o K&br @sddlZddlZddlmZddlmZmZmZmZm Z m Z m Z ddl m Z edZe Zdejdeded dfd d Zdejd ejd dfd dZddddejdededefddZddddejdeedefddZdS)N)Optional)cloudsconfigcontract entitlements event_logger exceptionsmessages)identityz ua.actionscfgtoken allow_enablereturnc Csddlm}z tj|||dWn6tjy'}z ||||d}~wtjyF}ztj |j t j d||||d}~wwt }|rS|d|||dS)a: Common functionality to take a token and attach via contract backend :raise UrlError: On unexpected connectivity issues to contract server or inability to access identity doc from metadata service. :raise ContractAPIError: On unexpected errors when talking to the contract server. r)update_apt_and_motd_messages)r N)Z file_typez instance-id)Zuaclient.jobs.update_messagingrrZrequest_updated_contractrZUrlErrorstatusZUserFacingErroreventinfomsgsysstderrr Zget_instance_idZ write_cache)r r r rexcZ current_iidr2/usr/lib/python3/dist-packages/uaclient/actions.pyattach_with_tokens*    rcloudc Cst|}z|j|d}Wn&tjy3}z|jr-d|jkr$dkr-n|ttj|d}~ww|d}t ||dddS)aS :raise UrlError: On unexpected connectivity issues to contract server or inability to access identity doc from metadata service. :raise ContractAPIError: On unexpected errors when talking to the contract server. :raise NonAutoAttachImageError: If this cloud type does not have auto-attach support. )instanceiiNZ contractTokenT)r r ) rZUAContractClientZ"request_auto_attach_contract_tokenrZContractAPIErrorcodeZNonAutoAttachImageErrorr ZUNSUPPORTED_AUTO_ATTACHr)r rZcontract_clientZ tokenResponseer rrr auto_attach5s  rF) assume_yes allow_betanamerr cCs"t|}|||||d}|S)z Constructs an entitlement based on the name provided. Passes kwargs onto the entitlement constructor. :raise EntitlementNotFoundError: If no entitlement with the given name is found, then raises this error. )rr Z called_name)rZentitlement_factoryZenable)r r!rr Zent_clsZ entitlementrrrenable_entitlement_by_nameQs r")simulate_with_token show_betar#r$cCs6|r|j||d\}}||fS|j|d}d}||fS)z5 Construct the current UA status dictionary. )r r$)r$r)Zsimulate_statusr)r r#r$rretrrrres   r)ZloggingrtypingrZuaclientrrrrrrr Zuaclient.cloudsr Z getLoggerZLOGZget_event_loggerrZUAConfigstrboolrZAutoAttachCloudInstancerr"rrrrrsV $  !