Hcf# ddlZddlmZmZddlmZmZmZmZm Z ddl m Z m Z m Z ddlmZddlmZddlmZddlmZdd lmZdd lmZmZmZmZmZej<e j>e Z!Gd d eZ"Gd deeZ#dedee$fdZ% dde"dee de#fdZ& dde"dedee de#fdZ'edde'e"dZ(y)N)ListOptional) entitlementslockmessagesstatusutil)AbstractProgressProgressWrapper exceptions) APIEndpoint)AdditionalInfo)_enabled_services) _is_attached)UAConfig) BoolDataValue DataObjectFieldStringDataValue data_listcfeZdZedeededededgZddddedeede fdZ y) EnableOptionsservicevariantF access_onlyN)rrc.||_||_||_yN)rrr)selfrrrs G/usr/lib/python3/dist-packages/uaclient/api/u/pro/services/enable/v1.py__init__zEnableOptions.__init__s  &) __name__ __module__ __qualname__rrrfieldsstrrboolr r!rrrsZ i) i%0 m]E2F"&! ' '# '  'r!rc eZdZedeeedeeedeedeegZdee dee de dee fdZ y) EnableResultenableddisabledreboot_requiredrc<||_||_||_||_yrr+r,r-r)rr+r,r-rs rr zEnableResult.__init__1s!   .  r!N) r"r#r$rrrrr%rr&r'r r(r!rr*r*)sz i?34 j)O45 / j)O45 F !c !s) !  ! s) !r!r*cfgreturncft|jDcgc]}|jc}Scc}wr)renabled_servicesname)r0ss r_enabled_services_namesr6?s%-c2CC DqAFF DD Ds.optionsprogress_objectc.t|t|S)N)r8)_enabler)r7r8s renabler;Cs 7HJ HHr!c |t|}tjstjt |j stj|jdk(rtjt|}|j|vrtggdgStj||j|jxsd}||dd|j|j}|j!|_d}d} t%j&d 5|j)|\}}ddd|sL||j. |j.} nt0j2} tj4|j| t|} |j6j9d gxsgD cgc]} t;| t<r| } } t?j>| |jAttCtEtG| jItG|tCtEtG|jItG| |jK| S#1swYIxYw#t*$r} t%j,| d} ~ wwxYwcc} w) N landscapeFr/)r0r4rT) assume_yes allow_beta called_namerzu.pro.services.enable.v1) lock_holder)rreason post_enable)r0)&r r we_are_currently_rootr NonRootUserErrorr is_attachedUnattachedErrorr NotSupportedr6r*rentitlement_factoryrrcalculate_total_enable_steps total_stepsr RetryLockr; Exceptionclear_lock_file_if_presentmessagerGENERIC_UNKNOWN_ISSUEEntitlementNotEnabledError messagingget isinstancer&rfinishsortedlistset difference_check_for_reboot)r7r0r8progressenabled_services_beforeent_cls entitlementsuccess fail_reasonerCenabled_services_aftermsgpost_enable_messagess rr:r:Is /H  % % '))++   ( (((**+%%%''5c:11!   .. goow/D"G OO'' K'CCEHGK ^^2  @$/#5#5h#? G[ @   "{':':'F ((F33F33OOF  5S9 ))--mR@FB  c3    MMc OO  *+66/0   +,77./  $557%! 9 @ @  '')s6J)J>JJ9J J J6J11J6v1 EnableServiceT)versionr4fn options_clssupports_progressr))loggingtypingrruaclientrrrrr uaclient.apir r r uaclient.api.apir uaclient.api.data_typesr-uaclient.api.u.pro.status.enabled_services.v1r(uaclient.api.u.pro.status.is_attached.v1ruaclient.configruaclient.data_typesrrrrr getLoggerreplace_top_level_logger_namer"LOGrr*r&r6r;r:endpointr(r!rrzs !??FF(2KA$g:::8DE'J'&!:~!,EEd3iE KOI I-56F-GII37X X X./X Xv     r!