i! 0UddlZddlZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z ejeZdZd Zd Zd d d dZdZdZdZdZdZddge egdZe ed<dedefdZdefdZdeeefdefdZ de!dedefdZ"d e!de d!ed"e#ddf d#Z$y)$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`tjd|tjj t stj dt y tjt |rdnddgddy #tj$r }tj d |Yd }~yd }~wwxYw) 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)res B/usr/lib/python3/dist-packages/cloudinit/config/cc_raspberry_pi.pyconfigure_usb_gadgetr-(s II.7 77>>/ 0 79NO %E    % % :A>s#A::B- B((B-cf tjgdy#tj$rYywxYw)N)/usr/bin/raspi-confignonint is_pifiveTF)rr*r,r1r1Ks/ BC  % %s 00cfgc8ddtdtfd}d}d}t|tr||d}||d}nt|tr|}t s|rd} t j dd t t|rd nd g t j dd tt|rd nd gy#t j$r }tjd |Yd }~yd }~wwxYw#t j$r }tjd|Yd }~yd }~wwxYw)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_dictnamec|j||}t|tstj dt |||S|S)Nz"Invalid value for %s.serial.%s: %s)get isinstanceboolr#warningRPI_INTERFACES_KEY)r6r7defaultvals r,get_bool_fieldz2configure_serial_interface..get_bool_fieldgs@ll4)#t$ KK4"   N r3consolehardwareTr/r0rz&Failed to configure serial console: %sNz'Failed to configure serial hardware: %s)F) dictstrr:r;r1rRASPI_CONFIG_SERIAL_CONS_FNr*r#r)RASPI_CONFIG_SERIAL_HW_FN)r4r@enable_console enable_hwr+s r,configure_serial_interfacerJSs(  S NI#t'Y7"3 3 C  ;>   '+AQ/     ')A*     % % :A>  % % ;Q?s0*B0*C&0C#CC#&D9DDifacec|tjvs Jd| tjddt|t|rdndgy#tj$r!}t j d||Yd}~y d}~wwxYw) 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/r0rrCTzFailed to configure %s: %sNF)SUPPORTED_INTERFACESkeysrrEr*r#r))rKrr+s r,configure_interfacerOs %**,,) (),  '$U+AQ'     % % .q9s-AB#A??Br7cloudargsct|vryt|tts#tj dt|ty|tstj dtyd}|tD]}|t k(r>|t|}t|tr|t|z}ntdt K|tk(rbt|t|ts'tj dt|t|y|t|stj d|yt|t|j}|D]}|tjvr%|tk7rtj dt|>|t||}|tk(rBt|ttfstj dt||n|t|z}t|tr|t!||z}tj dt||tj d||dur8|j"j%d d t& } t)j(| yy) 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_KEYr:rDr#r<r$ENABLE_USB_GADGET_KEYr;r- ValueErrorr=listrNrMSERIAL_INTERFACErJrOdistroshutdown_command REBOOT_MSGr) r7r4rPrQ want_rebootkeyrsubkeyssubkeycmds r,handleres[3 L)4 0 &       3\BK< ; ' '&s+F&$'3F;;  #56K5L!MNN  & &c,/4d; .  %c* &s+ ;SA3|,S16689G" "6";";"=="22KK02Df\*3/7--%ftTl; 9."" $'A&'II fd+#6vv#FFKKK5* 7 D KK-s 3 w;zdll++,  #r3)%loggingr%typingr cloudinitrcloudinit.cloudrcloudinit.configrcloudinit.config.schemarcloudinit.settingsr getLogger__name__r#rXr=rYrMr\rFrGr(r_r__annotations__r;r-r1rDrJrErOr[rer2r3r,rps !#.+g! !+   .*1 7  !" ,~ j  $ F4KE$*$5K$K\sDTDTT6T%TtTTr3