o i! @sUddlZddlZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z eeZdZd Zd Zd d d dZdZdZdZdZdZddge egdZe ed<dedefddZdefddZdeeefdefdd Z d!e!dedefd"d#Z"d$e!de d%ed&e#ddf d'd(Z$dS))N)Union)subp)Cloud)Config) MetaSchema) PER_INSTANCErpi interfacesenable_usb_gadgetdo_spido_i2c do_onewire)spii2conewireserialdo_serial_cons do_serial_hwz/usr/bin/rpi-usb-gadgetz(Rebooting to apply config.txt changes...cc_raspberry_pizraspberry-pi-os)iddistros frequencyactivate_by_schema_keysmetaenablereturnc Cstd|tjtstdtdSztjt|rdnddgdddWd Stjy@}z td |WYd }~dSd }~ww) zEnable / Disable Raspberry Pi USB gadget mode when the script is present. Returns True when a reboot is required. zEnable rpi-usb-gadget mode: %sz#rpi-usb-gadget script not found: %sFonoffz-f)capturetimeoutTz&Failed to configure rpi-usb-gadget: %sN) LOGdebugospathexistsRPI_USB_GADGET_SCRIPTerrorrProcessExecutionError)rer*B/usr/lib/python3/dist-packages/cloudinit/config/cc_raspberry_pi.pyconfigure_usb_gadget(s&      r,cCs,z tgdWdStjyYdSw)N)/usr/bin/raspi-confignonint is_pifiveTF)rr(r*r*r*r+r/Ks r/cfgc Cs ddtdtfdd}d}d}t|tr||d}||d}nt|tr%|}ts,|r,d}ztd d tt|r7d nd gWntjyW}z t d |WYd}~dSd}~wwztd d t t|rcd nd gWdStjy}z t d|WYd}~dSd}~ww)auConfigure the serial (UART) interface on Raspberry Pi. Depending on the configuration, this enables or disables the serial console and the hardware UART by invoking `raspi-config` non-interactively. Args: cfg: Either a boolean or a dictionary: - If a bool, enables or disables the serial console and hardware UART together. - If a dict, may contain the keys `console` and `hardware` with boolean values controlling each individually. Returns: True if the configuration was applied successfully, False otherwise. Fcfg_dictnamecSs.|||}t|tstdt|||S|S)Nz"Invalid value for %s.serial.%s: %s)get isinstanceboolr!warningRPI_INTERFACES_KEY)r1r2defaultvalr*r*r+get_bool_fieldgs  z2configure_serial_interface..get_bool_fieldconsolehardwareTr-r.rz&Failed to configure serial console: %sNz'Failed to configure serial hardware: %s)F) dictstrr4r5r/rRASPI_CONFIG_SERIAL_CONS_FNr(r!r'RASPI_CONFIG_SERIAL_HW_FN)r0r:enable_console enable_hwr)r*r*r+configure_serial_interfaceSsJ        rDifacec Csz|tvs Jd|ztddt|t|rdndgWdStjy<}ztd||WYd}~d Sd}~ww) aEnable or disable a supported hardware interface. Supported interfaces are defined in `SUPPORTED_INTERFACES` and currently include SPI, I2C, and 1-Wire. The function applies the change using `raspi-config` in non-interactive mode. Args: iface: The interface name (e.g. `"spi"` or `"i2c"`). enable: True to enable the interface, False to disable it. Returns: True if the interface configuration succeeded, False otherwise. zUnsupported interface: r-r.rr=TzFailed to configure %s: %sNF)SUPPORTED_INTERFACESkeysrr?r(r!r')rErr)r*r*r+configure_interfaces" rHr2cloudargsc Cst|vrdSt|ttstdt|tdS|ts$tdtdSd}|tD]}|tkrJ|t|}t|trB|t|O}nt dtq*|t krt|t|tsetdt|t|dS|t|sttd|dSt |t| }|D]L}|t vr|tkrtdt |q|t||}|tkrt|ttfstdt ||n|t|O}qt|tr|t||O}qtdt ||qq*td|q*|dur|jjd d td } t| dSdS) NzInvalid value for %s: %szEmpty value for %s. Skipping...FzInvalid value for zInvalid key for %s: %szInvalid value for %s.%s: %szUnsupported key: %sTrebootnow)modedelaymessage) RPI_BASE_KEYr4r>r!r6r"ENABLE_USB_GADGET_KEYr5r, ValueErrorr7listrGrFSERIAL_INTERFACErDrHdistroshutdown_command REBOOT_MSGr) r2r0rIrJ want_rebootkeyrsubkeyssubkeycmdr*r*r+handles           "r])%loggingr#typingr cloudinitrcloudinit.cloudrcloudinit.configrcloudinit.config.schemarcloudinit.settingsr getLogger__name__r!rPr7rQrFrTr@rAr&rWr__annotations__r5r,r/r>rDr?rHrSr]r*r*r*r+s>        #N""