File manager - Edit - /usr/lib/python3/dist-packages/uaclient/__pycache__/cli.cpython-310.pyc
Back
o =�xd� � @ s| d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl m Z ddl mZmZmZ ddlmZmZmZmZmZmZmZmZmZmZm Z ddlm Z ddlm Z m Z m!Z! ddlm"Z# dd lm$Z$m%Z%m&Z& dd l'm(Z( dd l)m*Z*m+Z+ dd l,m-Z- dd l.m/Z/m0Z0 ddl1m2Z2m3Z3 ddl4m5Z5 ddl6m7Z7 ddl8m9Z9m:Z: ddl;m<Z<m=Z= ddl>m?Z? ddl@mAZAmBZBmCZC ddlDmEZEmFZFmGZGmHZH ddlImJZJmKZK ddlLmMZM ddlNmOZO ddlPmQZQmRZR ddlSmTZTmUZU dZVd ZWd ZXd ZYd ZZd Z[g d!�Z\d"Z]d#e j^ d$ Z_e�`� ZaG d%d&� d&ejb�Zcd�d'd(�Zdd)d*� Zed+d,� Zfd�d-d.�Zgd/d0� Zhd1d2� Zid3d4� Zjd5d6� Zkd7elfd8d9�Zmd7elfd:d;�Znd7elfd<d=�Zod>d?� Zpd@dA� ZqdBdC� ZrdDdE� ZsdFdG� ZtdHdI� ZudJdK� ZvdLdM� ZwdNejxfdOdP�ZydNejxfdQdR�ZzdNejxfdSdT�Z{dUdV� Z|dWdX� Z}dYdZ� Z~dNejxd[eld\elfd]d^�Zd_d`�dadb�Z�dcdd� Z�dedf� Z�eedgdh� �Z�eedidj� �Z�dkdl� Z�efeeege��eddm�dndo� ����Z�efeeege��eddp�dqdr� ����Z�efeeeg� edds�dte�fdudv�����Z�dNejxdwe�dte�fdxdy�Z�dNejxdtdfdzd{�Z�d|d}� Z�eedNejxdte�fd~d��Z�d�d�� Z�eheeedd��d�d�� ���Z�dNejxfd�d��Z�dNejxfd�d��Z�dNejxfd�d��Z�d�d�� Z�dNejxfd�d��Z�d�d�d��Z�dNejxfd�d��Z�eg� dNejxfd�d���Z�dNejxfd�d��Z�eeedd��dNejxfd�d����Z�dNejxd�e9d�eld�eel dtdf d�d��Z�d�d�� Z�d�d�d�d��Z�d�d�d��Z�d�d�d��Z�d�d�� Z�d�d�� Z�e�d�d�d���Z�e�d�k�r�e��e�� � dS dS )�z2Client to manage Ubuntu Pro services on a machine.� N��wraps)�List�Optional�Tuple) �actions�apt�apt_news�config�contract�daemon�defaults� entitlements� event_logger� exceptions�lock)�log)�messages�security�security_status)�status)�timer�util�version)�call_api)�FullAutoAttachOptions�_full_auto_attach)� _initiate)�MagicAttachRevokeOptions�_revoke)�MagicAttachWaitOptions�_wait)�_reboot_required)� _is_attached)� AptProxyScope�setup_apt_proxy)�AttachActionsConfigFile�IncorrectTypeError)�PRINT_WRAP_WIDTH)�,create_enable_entitlements_not_found_message�entitlements_disable_order�get_valid_entitlement_names)�ApplicationStatus�CanDisableFailure�CanEnableFailure�CanEnableFailureReason)�notices� state_files)�Notice)�JsonArrayFormatter)� refresh_motd�update_motd_messages)� safe_dump� safe_load�proz{name} {command} [flags]zAUse {name} {command} --help for more information about a command.z&Try 'pro --help' for more information.z}Account: {account} Subscription: {subscription} Valid until: {contract_expiry} Technical support level: {tech_support_level} z$https://auth.contracts.canonical.com)�tabular�json�yamlzua_logs.tar.gz� z� A new version is available: {version} Please run: sudo apt-get install ubuntu-advantage-tools to get the latest version with new features and bug fixes.c sl e Zd Zdddejdfdee f� fdd� Zdd� Zd � fdd � Z e d e e e e e f fd d ��Z � ZS )�UAArgumentParserN� base_descc s t � j||||d� || _d S )N)�prog�usage�epilog�formatter_class)�super�__init__r> )�selfr? r@ rA rB r> �� __class__� �./usr/lib/python3/dist-packages/uaclient/cli.pyrD i s � zUAArgumentParser.__init__c C s, | � tj� |dkr t}| �d|d � d S )Nz&the following arguments are required: � r< )� print_usage�sys�stderr�TRY_HELP�exit)rE �messagerH rH rI �errorz s zUAArgumentParser.errorFc sf | j r*t�� \}}t|�}|rt|| �}d�| j g| �| _| jdtjj 7 _t � j |d� d S )Nr< � ��file) r> r= �_get_service_descriptions�sorted�join� descriptionr �PRO_HELP_SERVICE_INFO�msgrC � print_help)rE rT �show_all�non_beta_services_desc�beta_services_desc�service_descriptionsrF rH rI r[ � s �� �zUAArgumentParser.print_help�returnc C s� t �� } d}g }g }t�t �� �}|D ]K}z tj| |d d�}W n tjy* Y qw |�d|d �}|j r=d� |j �}nd}t j |j ||j |d�tdd d d �} |jrY|�| � q|�| � q||fS ) Nz - {name}: {description}{url}�name��cfgra � presentedAsz ({})� )ra rX �urlz F)�width�subsequent_indent�break_long_words�break_on_hyphens)r �UAConfigr �get_available_resourcesr �entitlement_factoryr �EntitlementNotFoundError�get� help_doc_url�format�textwrap�fillrX r( �is_beta�append) rc �service_info_tmplr] r^ � resources�resource�ent_cls�presentation_namerf � service_inforH rH rI rU � s@ ���� z*UAArgumentParser._get_service_descriptions)NF)�__name__� __module__� __qualname__�argparse� HelpFormatterr �strrD rQ r[ � staticmethodr r rU � __classcell__rH rH rF rI r= h s ��(r= c � � fdd�}|S )z1Decorator asserting exclusive access to lock filec � t � �� �fdd��}|S )Nc sF t j| �d�� � |d| i|��}W d � |S 1 s w Y |S )N)rc � lock_holderrc )r �SingleAttemptLock)rc �args�kwargs�retval)�fr� rH rI �new_f� s ��z0assert_lock_file.<locals>.wrapper.<locals>.new_fr �r� r� �r� �r� rI �wrapper� s z!assert_lock_file.<locals>.wrapperrH )r� r� rH r� rI �assert_lock_file� s r� c � t � �� fdd��}|S )z Decorator asserting root userc s t �� st�� �� | i |��S �N)r �we_are_currently_rootr �NonRootUserError)r� r� r� rH rI r� � s zassert_root.<locals>.new_fr r� rH r� rI � assert_root� s r� c r� )z>Decorator to verify if correct params are used for json formatc sX | s � | g|�R i |��S | j dkr!| js!tj}tj|j|jd��� | g|�R i |��S )Nr: �rZ �msg_code)rq � assume_yesr � JSON_FORMAT_REQUIRE_ASSUME_YESr �UserFacingErrorrZ ra )�cmd_argsr� r� rZ r� rH rI r� � s z&verify_json_format_args.<locals>.new_fr r� rH r� rI �verify_json_format_args� s r� c r� )z�Decorator asserting attached config. :param msg_function: Optional function to generate a custom message if raising an UnattachedError c r� )Nc sb t |�js'�r!t| dd�}t| dd�}�|||d�}t�|�}|�t�� }|�� | fd|i|��S )N�commandre �service)r� � service_namesrc rc )r# � is_attached�getattrr �UnattachedError)r� rc r� r� r� rZ � exception)r� � msg_functionrH rI r� � s � �z/assert_attached.<locals>.wrapper.<locals>.new_fr r� �r� r� rI r� � s z assert_attached.<locals>.wrapperrH )r� r� rH r� rI �assert_attached� s r� c r� )z&Decorator asserting unattached config.c s, t |�jrt�|jj�dd���� | |d�S )Nra re �rc )r# r� r �AlreadyAttachedError�machine_token_file�accountro �r� rc r� rH rI r� s � z"assert_not_attached.<locals>.new_fr r� rH r� rI �assert_not_attached s r� c C s4 d| _ d| _| jdddd� | jddg d d d � | S ) z5Build or extend an arg parser for the api subcommand.�api� Calls the Client API endpoints.� endpoint_path�endpointzAPI endpoint to call)�metavar�helpz--args�options�*z;Options to pass to the API endpoint, formatted as key=value)�dest�default�nargsr� )r? rX � add_argument��parserrH rH rI � api_parser s ��r� c C s* d| _ d| _tjt| j d�| _d| j_| S )z9Build or extend an arg parser for auto-attach subcommand.� auto-attachz5Automatically attach on an Ubuntu Pro cloud instance.�ra r� �Flags)r? rX � USAGE_TMPLrq �NAMEr@ � _optionals�titler� rH rH rI �auto_attach_parser- s �r� c C s2 d| _ d| _tjt| j d�| _| jdddd� | S )z<Build or extend an arg parser for 'collect-logs' subcommand.� collect-logsz<Collect logs and relevant system information into a tarball.r� z-oz--outputzEtarball where the logs will be stored. (Defaults to ./ua_logs.tar.gz)�r� )r? rX r� rq r� r@ r� r� rH rH rI �collect_logs_parser8 s ��r� �parent_commandc C s6 t jtd�|�d�| _d| _d| _| jdddd� | S ) z;Build or extend an arg parser for 'config show' subcommand.z {} show [key]r� �showz(Show customisable configuration settings�key�?z6Optional key or key(s) to show configuration settings.�r� r� )r� rq r� r@ r? rX r� �r� r� rH rH rI �config_show_parserJ s ��r� c C sJ t jtd�|�d�| _d| _d| _d| j_| jdd�d� t j ��d � | S ) z:Build or extend an arg parser for 'config set' subcommand.z{} set <key>=<value>r� �asetz/Set and apply Ubuntu Pro configuration settingsr� �key_value_pairzKkey=value pair to configure for Ubuntu Pro services. Key must be one of: {}�, r� ) r� rq r� r@ r? rX r� r� r� rW r �UA_CONFIGURABLE_KEYSr� rH rH rI �config_set_parserY s � �� r� c C sL t jtd�|�d�| _d| _d| _| jdd�d�tj ��dd� d | j _ | S ) z<Build or extend an arg parser for 'config unset' subcommand.z{} unset <key>r� �unsetz&Unset Ubuntu Pro configuration settingr� z?configuration key to unset from Ubuntu Pro services. One of: {}r� )r� r� r� ) r� rq r� r@ r? rX r� rW r r� r� r� r� rH rH rI �config_unset_parserm s ��r� c C s� d}t jtd�|�d�| _|| _d| _d| j_| jdddd �}|j d d d �}|j t d � t ||d� |j ddd �}|j t d � t||d� |j ddd �}|j td � t||d� | S )z4Build or extend an arg parser for config subcommand.r z {} <command>r� z Manage Ubuntu Pro configurationr� �Available Commandsr� re �r� r� r� r� z,show all Ubuntu Pro configuration setting(s)r� ��action)r� �setz$set Ubuntu Pro configuration settingr� z&unset Ubuntu Pro configuration setting)r� rq r� r@ r? rX r� r� �add_subparsers� add_parser� set_defaults�action_config_showr� �action_config_setr� �action_config_unsetr� )r� r� � subparsers� parser_show� parser_set� parser_unsetrH rH rI � config_parser� s4 ��� � � r� c C s� t jtdd�| _tj| _d| _d�tj �}|d d | _ d| j _ | j dd d �t�d � | j d d ddd� | j dt�d�dd� | j ddddgddd� | S )z4Build or extend an arg parser for attach subcommand.zattach <token>r� �attachz@Attach this machine to Ubuntu Pro with a token obtained from: {}rR z�When running this command without a token, it will generate a short code and prompt you to attach the machine to your Ubuntu Pro account using a web browser.r� �tokenr� z0token obtained for Ubuntu Pro authentication: {}r� z--no-auto-enable� store_false� auto_enablez4do not enable any recommended services automatically)r� r� r� z--attach-config�rzKuse the provided attach config file instead of passing the token on the cli)�typer� �--format�store�clir: �4output enable in the specified format (default: cli)�r� �choicesr� r� )r� rq r� r@ r �RawDescriptionHelpFormatterrB r? r � BASE_UA_URLrX r� r� r� �UA_AUTH_TOKEN_URL�FileType)r� r> rH rH rI � attach_parser� sN ���� �����r� c C sV t jtdd�| _d| _d| _d| j_| jddd� | jd d d d � | jd d dd � | S )z1Build or extend an arg parser for fix subcommand.z"fix <CVE-yyyy-nnnn+>|<USN-nnnn-d+>r� �fixzLInspect and resolve CVEs and USNs (Ubuntu Security Notices) on this machine.r� �security_issuezwSecurity vulnerability ID to inspect and resolve on this system. Format: CVE-yyyy-nnnn, CVE-yyyy-nnnnnnn or USN-nnnn-ddr� z --dry-run� store_truezrIf used, fix will not actually run but will display everything that will happen on the machine during the command.�r� r� z --no-relatedz`If used, when fixing a USN, the command will not try to also fix related USNs to the target USN.� r� rq r� r@ r? rX r� r� r� r� rH rH rI � fix_parser� s, ���� � r� c C sx d| _ tj| _t�d�| _| jddddd� | �� }|jdd d d � |jd d d d � |jddd d � |jddd d � | S )z=Build or extend an arg parser for security-status subcommand.�security-statusa] Show security updates for packages in the system, including all available Expanded Security Maintenance (ESM) related content. Shows counts of how many packages are supported for security updates in the system. If called with --format json|yaml it shows a summary of the installed packages based on the origin: - main/restricted/universe/multiverse: packages from the Ubuntu archive - esm-infra/esm-apps: packages from the ESM archive - third-party: packages installed from non-Ubuntu sources - unknown: packages which don't have an installation source (like local deb packages or packages for which the source was removed) The output contains basic information about Ubuntu Pro. For a complete status on Ubuntu Pro services, run 'pro status'. r� zFormat for the output)r: r; �textr� )r� r� r� z --thirdpartyz7List and present information about third-party packagesr� )r� r� z --unavailablez7List and present information about unavailable packagesz --esm-infraz5List and present information about esm-infra packagesz --esm-appsz4List and present information about esm-apps packages) r? r r� rB rr �dedentrX r� � add_mutually_exclusive_group)r� �grouprH rH rI �security_status_parser� sB ������r c C sT d| _ d| _tjtdd�| _d| j_tj | _ t � d�| _| j dg d�d d d d � | S ) z5Build or extend an arg parser for refresh subcommand.�refreshz9Refresh existing Ubuntu Pro contract and update services.z"refresh [contract|config|messages]r� r� a� Refresh three distinct Ubuntu Pro related artifacts in the system: * contract: Update contract details from the server. * config: Reload the config file. * messages: Update APT and MOTD messages related to UA. You can individually target any of the three specific actions, by passing it's target to nome to the command. If no `target` is specified, all targets are refreshed. �target)r r r r� NzTarget to refresh.)r� r� r� r� )r? rX r� rq r� r@ r� r� r r� rB rr r� r� r� rH rH rI �refresh_parser2 s& ��� �r c K s� | j dkr2| jrt�� dS | jrt�� dS | jr!t�|� dS | jr+t� |� dS t�|� dS | j dkrGt t j t� |�dtjd�� dS t tt� |�dd�� dS )Nr� r: T)� sort_keys�clsF)�default_flow_styler )rq � thirdpartyr �list_third_party_packages� unavailable�list_unavailable_packages� esm_infra�list_esm_infra_packages�esm_apps�list_esm_apps_packages�printr: �dumps�security_status_dictr �DatetimeAwareJSONEncoderr6 �r� rc r� rH rH rI �action_security_statusU s> �� � � ������r c K sD t �tj| j�sd�| j�}t�|��tj|| j| j | j d�}|j S )NzYError: issue "{}" is not recognized. Usage: "pro fix CVE-yyyy-nnnn" or "pro fix USN-nnnn")rc �issue_id�dry_run� no_related) �re�matchr �CVE_OR_USN_REGEXr� rq r r� �fix_security_issue_idr r � exit_code)r� rc r� rZ � fix_statusrH rH rI � action_fixs s � �r c C sT t jtdd�}|| _d| _d| _d| j_| jdddd� | jd d d d gd d d� | S )z4Build or extend an arg parser for detach subcommand.�detachr� z-Detach this machine from Ubuntu Pro services.r� � --assume-yesr� z;do not prompt for confirmation before performing the detachr� r� r� r� r: r� r� r� �r� r@ rH rH rI � detach_parser� s$ ��r$ rc c C s� t jtdd�}|| _d| _d| _d| j_| jdddd �d � t j |d ���d � | jd dt t d d�t d �d� | jdddd� | S )z2Build or extend an arg parser for help subcommand.zhelp [service]r� r� z7Provide detailed information about Ubuntu Pro services.� Argumentsr� r� r� z-a service to view help output for. One of: {}r� r� �r� r� r� r� r z1output help in the specified format (default: {})r� �--allr� �(Allow the visualization of beta servicesr� ) r� rq r� r@ r? rX � _positionalsr� r� rW r �valid_services�STATUS_FORMATS�r� rc r@ rH rH rI � help_parser� s: ��� �� �r- c C s� t jtdd�}d| _|| _d| _d| j_d| j_| j ddd d �d � t j |d ���d � | j dddd� | j dddd� | j dddd� | j ddddgddd� | j ddd d� | S ) z4Build or extend an arg parser for enable subcommand.z enable <service> [<service>]r� z Enable an Ubuntu Pro service.�enabler% r� r� r� �+z<the name(s) of the Ubuntu Pro services to enable. One of: {}r� r� r& r" r� z;do not prompt for confirmation before performing the enabler� � --access-onlyzHdo not auto-install packages. Valid for cc-eal, cis and realtime-kernel.z--betaz allow beta service to be enabledr� r� r: r� r� � --variantz8The name of the variant to use when enabling the service� r� rq r� rX r@ r? r) r� r� r� rW r r* r, rH rH rI � enable_parser� sV ��� �����r3 c C s� t jtdd�}d| _|| _d| _d| j_d| j_| j ddd d �d � t j |d ���d � | j dddd� | j ddddgddd� | S )z5Build or extend an arg parser for disable subcommand.z disable <service> [<service>]r� z Disable an Ubuntu Pro service.�disabler% r� r� r� r/ z=the name(s) of the Ubuntu Pro services to disable. One of: {}r� r� r& r" r� z<do not prompt for confirmation before performing the disabler� r� r� r: z5output disable in the specified format (default: cli)r� r2 r, rH rH rI �disable_parser� s< ��� ��r5 c C sZ t jtdd�| _d| _d| _d| j_| jdddd �}|j d d d �}|j t d � t |� | S )z4Build or extend an arg parser for system subcommand.zsystem <command>r� z9Output system related information related to Pro services�systemr� r� r� re r� �reboot-requiredz#does the system need to be rebootedr� r� ) r� rq r� r@ rX r? r� r� r� r� r� � action_system_reboot_required�reboot_required_parser)r� r� �parser_reboot_requiredrH rH rI � system_parser s ��� r; c C s. t jtdd�| _d| _tj| _t� d�| _ | S )Nzsystem reboot-requiredr� r7 al Report the current reboot-required status for the machine. This command will output one of the three following states for the machine regarding reboot: * no: The machine doesn't require a reboot * yes: The machine requires a reboot * yes-kernel-livepatches-applied: There are only kernel related packages that require a reboot, but Livepatch has already provided patches for the current running kernel. The machine still needs a reboot, but you can assess if the reboot can be performed in the nearest maintenance window. ) r� rq r� r@ r8 r r� rB rr r� rX r� rH rH rI r9 - s ��r9 c C s� t jtdd�}|| _d| _d| _tj| _t � d�| _| j ddddd � | j d d t t d d �t d �d� | j ddd dd� | j dddd� d| j _| S )z4Build or extend an arg parser for status subcommand.r r� z6Output the status information for Ubuntu Pro services.u; Report current status of Ubuntu Pro 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. If --simulate-with-token is used, then the output has five columns. SERVICE, AVAILABLE, ENTITLED and DESCRIPTION are the same as mentioned above, and AUTO_ENABLED shows whether the service is set to be enabled when that token is attached. If the --all flag is set, beta and unavailable services are also listed in the output. z--waitr� Fz Block waiting on pro to complete)r� r� r� r� r� r z3output status in the specified format (default: {})r� z--simulate-with-token�TOKENz1simulate the output status using a provided token)r� r� r� r' r( r� r� )r� rq r� r@ rX r? r r� rB rr r� r� r+ r� r� r# rH rH rI � status_parserG sJ ��%��� ��r= �cmd_name� subcmd_namec C sf t | d�}|�� d j| }|�� d j�� }||vr1|�� d j| �� t�d�d�|����d S )Nr� r z <command> must be one of: {}r� ) � get_parser�_get_positional_actionsr� �keysr[ r r� rq rW )rc r> r? r� � subparser� valid_choicesrH rH rI �_print_help_for_subcommand� s ��rE T)� update_statusc C s� | j }|dur |} | �� \}}|s:t�| j� |dur9t|t�r9|jdur9t�|jj � tj |jj |jj| jd� nt� | j� |rHt j |d� |S )a= Perform the disable action on a named entitlement. :param entitlement_name: the name of the entitlement to enable :param cfg: the UAConfig to pass to the entitlement :param assume_yes: Assume a yes response for any prompts during service enable @return: True on success, False otherwise N�� error_msg� error_coder� r� )�enabled_variantr4 �event�service_failedra � isinstancer- rP �inforZ rQ �service_processed� ua_statusr )� entitlementrc r� rF �variant�ret�reasonrH rH rI �_perform_disable� s& �� rU c K � t |d| jd� dS )zGPerform the config action. :return: 0 on success, 1 otherwise r �r> r? r �rE r� r rH rH rI � action_config� � �rY c K s� | j r9| j tjvr(d�| j d�tj��}|�d�d }t�tj |t d| d���t dj| j t || j d�d �� d S t td d � tjD ��d �}d| d }tjD ]}t |j|t ||d�d �� qP|jsf|jrr|jsl|jrtt d� dS dS dS )z�Perform the 'config show' action optionally limit output to a single key :return: 0 on success :raise UserFacingError: on invalid keys z '{}' must be one of: {}r� �:rJ � )rg rh z {key} {value}N)r� �valuer c S s g | ]}t |��qS rH )�len)�.0�xrH rH rI � <listcomp>� s z&action_config_show.<locals>.<listcomp>� z{key: <z } {value}zo Error: Setting global apt proxy and pro scoped apt proxy at the same time is unsupported. No apt proxy is set.)r� r r� rq rW �findr r� rr rs r( r r� r� �max�global_apt_http_proxy�global_apt_https_proxy�ua_apt_http_proxy�ua_apt_https_proxy)r� rc r� rZ �indent_position� col_width�row_tmplr� rH rH rI r� � s@ ����� ����r� c K s ddl m} ddlm} t|d�}|�� d jd }|�� d jd }z | j�d�\}} W n t yA |� � t � d� | j���w |tjvrW|� � t � d � d �tj����|d v r�|�d �d } | d krjtj} ntj} t�| | | � || i}|d i |�� tj�|�} | �� \} }| tjkr�|d i |�� n�||jv r�|�d �d } | d kr�tj } ntj } t�| | | � t |j p�|j �}|r�t!t"j#j ddd�� t$|t%j&|| � d|_ d|_ n�||j'|j( v �r4d|v r�dnd } | d kr�tj } ntj } ||j'v �rt!t"j)j | d�� d| }t�| | | � t |j*�p|j+�}|�r%t!t"j#j ddd�� t$|t%j,|| � d|_*d|_+nJ|dv �rezt-| �} | dk �rJt d� |���W n2 t �yd |� � t!d� t � d� || ���w |dk�r~| �.� dk} | �ryt/�0|� nt1j2�3� t4||| � dS ) zMPerform the 'config set' action. @return: 0 on success, 1 otherwise r )�configure_livepatch_proxy)�configure_snap_proxyr� r r� �=z% Expected <key>=<value> but found: {}� <key> must be one of: {}r� �� http_proxy� https_proxy�_�httprJ zpro scoped aptz global apt)� current_proxy�previous_proxyN�https�� protocol_type�global_)�update_messaging_timer�metering_timerzInvalid interval for {}re zECannot set {} to {}: <value> for interval must be a positive integer.r �truerH )5�uaclient.livepatchrl � uaclient.snaprm r@ rA r� r� �split� ValueErrorr[ r r� rq r r� rW r � PROXY_VALIDATION_SNAP_HTTP_URL� PROXY_VALIDATION_SNAP_HTTPS_URL�validate_proxyr � livepatch�LivepatchEntitlement�application_statusr, �ENABLED�ua_scoped_proxy_options� PROXY_VALIDATION_APT_HTTP_URL� PROXY_VALIDATION_APT_HTTPS_URL�boolre rf r r �WARNING_APT_PROXY_OVERWRITE�configure_apt_proxyr$ �UACLIENT�&deprecated_global_scoped_proxy_options�global_scoped_proxy_options�WARNING_APT_PROXY_SETUPrg rh �GLOBAL�int�lowerr �update_apt_newsr1 �apt_news_contents_file�delete�setattr)r� rc r� rl rm r� r� rC �set_key� set_valuery � validate_urlrQ �livepatch_statusrs � unset_currentrH rH rI r� � s� ��� �� � ����� ���� ��� r� c K sd ddl m} ddlm} ddlm} | jtjvr?t |d�}|� � d j d }|� � d j d }|� � t �d�d �tj����| jd v rh| j�d �d } || d � tj�|�} | �� \} } | tjkrg|| d � nA| j|jv rxt||j| jd � n1| j|j|j v r�| j|jv r�d| jv r�dnd} t � t j j| d �� d| j | _t||j!| jd � t"|| jd � dS )zOPerform the 'config unset' action. @return: 0 on success, 1 otherwise r )r$ )�unconfigure_livepatch_proxy)�unconfigure_snap_proxyr� r r� ro r� rp rs rx Nrw rt rz )#� uaclient.aptr$ r~ r� r r� r� r r� r@ rA r� r[ r r� rq rW r� r r� r� r� r, r� r� r� r� r� r� rK rN r r� r� r� ) r� rc r� r$ r� r� r� r� rC ry rQ r� rs rH rH rI r� { sN �� � �� �� r� c C s� t ||d�\}}|r |r tj}|jd�|�| d�|�dd�}|S |r.tjjd�|�d�}|S tjj| d�|�d�tj�d�}|S )z�Generates a custom message for enable/disable commands when unattached. Takes into consideration if the services exist or not, and notify the user accordingly.)�namesrc r� re )� valid_service� operation�invalid_service� service_msg)r� zSee {}�r� r� r� ) r+ r �!MIXED_SERVICES_FAILURE_UNATTACHEDrq rW � VALID_SERVICE_FAILURE_UNATTACHED�INVALID_SERVICE_OP_FAILUREr r� )r� r� rc �entitlements_found�entitlements_not_foundrZ rH rH rI �%_create_enable_disable_unattached_msg� s. ���� � �r� z pro disablec K s� t | dg �}t||�\}}d}|D ]}tj||d�}||| jd�} |t| || jd�M }q|rSdd�tj|dd�� d } d �tj | d d d d ��} t j d d�|�| d��t � � |r[dS dS )zbPerform the disable action on a list of entitlements. @return: 0 on success, 1 otherwise r� Trb �r� zTry r� �rc � allow_beta�.r< �P F)rg ri rj r4 r� r rb )r� r+ r rm r� rU rW r* rr �wrapr � InvalidServiceToDisableErrorrK �process_events) r� rc r� r� r� r� rS �ent_namery �ent� valid_namesr� rH rH rI �action_disable� s> ������� r� z pro enablec K s� t | dd�}| j}|r|rtjddd��t�tj� zt� |� W n tj tj fy= t j tjdd� tjtjd� Y nw t | d g �}t||�\}}d}|D ]�} z_tj|| | j| j||d �\} } tj|d � | s�| d ur�t| t�r�| jd ur�t�| jj� tj| jj| jj | d � | j t j kr�|� | � n| r�tj!| d� n | s�| d u r�tj"| d� || M }W qO tj y� } zt�| j� tj| j| j#| d � d}W Y d } ~ qOd } ~ ww |r�t$||| jd�} t�%|� tj | j| j d��t�&� |r�dS dS )z^Perform the enable action on a named entitlement. @return: 0 on success, 1 otherwise rR re r0 r1 )�option1�option2T)�exc_info)� warning_msgr� )r� r� � access_onlyrR r� NrG )r� Fr� r� r rb )'r� r� r �InvalidOptionCombinationrK rN r �REFRESH_CONTRACT_ENABLEr �request_updated_contract�UrlErrorr� �logging�debug�REFRESH_CONTRACT_FAILURE�warningr+ r �enable_entitlement_by_namer� �betarP r rM r. rP rZ rQ ra rT r/ �IS_BETAru rO rL r� r) �services_failedr� )r� rc r� rR r� r� r� r� rS r� �ent_retrT �erZ rH rH rI � action_enable� s� � � � � �� � � ���� r� z pro detachr` c C s2 t || jd�}|dkrt�� t�� t�� |S )zXPerform the detach action for this machine. @return: 0 on success, 1 otherwise r� r )�_detachr� r �startr �stoprK r� )r� rc rS rH rH rI � action_detachG s r� r� c C s� g }t | �D ]+}z tj| |d�}W n tjy Y qw || |d�}|jdd�\}}|r1|�|� q|rTt|�dkr<dnd}t� d� |�� |D ] }t� d � |j �� qHt j |d �s\dS |D ] }t|| |d d � q^| �� | j�� t| � t� tj� d S )a7 Detach the machine from the active Ubuntu Pro 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 rb )rc r� T)�ignore_dependent_servicesrb �sre z,Detach will disable the following service{}:z {}r� F)r� rF r )r* r rm r rn � can_disableru r^ rK rN rq ra r �prompt_for_confirmationrU � delete_cacher� r� r5 r �DETACH_SUCCESS) rc r� � to_disabler� ry r� rS rs �suffixrH rH rI r� X s8 � �� r� c C s� d }| j r| j �di ��di ��d�}|r t�tjj|d�� nt�tj� t� � t� | � t � | �\}}t �|�}t�t�|�� t�� d S )N�machineTokenInfo� contractInfora )� contract_name)� machine_tokenro rK rN r �ATTACH_SUCCESS_TMPLrq � ATTACH_SUCCESS_NO_CONTRACT_NAMEr r� �cleanupr r rP �format_tabularr �handle_unicode_charactersr� )rc r� r �_ret�outputrH rH rI �_post_cli_attach� s" � � r� c C s. t | j| j|�}t|�� � |jdkrdS dS )N�successr rb )r r� r� r �to_json�result�r� rc r� rH rH rI � action_api� s r� c C sJ z t t� |tjjd� W n tjy t�t j j � Y dS w t |� dS )N)rc �moderb r ) r r r �EventLoggerMode�CLIr r� rK rN r �ATTACH_FAILURErZ r� r� rH rH rI �action_auto_attach� s ��r� c K s� | j dkr tjd| j d��t�tj� t|d�}t�dtjj |j d� � t |j d�}zt ||d�}W n! tj yT } zt�tj� t|j d�}t||d� |�d }~ww t�dtj � |jS ) Nr: r� )�paramr] r� r< )� user_code)� magic_token)r� rc )rq r �MagicAttachInvalidParamrK rN r �CLI_MAGIC_ATTACH_INITr �CLI_MAGIC_ATTACH_SIGN_INr� r r� r! �MagicAttachTokenError�CLI_MAGIC_ATTACH_FAILEDr r �CLI_MAGIC_ATTACH_PROCESSING�contract_token)r� rc r� � initiate_resp� wait_options� wait_respr� �revoke_optionsrH rH rI � _magic_attach� s8 � ��� � �� r� z pro attachc C s� | j r| jrtjtjjtjjd��| j s | js t| |d�}d }n0| j r)| j }d }n'z t � t | j��}W n t yI } z tj | jj|jd��d }~ww |j }|j}| joV|d u }z tj|||d� W n tjym t�� �w d}|d ur�| jr�t||�\}} |D ]8} tj|| ddd�\} } | s�d}| d ur�t| t�r�| jd ur�t�| jj� tj| jj| jj| d � q�t� | � q�| r�t | |dd �} tj| jt j d � tj| j| jd � d}t |� |S ) Nr� r� )� config_namerQ )r� � allow_enabler T)r� r� rb rG r� )� file_type�rH rI )!r� � attach_configr r� r �ATTACH_TOKEN_ARG_XOR_CONFIGrZ ra r� r& � from_dictr7 r' � AttachInvalidConfigFileError�enable_servicesr� r �attach_with_tokenr� � AttachErrorr+ r� rM r. rP rK rN rQ rO r) rL rM r� )r� rc r� �enable_services_overrider r� r rS �found� not_foundra r� rT rZ rH rH rI � action_attach� sz � � ����� �� �� �r c C s� | j pt}t�� �M}t�||� z t�|d��}|j|dd� W d � n1 s)w Y W n t yM } zt � |� W Y d }~W d � dS d }~ww W d � dS 1 sYw Y dS )Nzw:gzzlogs/)�arcnamerb r ) r� �UA_COLLECT_LOGS_FILE�tempfile�TemporaryDirectoryr � collect_logs�tarfile�open�add�PermissionErrorr� rQ )r� rc � output_file� output_dir�resultsr� rH rH rI �action_collect_logs s& �� ��� ��r c C s� t }tttjtjtdd�tjtdd�|d�}|jdddd� |jdd t � � d �t�d � d |j _ |j d ddd�}d|_|jddd�}t|� |jtd� |jddd�}t|� |jtd� |jddd�}t|� |jtd� |jddd�}t|� |jtd� |jd d d�}t|� |jtd� |jd d d�} t| � | jt d� |jd d!d�} t | | d"� | jt d� |jd#d$d�} t | | d"� | jt d� |jd%d&d�} | jt!d� t"| � |jd'd(d�} t#| � | jt$d� |jd)d*d�}t%|| d"� |jt&d� |jd+d,d�}|jt'd� t(|� |jd-d.d�}|jt)d� t*|� |jd d �t�d�}|jt+d� |jd/d0d�}|jt,d� t-|� |S )1Nz <command>r� )r? rB r@ rA r> z--debugr� z&show all debug log messages to consoler� z --versionr zshow version of {})r� r r� r� r� r� re r� Tr� z1attach this machine to an Ubuntu Pro subscriptionr� r� r� r� r� z+automatically attach on supported platformsr� z&collect Pro logs and debug informationr z/manage Ubuntu Pro configuration on this machiner! z3remove this machine from an Ubuntu Pro subscriptionr4 z5disable a specific Ubuntu Pro service on this machiner� r. z4enable a specific Ubuntu Pro service on this machiner� z=check for and mitigate the impact of a CVE/USN on this systemr� z.list available security updates for the systemr� z3show detailed information about Ubuntu Pro servicesr zrefresh Ubuntu Pro servicesr z)current status of all Ubuntu Pro servicesr6 z/show system information related to Pro services).�__doc__r= r� r r� r� rq � EPILOG_TMPLr� r � get_versionr� r� r� �requiredr� r� r� r r� r� r� r� r� r r� rY r$ r� r5 r� r3 r� r r� r r r- � action_help�action_refreshr � action_statusr= � print_version� action_systemr; )rc r> r� r� � parser_attach� parser_api�parser_auto_attach�parser_collect_logs� parser_config� parser_detach�parser_disable� parser_enable� parser_fix�parser_security_status� parser_help�parser_refresh� parser_status�parser_version� parser_systemrH rH rI r@ ) s� ����� � � � � � � � � � � � � � � r@ c C s� |st �� }| r | jnd}| r| jnd }tjjj}tj |||d�\}}t |d |k�}| rY| j rY|rY|d |krTt j ddd� t�d� tj |||d�\}}|d |ks8t � d� t �|� tj||d�}t � t�|�� t �� |S ) NF)�simulate_with_tokenr\ �execution_statusr� re )�endrb �r\ )r rk �allr3 rP �UserFacingConfigStatus�ACTIVEr] r r r� �waitrK rN �time�sleep�set_output_contentr� r r� r� ) r� rc r\ r� � active_valuer rS � config_activer� rH rH rI r! � s2 � � � r! c K rV )zGPerform the system action. :return: 0 on success, 1 otherwise r6 rW r rX r rH rH rI r# � rZ r# c C s t |�}t�|j� dS )Nr )r" rK rN �reboot_requiredr� rH rH rI r8 � s r8 c C s t t�� � d S r� )r r r )�_argsrc rH rH rI r" � s r" c C sr z|� � W n+ ty1 } z t�� � t�|� W d � n1 s"w Y t�tj ��d }~ww t tj � d S r� ) �process_config� RuntimeErrorr �disable_log_to_consoler� r� r r� r �REFRESH_CONFIG_FAILUREr �REFRESH_CONFIG_SUCCESS)r� rc �excrH rH rI �_action_refresh_config� s � ��rH c C sv zt �|� W n, tjy3 } z t�� � t�|� W d � n1 s$w Y t�t j ��d }~ww t t j � d S r� ) r r� r r� r rD r� r� r� r r� r �REFRESH_CONTRACT_SUCCESS�rA rc rG rH rH rI �_action_refresh_contract� s � ��rK c C s� zt |� t� |jrt�|� W n+ ty< } z t�� � t�|� W d � n1 s-w Y t � t j ��d }~ww t t j� d S r� )r5 r4 r r� � Exceptionr rD r� r� r r� r �REFRESH_MESSAGES_FAILUREr �REFRESH_MESSAGES_SUCCESSrJ rH rH rI �_action_refresh_messages� s � � ��rO z pro refreshc C sj | j d u s | j dkrt| |� | j d u s| j dkr$t| |� t�tj� | j d u s.| j dkr3t| |� dS )Nr r r r )r rH rK r0 �remover2 �CONTRACT_REFRESH_WARNINGrO r� rH rH rI r � s r �scoper� r� c C sR |t jkr | j}| j}n |t jkr| j}| j}d|v r |}n|}t|||d� dS )zS Handles setting part the apt proxies - global and uaclient scoped proxies rw )rq rr � proxy_scopeN)r$ r� re rf r� rg rh r% )rc rR r� r� rq rr rH rH rI r� s �r� c C s� | j }| j}|st|d�j|d� dS |st�� }t�||�}| jdkr-t t � |�� dS |� � D ]\}}t d�|� � |�� q1dS )Nr� r6 r r: z{}: {} )r� r7 r@ r[ r rk rP r� rq r r: r �itemsr� )r� rc r� r\ � help_responser� r] rH rH rI r & s �r c C sH | r| j dkst| dd�dv rd S t�� }|r"t�tj|d�� d S d S )Nr� rq re �r: r; )r )r� r� r �check_for_new_versionr� r� �NEW_VERSION_NOTICErq )r� � new_versionrH rH rI �_warn_about_new_version< s � �rZ c C sN | j dv r#tj�� s%t| d�r| jdv rdS t�tj j| j d�� dS dS dS )z4Warn users that the user readable output may change.)r r� rq rV N)r� ) r� rL �stdout�isatty�hasattrrq r� r� r � WARNING_HUMAN_READABLE_OUTPUT�r� rH rH rI � _warn_about_output_redirectionL s ����r` c C s |du r t �� }|j}t�� st�� }t|t�r |� � }t� � }|du r(t � � }|� |� |�t�� � g |_t �tj�}|�|� |� | � |�d� |�|� t�|�}|�� sh|jjddd� |j dd� t � |�}|�t � � |� |� |�d� |�|� dS )z�Setup console logging and debug logging to log_file If run as non_root and cfg.log_file is provided, it is replaced with another non-root log file. Nz ua-consoleT)�parents�exist_oki� )r� zua-file) r rk �log_filer r� �pro_log�get_user_log_filerM r� �upper� LogFormatterr� � getLogger�setLevel� addFilter�RedactionFilter�handlers� StreamHandlerrL rM � setFormatter�set_name� addHandler�pathlib�Path�exists�parent�mkdir�touch� FileHandlerr3 ) � console_level� log_levelrc �loggerrc �console_formatter�console_handler� log_file_path� file_handlerrH rH rI � setup_logging[ s6 r c C s` | j dv r*t�| j � t| d�r,| jdkr t�tjj� | jdkr.t�tjj � dS dS dS dS )z3Set the right event mode based on the args provided)r� r! r. r4 r rq r: r; N) r� rK � set_commandr] rq �set_event_moder r� �JSON�YAMLr_ rH rH rI r� � s �r� c r� )Nc s� z� | i |��W S t y; t�� � t�d� W d � n1 s#w Y tdtjd� t� � t� d� Y d S t j y� } z�dt |�v rntj}t�d�rTtj}|j|jd�}tj|j|jd� tj|jtjd � nDt�� �# |j|d �}|jr�tj}ntj}t�|jdi |��� W d � n1 s�w Y tj }tj|j|jd� tj|jtjd � t� � t� � t � t� d� W Y d }~d S d }~w t j �y* } zPt�� � t�|j� W d � n1 s�w Y tj|j|j |j!d � tjd �|j�tjd � t"|t j#��st� � t� � t � t� |j$� W Y d }~d S d }~w t%�y~ } zHt�� � t�d � W d � n 1 �sGw Y t� � tjtj&jtjd � tjt'|dt |��dd� t� � t � t� d� W Y d }~d S d }~ww )N�KeyboardInterruptz Interrupt received; exiting.rS rb �CERTIFICATE_VERIFY_FAILEDzca-certificates)rf r )�info_msgr )rf rQ )rH rI �additional_infoz{}z&Unhandled exception, please file a bugrZ r� )rH � error_typerH )(r� r rD r� rQ r rL rM r �clear_lock_file_if_presentrO r r� r� r �&SSL_VERIFICATION_ERROR_CA_CERTIFICATESr � is_installed�%SSL_VERIFICATION_ERROR_OPENSSL_CONFIGrq rf rK rZ ra rN �$LOG_CONNECTIVITY_ERROR_WITH_URL_TMPL�LOG_CONNECTIVITY_ERROR_TMPLr� �CONNECTIVITY_ERRORr� rZ r� r� r� rM � LockHeldErrorr rL �UNEXPECTED_ERRORr� )r� r� rG �tmplrZ �msg_args�msg_tmplr� ��funcrH rI r� � s� � � � � ��� � ����z#main_error_handler.<locals>.wrapperrH )r� r� rH r� rI �main_error_handler� s Hr� c C s@ t tjtjd tjd � | stj} t�� }t |d�}| dd � }|s0|� � t t � t� d� |j|d�}t|� |j}|j}tj||d� |j}|jrPtjntj}t |||j� t�d| � t�� � |�� W d � n1 stw Y dd � tt�� � � �D �} | r�t�d | � t |� |j ||d�} t |� | S ) Nry rc r� rb )r� rp zExecuted with sys.argv: %rc S s g | ] \}}d � ||��qS )z{}={})rq )r_ �k�vrH rH rI ra � s ��zmain.<locals>.<listcomp>z'Executed with environment variables: %r) r r� �INFOr �CONFIG_DEFAULTSrL �argvr rk r@ rK r rN rO � parse_argsr� rq rr r �configure_web_proxyry r� �DEBUGrc rD �warn_about_invalid_keysrV �get_pro_environmentrT r` r� rZ ) �sys_argvrc r� � cli_argumentsr� rq rr ry rx �pro_environment� return_valuerH rH rI �main� sJ � ���r� �__main__r� )NN)r` N)�r r r: r� rq r rL r r rr r; � functoolsr �typingr r r �uaclientr r r r r r r r r r r r rd r r r r rP r r r �uaclient.api.apir �2uaclient.api.u.pro.attach.auto.full_auto_attach.v1r r �+uaclient.api.u.pro.attach.magic.initiate.v1r �)uaclient.api.u.pro.attach.magic.revoke.v1r r �'uaclient.api.u.pro.attach.magic.wait.v1r r! �5uaclient.api.u.pro.security.status.reboot_required.v1r" �(uaclient.api.u.pro.status.is_attached.v1r# r� r$ r% �uaclient.data_typesr& r' �uaclient.defaultsr( �uaclient.entitlementsr) r* r+ �(uaclient.entitlements.entitlement_statusr, r- r. r/ �uaclient.filesr0 r1 �uaclient.files.noticesr2 � uaclient.logr3 � uaclient.timer.update_messagingr4 r5 � uaclient.yamlr6 r7 r� r� r rN �STATUS_HEADER_TMPLr� r+ r � BLUE_INFOrX �get_event_loggerrK �ArgumentParserr= r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r$ rk r- r3 r5 r; r9 r= rE rU rY r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r@ r! r# r8 r" rH rK rO r r� r rZ r` r r� r� r� r| rO rH rH rH rI �<module> s0 4 ���� ^ 2'9# '4$O�� � & ) } / )N ,!D } ���� � - L 1�
| ver. 1.4 |
Github
|
.
| PHP 8.2.9 | Generation time: 0.01 |
proxy
|
phpinfo
|
Settings