/usr/lib/python3/dist-packages/uaclient/__pycache__
Edit: /usr/lib/python3/dist-packages/uaclient/__pycache__/actions.cpython-310.pyc (11381B)
o
SKjS; @ sT d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlmZm Z d dl
m Z m Z m
Z
mZmZmZmZmZ d dl
mZ d dl
mZmZ d dl
mZ d dl
mZmZmZ d dl m Z m Z m Z m Z d d l!m"Z"m#Z# d d
l$m%Z%m&Z&m'Z'm(Z(m)Z) e* Z+e,e-e.Z/d Z0d Z1d
e
j2dej3de j fddZ4 d8d
e
j2deej5 dej3de j de6f
ddZ7 d8d
e
j2de8de6de6ddf
ddZ9 d9d
e
j2d e j:ddfd d Z;dd ddd!d
e
j2d"e8d#e6d$e8de6d%e ee8 f d&d'Z
ddfd1d2Z?d
e
j2fd3d4Z@d
e
j2d5e8fd6d7ZAdS ); N)ListOptional)apicloudsconfigcontract entitlements event_logger
exceptions livepatch)log)messagessecret_manager)status)systemtimerutil)APPARMOR_PROFILESCLOUD_BUILD_INFODEFAULT_CONFIG_FILEDEFAULT_LOG_PREFIX)
machine_tokennotices)AttachmentDataattachment_data_filemachine_id_file only_series_check_marker_filetimer_jobs_state_file)zapt-news.servicezesm-cache.servicezua-timer.servicezua-timer.timerzua-auto-attach.pathzua-auto-attach.servicezua-reboot-cmds.servicezubuntu-advantage.service
cfgcontract_client attached_atc C s< ddl m} tt|d tj| d || | d S )Nr update_motd_messagesr! r ) uaclient.timer.update_messagingr# r writer ua_statusr update_activity_token)r r r! r# r* 2/usr/lib/python3/dist-packages/uaclient/actions.py_handle_partial_attach9 s
r, Fservices_to_be_enabledsilentc
C sD d}g }g }z'|D ]"}t | |j|j|d\} }
|| M }| s$||j q tj|jd q W nJ tjyH } zt|j t | || | d } ~ w tj
yY ||j d}Y n t yw } zd}||j || W Y d } ~ nd } ~ ww |st | || t | |rtj
dd t||D dtjdd |D dd S ) NT)r namevariantr. )serviceFc S s, g | ]\}}|t jjt|t d fqS )) error_msglog_path)r
UNEXPECTED_ERRORformatstrpro_log get_user_or_root_log_file_path).0r/ exceptionr* r* r+
r s z,_enable_default_services..)failed_servicesc S s g | ]}|t jfqS r* )r
!E_ATTACH_FAILURE_DEFAULT_SERVICES)r9 r/ r* r* r+ r; s )enable_entitlement_by_namer/ r0 appendeventservice_processedr
ConnectivityErrorservice_failedr, UbuntuProError Exceptionservices_failedAttachFailureUnknownErrorzip AttachFailureDefaultServices)
r r- r r! r. retr< unexpected_errorsenable_by_default_serviceent_retreasonexcer* r* r+ _enable_default_servicesF s^
rQ token allow_enablereturnc
C s ddl m} ddlm} tj| t| }t
| }t j j t j
jd}|j||d} t j}
| di di } tdd | d
g D } | d i d i d
d}
|
rzt|
}W n
tjyi Y n&w t|
}|j|jkr~tj|j|jdt j t j j |j|jd t !|
|!| z|| W n tj"y } z|# |d}~ww tj$% | di dt$| }t&!| |rt '| |( }t)| ||||d t*!t+|d || t,- dS )aC
Common functionality to take a token and attach via contract backend
:raise ConnectivityError: 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 )+check_entitlement_apt_directives_are_uniquer" )tz)contract_token
attachment_dtmachineTokenInfo contractInfoc s s, | ]}| d dkr| d |fV qdS )typesupportN)get)r9 rP r* r* r+ s z$attach_with_token..resourceEntitlementsr\ affordances
onlySeriesN)releaseseries_codename machineId)r r- r r! r. r$ ).uaclient.entitlementsrU r& r# r secrets
add_secretr get_machine_token_filer UAContractClientdatetimenowtimezoneutcadd_contract_machiner get_release_infoseriesr] dictget_distro_infor
MissingSeriesInDistroInfoFileeol AttachFailureRestrictedReleaserb rc r addNoticeLIMITED_TO_RELEASEr r' %EntitlementsAPTDirectivesAreNotUniquedeleteget_machine_id cache_clearr get_enabled_by_default_servicesr rQ r r r start)r rR rS r. rU r# machine_token_filer r! new_machine_tokencurrent_seriesrZ support_resource only_seriesallowed_releasecurrent_releaserP
machine_idr- r* r* r+ attach_with_token s
r Tcloudc C s | | }t| ||d dS )a\
:raise ConnectivityError: 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.
)rR rS N)acquire_pro_tokenr )r r rS rR r* r* r+ auto_attach s
r ) access_onlyr0 r.
extra_argsr/ r r0 r c C sf t j| ||||d}|sttjj|jd |t
\}}|r/|s/ttj j|jd ||fS )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.
)r r/ r0 r r )title) r entitlement_factoryr@ infor
ENABLING_TMPLr5 r enabler ProgressWrapper ENABLED_TMPL) r r/ r r0 r. r entitlementrM rN r* r* r+ r> s r> )simulate_with_tokenshow_allr r c C s: |rt j| ||d\}}||fS t j| |d}d}||fS )z6
Construct the current Pro status dictionary.
)r rR r r r r )r( simulate_statusr )r r r r rJ r* r* r+ r s
r filenamec
C s g d}d}d}z t |\}}W n' tjy8 } ztdt| t d| t| W Y d}~dS d}~ww |r[g }| dD ]
}t
||rO| | qBt | d
| dS dS )z
Helper which gets ubuntu_pro apparmor logs from the kernel from the last
day and writes them to the specified filename.
)
journalctlz-bz-kz--since=1 day agoz7apparmor=\".*(profile=\"ubuntu_pro_|name=\"ubuntu_pro_)Nz!Failed to collect kernel logs:
%s{}-error
)r subpr
ProcessExecutionErrorLOGwarningr6
write_filer5 splitresearchr? join)r cmd apparmor_re kernel_logs_rP
apparmor_logs kernel_liner* r* r+ _write_apparmor_logs_to_file0 s$ $
r return_codesc
C sr z
t j| |d\}}W n tjy, } zt d|t| W Y d}~dS d}~ww t |d|| dS )zCHelper which runs a command and writes output or error to filename.)rcsr Nzstdout:
{}
stderr:
{})r r r r
r r r5 r6 )r r r outerrrP r* r* r+ _write_command_output_to_fileI s $ r c s, j pt jtjjtg fddtjD S )Nc 3 s( | ]}t |tjjr| jV qd S N)
issubclassr repoRepoEntitlement repo_file)r9 entitlement_clsr% r* r+ r^ ^ s
z#_get_state_files..) cfg_pathr log_filer ua_filepathr r ENTITLEMENT_CLASSESr% r* r% r+ _get_state_filesW s
r
output_dirc
C s t dd| t dtjd| t dd| t dd| t d d
d d tD d
| tD ]}t d|d||ddgd q9t| dd\}}td|t j
|t j d t
}td|t
| t| }t rt dt nt g}t|D ]D\}} tj| rtd| qzt t| }
ttj|d||
W q t y } ztd| t | W Y d} ~ qd} ~ ww |t t d D ]L} tj | r&zt| }
W n t y } ztd | t | W Y d} ~ qd} ~ ww t |
}
t rt| |
ttj|tj!| |
qt"d | t#D ]1} tj | r`z t$%| | W q0 t y_ } ztd | t | W Y d} ~ q0d} ~ ww q0dS ) zG
Write all relevant Ubuntu Pro logs to the specified directory
zcloud-idz{}/cloud-id.txtz {} statusz{}/livepatch-status.txtzsystemctl list-timers --allz{}/systemd-timers.txtzujournalctl --boot=0 -o short-precise -u cloud-init-local.service -u cloud-init-config.service -u cloud-config.servicez{}/cloud-init-journal.txtz journalctl -o short-precise {} c S s g | ] }d |v rd |qS )z.servicez-u {})r5 )r9 sr* r* r+ r; s z collect_logs..z{}/pro-journal.txtzsystemctl status {}z {}/{}.txtr )r Fr z{}/pro-status.json)clsz{}/environment_vars.jsonNz$Skipping symlinked user log file: %sz
user{}.logz&Failed to collect user log file: %s
%s*zFailed to load file: %s
%sz{}/apparmor_logs.txtzFailed to copy file: %s
%s)&r r5 r
LIVEPATCH_CMDr UA_SERVICESr r r jsondumpsr DatetimeAwareJSONEncoderget_pro_environmentr we_are_currently_rootr7 get_all_user_log_filesUSER_LOG_COLLECTED_LIMITget_user_log_file enumerateosr islinkr r redact_sensitive_logs load_filerE r6 globr isfilebasenamer r shutilcopy)
r r r1
pro_statusr env_vars state_filesuser_log_files log_file_idxr contentrP fr* r* r+ collect_logsf s
r )F)Tr )Brj r r loggingr r r typingr r uaclientr r r r r r r
r r r7 r
r r r( r r r uaclient.defaultsr r r r uaclient.filesr r uaclient.files.state_filesr r r r r get_event_loggerr@ getLogger replace_top_level_logger_name__name__r r r UAConfigri r, EnableByDefaultServiceboolrQ r6 r AutoAttachInstancer r> r intr r r r* r* r* r+ s (
F
b
%