/usr/lib/python3/dist-packages/uaclient/__pycache__
Edit: /usr/lib/python3/dist-packages/uaclient/__pycache__/util.cpython-310.pyc (13525B)
o
Ki= @ s\ d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlm Z d dl
m Z m Z m
Z
mZmZ d dlmZmZ d dlmZ d dlmZ e Zdedefd d
Zeee Z G d d d ej Z G d
d dej Z!dd Z" dJde ee f de ee f dede ee f fddZ#dg fdede
e defddZ$ dKded e%d e%de%fd d Z&d e ee f d!ede%fd"d#Z'g d$Z(e(fd%e
e defd&d'Z)d(ee d e%de%fd)d*Z*d+ede j fd,d-Z+d.edefd/d0Z,d1d2 Z-d3d4 Z.d5d6d7d8d9d:Z/d;edefd
e
e de
e fd?d@Z1de%fdAdBZ2dCedDedefdEdFZ3dGe
e fdHdIZ4dS )L Nwraps)AnyDictListOptionalUnion)
exceptionsmessages) CONFIG_FIELD_ENVVAR_ALLOWLIST)MessagingOperationsnamereturnc C s( | dkrdS | d}d|d< d|S )z1Replace the name of the root logger from __name__ . ubuntupror )splitjoin)r
names r //usr/lib/python3/dist-packages/uaclient/util.py replace_top_level_logger_name s
r c s e Zd ZdZ fddZ ZS )DatetimeAwareJSONEncoderzBA json.JSONEncoder subclass that writes out isoformat'd datetimes.c s t |tjr
| S t |S N)
isinstancedatetime isoformatsuperdefault)selfo __class__r r r " s z DatetimeAwareJSONEncoder.default)__name__
__module__ __qualname____doc__r
__classcell__r r r! r r s r c s, e Zd ZdZ fddZedd Z ZS )DatetimeAwareJSONDecodera,
A JSONDecoder that parses some ISO datetime strings to datetime objects.
Important note: the "some" is because we seem to only be able extend
Python's json library in a way that lets us convert string values within
JSON objects (e.g. '{"lastModified": "2019-07-25T14:35:51"}'). Strings
outside of JSON objects (e.g. '"2019-07-25T14:35:51"') will not be passed
through our decoder.
(N.B. This will override any object_hook specified using arguments to it,
or used in load or loads calls that specify this as the cls.)
c s. d|v r | d t j|d| ji| d S )N object_hook)popr __init__r) )r argskwargsr! r r r+ 6 s
z!DatetimeAwareJSONDecoder.__init__c C sL | D ] \}}t|tr#zt|}W n ty |}Y nw || |< q| S r )itemsr strparse_rfc3339_date
ValueError)r keyvalue new_valuer r r r) ; s
z$DatetimeAwareJSONDecoder.object_hook)r# r$ r% r& r+ staticmethodr) r' r r r! r r( ( s
r( c s fdd}|S )a Decorator to retry on exception for retry_sleeps.
@param retry_sleeps: List of sleep lengths to apply between
retries. Specifying a list of [0.5, 1] tells subp to retry twice
on failure; sleeping half a second before the first retry and 1 second
before the second retry.
@param exception: The exception class to catch and retry for the provided
retry_sleeps. Any other exception types will not be caught by the
decorator.
c s t fdd}|S )Nc
sp } z| i |W S y6 } z |s|tdt|t| t|d W Y d }~nd }~ww q)NTz%s: Retrying %d more times.r )copyLOGdebugr/ lentimesleepr* )r, r- sleepse) exceptionf retry_sleepsr r decoratorW s z)retry..wrapper..decoratorr )r? rA r> r@ )r? r wrapperV s
zretry..wrapperr )r> r@ rC r rB r retryJ s rD r orig_dictnew_dictpathc C s i }| D ]C\}}||t}|s|n|d | }t|tr8||v r3t||| |d}|r2|||< qt||< q||krItd|t| |||< q| D ] \}}|| vrZ|||< qN|S )z s z"prompt_choices..Tz> )uaclientrQ get_event_loggerformatr infoinputlower)rO rP rQ eventr3 error_msgr r r prompt_choices s
r^ F
assume_yesr c C s> |rdS | s t j} t| }|dkr|S |dv r dS dS )a
Display a confirmation prompt, returning a bool indicating the response
:param msg: String custom prompt text to emit from input call.
:param assume_yes: Boolean set True to skip confirmation input and return
True.
:param default: Boolean to return when user doesn't enter any text
This function will only prompt a single time, and defaults to "no" (i.e. it
returns False).
Tr )yyesF)r
PROMPT_YES_NOrZ r[ strip)rO r_ r r3 r r r prompt_for_confirmation s rd config
path_to_valuec C s | }i }| d}|d }|D ]}||krd}t|tr#|||}q dS t|}| dkr2dS | dkr:dS tj|d|d) aJ Check if value parameter can be translated into a boolean 'True' value.
@param config: A config dict representing
/etc/ubuntu-advantange/uaclient.conf
@param path_to_value: The path from where the value parameter was
extracted.
@return: A boolean value indicating if the value paramater corresponds
to a 'True' boolean value.
@raises exceptions.UbuntuProError when the value provide by the
path_to_value parameter can not be translated into either
a 'False' or 'True' boolean value.
r falseFtrueTz boolean string: true or false)rf expected_valuer3 )r r rJ rH r/ r[ r InvalidBooleanConfigValue)re rf r3
default_valuepaths
leaf_valuer2 value_strr r r is_config_value_true s(
rp )z(Bearer )[^\']+z(\'attach\', \')[^\']+z (\'machineToken\': \')[^\']+(\'token\': \')[^\']+z((\'X-aws-ec2-metadata-token\': \')[^\']+z*(.*\[PUT\] response.*api/token,.*data: ).*z(https://bearer:)[^\@]+z1(/snap/bin/canonical-livepatch\s+enable\s+)[^\s]+z>(Contract\s+value\s+for\s+'resourceToken'\s+changed\s+to\s+).*z (\'resourceToken\': \')[^\']+z (\'contractToken\': \')[^\']+zF(https://contracts.canonical.com/v1/resources/livepatch\?token=)[^\s]+z (\"identityToken\": \")[^\"]+zT(response:\s+http://metadata/computeMetadata/v1/instance/service-accounts.*data: ).*rq z(\'userCode\': \')[^\']+z(\'magic_token=)[^\']+z (--registration-key=\")[^\"]+z (--registration-key=\')[^\']+z(--registration-key=)[^ ]+z (--registration-key \")[^\"]+z (--registration-key \')[^\']+z(--registration-key )[^\s]+z
(-p \")[^\"]+z
(-p \')[^\']+z (-p )[^\s]+
redact_regexsc C s | }|D ] }t |d|}q|S )z4Redact known sensitive information from log content.z\g<1>)resub)logrr redacted_log redact_regexr r r redact_sensitive_logs s rx msg_opsc C sN | sdS | D ] }t |trt| q|\}}||d< |di |s$ dS qdS )ah Emit messages to the console for user interaction
:param msg_op: A list of strings or tuples. Any string items are printed.
Any tuples will contain a callable and a dict of args to pass to the
callable. Callables are expected to return True on success and
False upon failure.
:return: True upon success, False on failure.
Tr_ FNr )r r/ print)ry r_ msg_opfunctorr, r r r handle_message_operations s
r} dt_strc C sD t dd| }t dd|}|dd}t dd|}tj|d S )
aT
Parse a datestring in rfc3339 format. Originally written for compatibility
with golang's time.MarshalJSON function. Also handles output of pythons
isoformat datetime method.
This drops subseconds.
:param dt_str: a date string in rfc3339 format
:return: datetime.datetime object of time represented by dt_str
z(\d{2}:\d{2}:\d{2})\.\d+z\g<1>z(\d{2}:\d{2}:\d{2})$z\g<1>ZZz+00:00z(-|\+)(\d{2}):(\d{2})$z\g<1>\g<2>\g<3>z%Y-%m-%dT%H:%M:%S%z)rs rt replacer strptime)r~ dt_str_without_subseconds
dt_str_with_zdt_str_without_zdt_str_with_pythonish_tzr r r r0 & s r0 messagec C s` t jjdu sdt jj vr.| dd} | tjd d} | tjd d} | dd } | S ) z}
Verify if the system can output unicode characters and if not,
remove those characters from the message string.
NzUTF-8u —- r asciiignore)
sysstdoutencodingrR r r
OKGREEN_CHECKFAIL_Xencodedecode)r r r r handle_unicode_characters] s r c C s dd t j D S )Nc S s4 i | ]\}}| tv s|d s|dkr||qS ) UA_FEATURESUA_CONFIG_FILE)r[ r
startswith)rS kvr r r
x s z'get_pro_environment..)osenvironr. r r r r get_pro_environmentw s r c C s dd }| D ]D\}}| |}t|tr#t|tr#t| | | qt|trHt|trHt|rBt|d trB|| | ||d q || |< q|| |< qdS )a Merge the contents of overlay dict into base_dict not only on top-level
keys, but on all on the depths of the overlay_dict object. For example,
using these values as entries for the function:
base_dict = {"a": 1, "b": {"c": 2, "d": 3}}
overlay_dict = {"b": {"c": 10}}
Should update base_dict into:
{"a": 1, "b": {"c": 10, "d": 3}}
@param base_dict: The dict to be updated
@param overlay_dict: The dict with information to be added into base_dict
c
S sx dddd}g }| |}|D ]%}d}t| D ]\}} | || |kr,t| | d}q|s4|| q| | d S )Nr
typeselector)availableResourcesresourceEntitlements overridesFT)rH enumerate depth_first_merge_overlay_dictappendextend)
base_valuesoverlay_valuesr2 merge_id_key_mapvalues_to_appendid_key
overlay_value was_replacedbase_value_idx
base_valuer r r update_dict_list s"
z8depth_first_merge_overlay_dict..update_dict_listr )r2 N)r. rH r rJ r listr9 ) base_dict overlay_dictr r2 r3 r r r r r s
r amd64i386ppc64elarm64armhf)x86_64i686ppc64leaarch64armv7larchc C s | }t||S r )r[ ARCH_ALIASESrH )r
arch_lowerr r r standardize_arch_name s r archesc C s* t }| D ] }|t| qtt|S r )setaddr sortedr )r deduplicated_archesr r r r deduplicate_arches s r c C s t dkS )Nr )r getuidr r r r we_are_currently_root s r filename
new_extensionc C s t j| \}}|d | S )Nr )r rG splitext)r r r
_extensionr r r set_filename_extension s r package_listc
C s& d tjd | ddddddd S )N
r P Fz )widthbreak_long_wordsbreak_on_hyphensinitial_indentsubsequent_indent)r textwrapwrap)r r r r create_package_list_str s
r )r )r FF)5r jsonloggingr rs r r r: functoolsr typingr r r r r rV r r
uaclient.defaultsr uaclient.typesr objectrI r/ r getLoggerr# r7 JSONEncoderr JSONDecoderr( rD rK r^ boolrd rp REDACT_SENSITIVE_LOGSrx r} r0 r r r r r r r r r r r r r s "!
'
7
5