U M[^V@sVdZddlZddlmZddlZddlZddlZddlZddlZddl Z ddl m Z ddl m Z ddl m Z ddl mZddl mZdd l mZdd l mZdd lmZd Zd ZdZdZdZdZddgZddZdUddZddZedddZ ddZ!dd Z"d!d"Z#d#d$Z$d%d&Z%d'd(Z&ee d)eej'd*d+Z(d,d-ee j)e*e*d.d/d0Z+ee d1eej'd2d3Z,eee-d4d5d6Z.e j)e*e-d7d8d9Z/e j)ee*e-d:d;d<Z0e j)ed=d>d?Z1ed@dAZ2eedBdCZ3dDdEZ4dFdGZ5dVdHdIZ6eedJdKZ7dLdMZ8dWdNdOZ9dPdQZ:e:dXdRdSZ;e.new_frrrrrr assert_root-srcsfdd}|S)zDecorator asserting attached config. :param unattached_msg_tmpl: Optional msg template to format if raising an UnattachedError cstfdd}|S)NcsD|js:r.t|dd}j|d}t|}nt}|||S)NnameNone)r) is_attachedgetattrformatrZUnattachedError)rcfgrmsg exception)runattached_msg_tmplrrrAs   z/assert_attached..wrapper..new_frrr!rrwrapper@s z assert_attached..wrapperr)r!r#rr"rassert_attached9s r$cstfdd}|S)z&Decorator asserting unattached config.cs|jrt|||SN)rrAlreadyAttachedErrorrrrrrrUs z"assert_not_attached..new_frrrrrassert_not_attachedRsr( operationcsfdd}|S)z}Decorator ensuring that args.name is a valid service. :param operation: the operation name to use in error messages cstfdd}|S)Ncs>t|dr4|j}tj}|tjkr4t|j|d||S)Nr)r*r) hasattrr ua_statusZ'MESSAGE_INVALID_SERVICE_OP_FAILURE_TMPLrENTITLEMENT_CLASS_BY_NAMErUserFacingErrorr)rrrZtmpl)rr*rrres   z>require_valid_entitlement_name..wrapper..new_frrr)rrr#ds z/require_valid_entitlement_name..wrapperr)r*r#rr)rrequire_valid_entitlement_name^s r/cCs$d|_tjt|jd|_d|j_|S)z9Build or extend an arg parser for auto-attach subcommand. auto-attachrcommandFlags)prog USAGE_TMPLrNAMEusage _optionalstitleparserrrrauto_attach_parserusr<cCsJtjtdd|_d|_d|j_|jdddtd|jd d d d d |S)z4Build or extend an arg parser for attach subcommand.zattach r1attachr3token?z6token obtained for Ubuntu Advantage authentication: {})nargshelpz--no-auto-enableZ store_false auto_enablez4do not enable any recommended services automatically)actiondestrA) r5rr6r7r4r8r9 add_argumentUA_AUTH_TOKEN_URLr:rrr attach_parser}s"rGcCs6tjtdd}||_d|_d|j_|jdddd|S)z4Build or extend an arg parser for detach subcommand.detachr1r3z --assume-yes store_truez;do not prompt for confirmation before performing the detachrCrA)r5rr6r7r4r8r9rEr;r7rrr detach_parsersrLcCsBtjtddd}||_d|_d|j_d|j_|jdddd |S) z4Build or extend an arg parser for enable subcommand.enabler1 Argumentsr3rstorez2the name of the Ubuntu Advantage service to enablerJ r5rr6r7r4Z _positionalsr9r8rErKrrr enable_parsersrRcCsBtjtddd}||_d|_d|j_d|j_|jdddd |S) z5Build or extend an arg parser for disable subcommand.disabler1rNrOr3rrPz3the name of the Ubuntu Advantage service to disablerJrQrKrrrdisable_parsersrTc Cs\tjtdd}||_d|_tj|_t d|_ |j ddt t ddt ddd |j _|S) z4Build or extend an arg parser for status subcommand.rr1u Report current status of Ubuntu Advantage services on system. This shows whether this machine is attached to an Ubuntu Advantage support contract. When attached, the report includes the specific support contract details including contract name, expiry dates, and the status of each service on this system. The attached status output has four columns: * SERVICE: name of the service * ENTITLED: whether the contract to which this machine is attached entitles use of this service. Possible values are: yes or no * STATUS: whether the service is enabled on this machine. Possible values are: enabled, disabled, n/a (if your contract entitles you to the service, but it isn't available for this machine) or — (if you aren't entitled to this service) * DESCRIPTION: a brief description of the service The unattached status output instead has three columns. SERVICE and DESCRIPTION are the same as above, and there is the addition of: * AVAILABLE: whether this service would be available if this machine were attached. The possible values are yes or no. z--formatrPrz3output status in the specified format (default: {}))rCchoicesdefaultrAr3)r5rr6r7r4argparseRawDescriptionHelpFormatterformatter_classtextwrapdedent descriptionrESTATUS_FORMATSr8r9rKrrr status_parsers$ r^rScCs0tj|j}||}|r dnd}||S)z_Perform the disable action on a named entitlement. @return: 0 on success, 1 otherwise r)rr-rrSr)rrent_cls entitlementretrrraction_disables  rcFsilent_if_inapplicable)entitlement_namerrereturncCs*tj|}||}|j|d}||S)aPerform the enable action on a named entitlement. (This helper excludes any messaging, so that different enablement code paths can message themselves.) :param entitlement_name: the name of the entitlement to enable :param cfg: the UAConfig to pass to the entitlement :param silent_if_inapplicable: don't output messages when determining if an entitlement can be enabled on this system @return: True on success, False otherwise rd)rr-rMr)rfrrer`rarbrrr_perform_enables   rhrMc CsZttjzt|Wn,tjtjfk rDt j tj ddYnXt |j |rVdSdS)z^Perform the enable action on a named entitlement. @return: 0 on success, 1 otherwise T)exc_inforr_)printr,ZMESSAGE_REFRESH_ENABLErrequest_updated_contractr UrlErrorrr.loggingdebugMESSAGE_REFRESH_FAILURErhrr'rrr action_enables  rp)rgcCst||jdS)zXPerform the detach action for this machine. @return: 0 on success, 1 otherwise  assume_yes)_detachrrr'rrr action_detach+srt)rrrrgc Csg}tjD]"}||}|jddr ||q |rnt|dkrBdnd}td||D]}td|jqX|s~t s~dS|D]}|j ddqt |}|j d}|j d d d }||||ttjd S) a=Detach the machine from the active Ubuntu Advantage subscription, :param cfg: a ``config.UAConfig`` instance :param assume_yes: Assume a yes answer to any prompts requested. In this case, it means automatically disable any service during detach. @return: 0 on success, 1 otherwise T)Zsilentr_sz,Detach will disable the following service{}:z {}Z machineTokenmachineTokenInfo contractInfoidr)rZENTITLEMENT_CLASSESZ can_disableappendlenrjrrr Zprompt_for_confirmationrSrUAContractClient machine_tokenZdetach_machine_from_contractZ delete_cacher,ZMESSAGE_DETACH_SUCCESS) rrrZ to_disabler`Zentsuffixcontract_clientr}Z contract_idrrrrs5s*         rs)rr> allow_enablergc Csztj|||dWntjk rn}z:tt|W5QRXttj | WYdSd}~XYn>t j k r}zt |j| WYdSd}~XYnX|jddd}ttjj|dtd|dd S) zDCommon functionality to take a token and attach via contract backend)rr_Nrwrxr) contract_namer'r)rrkr rldisable_log_to_consolermr rjr,ZMESSAGE_ATTACH_FAILURErrr.Zwarningrr}ZMESSAGE_ATTACH_SUCCESS_TMPLr action_status)rr>rexcrrrr_attach_with_tokenVs4      r)rrgc Cs z t}Wn8tjk rD}z|jr0t||W5d}~XYnXt}|jr|d}||krpt|tdt |dddkrtt j t |}z|j|d}WnVt jk r}z4|jrd|jkrd krnn tt j|W5d}~XYnX|r|d||d S) aIDetect cloud_type and request a contract token from identity info. :param cfg: a ``config.UAConfig`` instance :raise NonAutoAttachImageError: When not on an auto-attach image type. :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. :return: contract token obtained from identity doc Nz instance-idz:Re-attaching Ubuntu Advantage subscription on new instanceTrqr)instanceiiZ contractToken)r Zcloud_instance_factoryrr.rr&Zget_instance_idZ read_cacherjrsr,Z!MESSAGE_DETACH_AUTOMATION_FAILURErr|Z"request_auto_attach_contract_tokenZContractAPIErrorcodeZNonAutoAttachImageErrorZMESSAGE_UNSUPPORTED_AUTO_ATTACHZ write_cache)rreZ current_iidZprev_iidrZ tokenResponserrr'_get_contract_token_from_cloud_identityus<        rcCst|}t||ddS)NTr>r)rr)rrr>rrraction_auto_attachsrcCs$|jsttjt||j|jdS)Nr)r>rr.r,ZMESSAGE_ATTACH_REQUIRES_TOKENrrBr'rrr action_attachsrc Cs4d}tg}ttj}|D]\}}|jr8d|j}nd}|j||j|d}t|dkrf| |qg}|}t|dkr| dd\}} | d| qn| |d d |gqtjttjd |tjtd d tjtd d d } | jddddd| j_| jdddd} d| _| jddd} | jtdt| | jddd} t| | jtd| jddd}t||jt d| jdd d}t!||jt"d| jd!d"d}t#||jt$d| jd#d$d}t%||jt&d| jd%d&d}|jt'd| jd'd(td}|jt(d| jd)d*d}|jt)d| S)+Nz - {name}: {description}{url}z ({})rv)rr\urlP r_rz  z [command]r1)r4rYr\r7Zepilogz--debugrIz&show all debug log messages to consolerJr3zAvailable Commandsr2)r9rDmetavarTrz/current status of all Ubuntu Advantage services)rA)rCr=z7attach this machine to an Ubuntu Advantage subscriptionr0zt |}t jjdks~dt jjkr|dd}t |dS)Nr ZexpireszUTF-8u—-r)rUAConfigrrr,ZUserFacingStatusZ INAPPLICABLEvaluestrrjr dumpsZformat_tabularsysstdoutencodingupperreplace)rrroutputrrrrs  rcCsttdSr%)rjr Z get_versionZ_argsZ_cfgrrrr)src Cslzt|WnNtjk r\}z.tt|W5QRXtt j W5d}~XYnXt t j dSr ) rrkr rlrrmr rr.r,rorjZMESSAGE_REFRESH_SUCCESS)rrrrrrr-s  rcCstdSr )rZ print_helprrrrr:s rc Cs|dkrtjd}t}tt}t}||d}|j D]L}t |dr@t |j dr@|j j dkr@||| ||dd}qq@|sttj}| ||||d||td krt|} | | d t|} | || ||| dS) z3Setup console logging and debug logging to log_fileNlog_fileFstreamrzconsoleTri)rZCONFIG_DEFAULTSr Z LogFormatterrmZ FormatterDEFAULT_LOG_FORMATZ getLoggerZsetLevelZhandlersr+rrZ setFormatterZset_nameZ StreamHandlerrstderrZ addHandlerrrpathlibPathZtouchchmodZ FileHandler) console_level log_levelrZconsole_formatterZ log_formatterrootZ stderr_foundZhandlerrZ log_file_pathZ filehandlerrrr setup_logging?s:                  rcsfdd}|S)Nc sz ||WStk rXttdW5QRXtdtjdtdYn6tj k r}zdt8|j |d}|j rt j }nt j }t|jf|W5QRXtt jtjdtdW5d}~XYntjk r@}zFtt|jW5QRXtd|jtjdt|jW5d}~XYnNtk rttdW5QRXtt jtjdtdYnXdS)NKeyboardInterruptzInterrupt received; exiting.)filer_)rerrorz{}z&Unhandled exception, please file a bug)rr rrmr rjrrexitrlrr,Z$LOG_CONNECTIVITY_ERROR_WITH_URL_TMPLZLOG_CONNECTIVITY_ERROR_TMPLrZMESSAGE_CONNECTIVITY_ERRORrr.rZ exit_code ExceptionZMESSAGE_UNEXPECTED_ERROR)rrrZmsg_argsZmsg_tmplfuncrrr#ds4      z#main_error_handler..wrapperr)rr#rrrmain_error_handlercs rcCs|s tj}t}|dd}|s:|tdtd|j|d}t}|j }|j r`t j nt j }t|||jt d||||S)Nr_z%Try 'ua --help' for more information.)rzExecuted with sys.argv: %r)rargvrZ print_usagerjr parse_argsrrrrnrmDEBUGINFOrrrC)Zsys_argvr;Z cli_argumentsrrrrrrrmains    r__main__)N)NN)N)N)>rrW functoolsrr rmrrrrZZuaclientrrrrrr,r r Zuaclient.cloudsr r6r5rZSTATUS_HEADER_TMPLrFrr]rr$r(rr/r<rGrLrRrTr^Z&MESSAGE_ENABLE_FAILURE_UNATTACHED_TMPLrcrboolrhrpintrtrsrrrrrrrrrrrr__name__rrrrrs           3 " 2  Z   $!