Jh&ddlZddlmZmZmZmZmZddlmZm Z m Z m Z m Z ddl mZmZmZddlmZddlmZmZddlmZddlmZdd lmZdd lmZmZmZm Z m!Z!ejDe jFe$Z%Gd d eZ&Gd deeZ'dededejPdeeejPde)deejPeeff dZ*dedee+fdZ, d*de&deede'fdZ- d*de&dedeede'fdZ.edde.e&dZ/ddd e'ej`d!fejbd"fejdd#fejfd$fejhd%fejjd&fgd'd(d)Z6y)+N)IterableListOptionalTupleType) entitlementslockmessagesstatusutil)AbstractProgressProgressWrapper exceptions) APIEndpoint)AdditionalInfoErrorWarningObject)_enabled_services) _is_attached)UAConfig) BoolDataValue DataObjectFieldStringDataValue data_listcreZdZededededdededdgZd dd dedeede fd Z y ) EnableOptionsservicezPro service to be enableddocvariantFz2Optional variant of the Pro service to be enabled. access_onlyz^If true and the target service supports it, only enable access to the service (default: false)N)r r!c.||_||_||_yN)rr r!)selfrr r!s G/usr/lib/python3/dist-packages/uaclient/api/u/pro/services/enable/v1.py__init__zEnableOptions.__init__*s  &) __name__ __module__ __qualname__rrrfieldsstrrboolr&r'r%rrsq i.IJ   D     :  F."&! ' '# '  'r'rc eZdZedeededeededededeed gZdee dee de dee fd Z y ) EnableResultenabledz#List of services that were enabled.rdisabledz$List of services that were disabled.reboot_requiredz?True if one of the services that was enabled requires a reboot.r zaList of information message strings about the service that was just enabled. Possibly translated.c<||_||_||_||_yr#r1r2r3r )r$r1r2r3r s r%r&zEnableResult.__init__Ts!   .  r'N) r(r)r*rrrrr+rr,r-r&r.r'r%r0r06s  o &5   o &6        o &:  'F: !c !s) !  ! s) !r'r0cfgprogress entitlementavailable_variantsr!returnc d}|D]O}|||}|jdtjjk(s<|j sM|}n||j |j ||}||j dtjdtjjtjj|jztjjzifgtj j|j}|t#|j$|j&d|jifS|dfS) N)r6r!rmessage_operationmsg)r ) variant_namer>)titlecodemeta)applicability_statusrApplicabilityStatus APPLICABLEvariant_auto_selectdefault_variantemitr prompt_for_confirmationr AUTO_SELECTING_VARIANTformatTxtColorBOLDr>ENDCAUTO_SELECTED_VARIANT_WARNINGrr=name) r6r7r8r9r!r v_clsvwarnings r%_auto_select_variantrSbsgG# c{ 3 " " $Q '//:: ;%%'G ;66B--#;-O 00x>>EE$,$5$5$:$:%22%3&//44%5 F    88?? --@ *++ '"6"67   D  r'cft|jDcgc]}|jc}Scc}wr#)renabled_servicesrO)r6ss r%_enabled_services_namesrWs%-c2CC DqAFF DD Ds.optionsprogress_objectc.t|t|S)N)rY)_enabler)rXrYs r%enabler\s 7HJ HHr'c t|}g}tjstjt |j stjjdk(rtjt|}tfdt|jDd}|rtggdgStj |jj"xsdj$}|j'\}} |j(} t+|tj,j.k(|j0 | gr>t3|||| j5j$\}} | r|j7| |j9|_d} d} t=j>d 5|jA|\} } ddd| sL| | jF | jF}ntHjJ}tjLj| t|}|jNjQd gxsgDcgc]}tS|tTr|}}tWjV| |jYtt[t]t_|jat_|t[t]t_|jat_||jc|}||_2|S#1swYRxYw#tB$r}t=jD|d}~wwxYwcc}w)z Enable a Pro service. This will automatically disable incompatible services and enable required services that that target service depends on. landscapec3K|]P}|jjk(r5jr%|jr|jjk(r|Rywr#)rOrr variant_enabledr>).0rVrXs r% z_enable..sH vv(OO%%!..GOO*K  sAANFr5)r6rOr r!)r6r7r8r9r!zu.pro.services.enable.v1) lock_holder)rreason post_enable)r6)3rr we_are_currently_rootrNonRootUserErrorr is_attachedUnattachedErrorr NotSupportedrWnextrrUr0rentitlement_factoryr r!rBvariantsallrCrD is_variantrSvaluesappendcalculate_total_enable_steps total_stepsr RetryLockr\ Exceptionclear_lock_file_if_presentmessager GENERIC_UNKNOWN_ISSUEEntitlementNotEnabledError messagingget isinstancer,r finishsortedlistset difference_check_for_rebootwarnings)rXr6rYr7renabled_services_beforealready_enabledr8 applicability_r9auto_select_warningsuccess fail_reasonereenabled_services_afterr=post_enable_messagesresults` r%r[r[s6/HH  % % '))++   ( (((**+%%%''5c: &s+<<   O!   22  __%2'' K#779M1$-- \==HH H&& &  ,@#188:++ , ( (  OO/ 0&CCEHGK ^^2  @$/#5#5h#? G[ @   "{':':'F ((F33F33OOF  5S9 ))--mR@FB  c3    MMc OO  *+66/0   +,77./  $557%!F$FO M_ @ @  '')s6-L5L(L5M(L2-L55 M>MMv1 EnableServiceT)versionrOfn options_clssupports_progress32zw from uaclient.api.u.pro.services.enable.v1 import enable, EnableOptions result = enable(EnableOptions(service="usg")) zWhen called as non-root userzCWhen called on a machine that is not attached to a Pro subscriptionz_When called for a service that doesn't support being enabled via API (currently only Landscape)zxWhen the service argument is not a valid Pro service name or if the variant is not a valid variant of the target servicez8When another Ubuntu Pro related operation is in progressz When the service fails to enablez3pro api u.pro.services.enable.v1 --args service=usgzp { "disabled": [], "enabled": [ "usg" ], "messages": [], "reboot_required": false } ) introduced_inrequires_networkexample_python result_classr example_cli example_jsonr#)7loggingtypingrrrrruaclientrr r r r uaclient.apir rruaclient.api.apiruaclient.api.data_typesrr-uaclient.api.u.pro.status.enabled_services.v1r(uaclient.api.u.pro.status.is_attached.v1ruaclient.configruaclient.data_typesrrrrr getLoggerreplace_top_level_logger_namer(LOGrr0 UAEntitlementr-rSr,rWr\r[endpointrhrjrkEntitlementNotFoundError LockHeldErrorrz_docr.r'r%rs88??FF(FKA$g:::8DE'J'@)!:~)!X,! ,!,!++,!!l&@&@!AB ,!  ,!  < % %x0B'C CD ,!^EEd3iE KOI I-56F-GII37{ { {./{ {|     !  $ $&DE  & &    # #6   / /O   $ $ F  1 1 . 7@I S3r'