Server : Apache/2.4.52 (Ubuntu) System : Linux webserver 6.8.0-49-generic #49~22.04.1-Ubuntu SMP PREEMPT_DYNAMIC Wed Nov 6 17:42:15 UTC 2 x86_64 User : www-data ( 33) PHP Version : 8.1.2-1ubuntu2.21 Disable Function : NONE Directory : /lib/python3/dist-packages/uaclient/__pycache__/ |
o
���frE � @ s
d dl Z d dlZd dlZd dlmZmZ d dlmZmZm Z d dl
m Z m Z m
Z
mZmZmZmZmZ d dlmZmZmZmZmZmZmZmZ d dl m Z d dl m Z e� e�!e"��Z#dd d
�Z$d Z%d Z&d
Z'e �(� Z)G dd� d�Z*de+fdd�Z,ddd�Z-de+fdd�Z.dS )� N)� lru_cache�wraps)�Any�Dict�Optional)�apt� event_logger�
exceptions�http�messages�snap�system�util)� APT_NEWS_URL�BASE_CONTRACT_URL�BASE_LIVEPATCH_URL�BASE_SECURITY_URL�CONFIG_DEFAULTS� CONFIG_FIELD_ENVVAR_ALLOWLIST�DEFAULT_CONFIG_FILE�DEFAULT_DATA_DIR)�user_config_file)� safe_load�name�type)�availableResources�resourceEntitlements�_unset) �
http_proxy� https_proxy�apt_http_proxy�apt_https_proxy�ua_apt_http_proxy�ua_apt_https_proxy�global_apt_http_proxy�global_apt_https_proxy�update_messaging_timer�metering_timer�apt_news� apt_news_url) � contract_url�data_dir�features�log_file� log_level� security_url�settings_overrides� ua_config�
livepatch_urlc @ s� e Zd ZdZdZdZ dCdeeee f dee
j dee ddfd d
�Z e
defd d ��Ze
defd
d��Ze
defdd��Ze
dee fdd��Zejdefdd��Ze
dee fdd��Zejdefdd��Ze
dee fdd��Zejdefdd��Ze
dee fdd ��Zejdefd d ��Ze
edd �dee fd d ���Zejdefd!d ��Ze
edd �dee fd"d#���Zejdefd$d#��Ze
defd%d&��Zejdefd'd&��Ze
defd(d)��Zejdefd*d)��Ze
de fd+d,��Z e jde fd-d,��Z e
defd.d/��Z e jdefd0d/��Z e
de fd1d2��Z e jde fd3d2��Z e
defd4d5��Z e jdefd6d5��Z e
d7d8� �Z!e
d9d:� �Z"e
defd;d<��Z#e
d=d>� �Z$d?d@� Z%dAdB� Z&dS )D�UAConfig)r" r# )r$ r% )r r! N�cfg� user_config�series�returnc
C s� |r d| _ || _d| _n
t� | _ t| j �\| _| _|r || _n,ztj�� | _W n# tyJ } zt j
d|d� t �
d� t� � | _W Y d}~nd}~ww d| jv rctj j i | j�
� �| jd �dd�| _|| _dS )� NzError loading user config)�exc_infozUsing default config valuesr1 T)� optional_type_errors_become_null)�cfg_pathr4 � invalid_keys�get_config_path� parse_configr5 r �read� Exception�LOG�warning�UserConfigData� from_dict�to_dictr6 )�selfr4 r5 r6 �e� rH �1/usr/lib/python3/dist-packages/uaclient/config.py�__init__R s,
��
�
zUAConfig.__init__c C � | j �dt�S )Nr* )r4 �getr �rF rH rH rI r* u � zUAConfig.contract_urlc C rK )Nr/ )r4 rL r rM rH rH rI r/ y rN zUAConfig.security_urlc C rK )Nr2 )r4 rL r rM rH rH rI r2 } rN zUAConfig.livepatch_urlc C � | j jS �N)r5 r rM rH rH rI r � � zUAConfig.http_proxy�valuec C � || j _tj �| j � d S rP )r5 r r �write�rF rR rH rH rI r � � c C rO rP )r5 r rM rH rH rI r � rQ zUAConfig.https_proxyc C rS rP )r5 r r rT rU rH rH rI r � rV c C rO rP )r5 r# rM rH rH rI r# � rQ zUAConfig.ua_apt_https_proxyc C rS rP )r5 r# r rT rU rH rH rI r# � rV c C rO rP )r5 r" rM rH rH rI r" � rQ zUAConfig.ua_apt_http_proxyc C rS rP )r5 r" r rT rU rH rH rI r" � rV )�maxsizec C �: | j j}|r|S | j j}|rt�tjjddd�� |S d S )Nr r$ ��old�new)r5 r$ r �event�infor �WARNING_DEPRECATED_FIELD�format�rF �
global_val� old_apt_valrH rH rI r$ � � ��z UAConfig.global_apt_http_proxyc C �. || j _d | j _tjj�� tj �| j � d S rP )r5 r$ r r3 �fget� cache_clearr rT rU rH rH rI r$ � � c C rX )Nr! r% rY )r5 r% r! r\ r] r r^ r_ r` rH rH rI r% � rc z UAConfig.global_apt_https_proxyc C rd rP )r5 r% r! r3 re rf r rT rU rH rH rI r% � rg c C � | j j}|d u r
dS |S )Ni`T )r5 r& �rF �valrH rH rI r& � � z UAConfig.update_messaging_timerc C rS rP )r5 r& r rT rU rH rH rI r& � rV c C rh )Ni@8 )r5 r' ri rH rH rI r' � rk zUAConfig.metering_timerc C rS rP )r5 r' r rT rU rH rH rI r' � rV c C rh )NF)r5 �poll_for_pro_licenseri rH rH rI rl � s z UAConfig.poll_for_pro_licensec C rS rP )r5 rl r rT rU rH rH rI rl � rV c C rh )NiX )r5 �polling_error_retry_delayri rH rH rI rm � s z"UAConfig.polling_error_retry_delayc C rS rP )r5 rm r rT rU rH rH rI rm rV c C rh )NT)r5 r( ri rH rH rI r(
rk zUAConfig.apt_newsc C rS rP )r5 r( r rT rU rH rH rI r( rV c C s | j j}|d u r
tS |S rP )r5 r) r ri rH rH rI r) rk zUAConfig.apt_news_urlc C rS rP )r5 r) r rT rU rH rH rI r) rV c C rK )Nr+ )r4 rL r rM rH rH rI r+ " rN zUAConfig.data_dirc C s8 | j �dd�}ztt|�� �W S ty tj Y S w )Nr. �DEBUG)r4 rL �getattr�logging�upper�AttributeErrorrn )rF r. rH rH rI r. & s
�zUAConfig.log_levelc C s | j �dtd �S )Nr- )r4 rL r rM rH rH rI r- . s zUAConfig.log_filec C s. | j �d�}|rt|t�r|S t�d|� i S )z>Return a dictionary of any features provided in uaclient.conf.r, zDUnexpected uaclient.conf features value. Expected dict, but found %s)r4 rL �
isinstance�dictrA rB )rF r, rH rH rI r, 2 s
�zUAConfig.featuresc C s dD ]"}t | |�}|d u rt�d|� qt|t�r |dk r$tj||d��q| js+| jr5| j s1| j
r5t� � �t �
d| jt j� t �
d| jt j� t �
d| j t j� t �
d| j
t j� t �
d| jt j� t �
d| jt j� | jsq| jr}t�| j| jtjj� n| j s�| j
r�t�| j | j
tjj� g }t�� r�t�| j| j� | js�t� tj �s�| js�t� tj �r�|� d� ddl m!} dd l"m#} || �j$}t%d
d � |D ��}|r�|�&| j| j� | js�|� |j �s�| js�|� |j �r�|� d � t'|�dk�rd
�(|�}t)t*j+j,|d�� d S d S )N)r& r' z1No config set for %s, default value will be used.r ��keyrR r
�httpsr )� livepatch)�_enabled_servicesc s s � | ]
}|j d kr|V qdS )rx N)r )�.0�entrH rH rI � <genexpr>� s � �z*UAConfig.process_config.<locals>.<genexpr>rx z, )�services)-ro rA �debugrs �intr �InvalidPosIntConfigValuer$ r% r" r# � InvalidProxyCombinationConfigr
�validate_proxy� PROXY_VALIDATION_APT_HTTP_URL� PROXY_VALIDATION_APT_HTTPS_URLr � PROXY_VALIDATION_SNAP_HTTP_URLr � PROXY_VALIDATION_SNAP_HTTPS_URLr �setup_apt_proxy�
AptProxyScope�GLOBAL�UACLIENTr �is_snapd_installed�configure_snap_proxy�get_config_option_value�HTTP_PROXY_OPTION�HTTPS_PROXY_OPTION�append�uaclientrx �-uaclient.api.u.pro.status.enabled_services.v1ry �enabled_services�any�configure_livepatch_proxy�len�join�printr �!PROXY_DETECTED_BUT_NOT_CONFIGUREDr_ ) rF �proprR �services_with_proxiesrx ry r� �livepatch_enabledr} rH rH rI �process_configA s�
��� ����
��
�
� � ��
��
�
��������
���zUAConfig.process_configc C s` | j d urt| j �D ]}t�d|� q
d| jv r.t�d� t�d� t�d� t�d� d S d S )Nz&Ignoring invalid uaclient.conf key: %sr1 z)legacy "ua_config" found in uaclient.confzPlease do the following:z� 1. run `sudo pro config set field=value` for each field/value pair present under "ua_config" in /etc/ubuntu-advantage/uaclient.confzQ 2. Delete "ua_config" and all sub-fields in /etc/ubuntu-advantage/uaclient.conf)r< �sortedrA rB r4 )rF � invalid_keyrH rH rI �warn_about_invalid_keys� s
�
���z UAConfig.warn_about_invalid_keys)NNN)'�__name__�
__module__� __qualname__�ua_scoped_proxy_options�global_scoped_proxy_options�&deprecated_global_scoped_proxy_optionsr r �strr r rC rJ �propertyr* r/ r2 r �setterr r# r" r r$ r% r r& r' �boolrl rm r( r) r+ r. r- r, r� r� rH rH rH rI r3 G s� ����
�#
jr3 r7 c C s t j�d�} | r
| S tS )z4Get config path to be used when loading config dict.�UA_CONFIG_FILE)�os�environrL r )� config_filerH rH rI r= � s r= c C sp t � t�}| s
t� } t�d| � tj�| �r |�t t
� | ��� i }tj �
� D ]Q\}}|�� }|�d�rx|dd� }|�d�rp|dd� }|�d�r^tj�|�rXt t
� |��}ntj|d��d |vri||i|d < q'||d |< q'|tv rx|||< q'|�|� d
|v r�tj�|d
�|d
< d D ]}t�|| �s�tj||| d ��q�t|�� ��t�}|D ]}|� |� q�||fS )
a� Parse known Pro config file
Attempt to find configuration in cwd and fallback to DEFAULT_CONFIG_FILE.
Any missing configuration keys will be set to CONFIG_DEFAULTS.
Values are overridden by any environment variable with prefix 'UA_'.
@param config_path: Fullpath to pro configfile. If unspecified, use
DEFAULT_CONFIG_FILE.
@return: Dict of configuration values.
z%Using client configuration file at %s�ua_� N� features_� �yaml)�filepathr, r+ )r* r/ ru ) �copyr r= rA r~ r� �path�exists�updater r
� load_filer� �items�lower�
startswith�endswithr � InvalidFeatureYamlConfigValuer �
expanduserr
�is_service_url�InvalidURLConfigValue�set�keys�
difference�VALID_UA_CONFIG_KEYS�pop) � config_pathr4 �env_keysrv rR �
field_name�feature_field_namer< r� rH rH rI r> � sH
��
� r> � override_keyc s � fdd�}|S )a� Decorator used to override function return by config settings.
To identify if we should override the function return, we check
if the config object has the expected override key, we use it
has, we will use the key value as the function return. Otherwise
we will call the function normally.
@param override_key: key to be looked for in the settings_override
entry in the config dict. If that key is present, we will return
its value as the function return.
c s t � �� �fdd��}|S )Nc s@ t � \} }| �di ���t�}|tkr �dkr|d fS |S � � S )Nr0 �
cloud_type)r> rL �UNSET_SETTINGS_OVERRIDE_KEY)r4 �_�value_override)�fr� rH rI �new_f s
�z>apply_config_settings_override.<locals>.wrapper.<locals>.new_f)r )r� r� �r� )r� rI �wrapper s
z/apply_config_settings_override.<locals>.wrapperrH )r� r� rH r� rI � apply_config_settings_override s
r� rP )/r� rp r� � functoolsr r �typingr r r r� r r r r
r r r
r �uaclient.defaultsr r r r r r r r �uaclient.filesr �
uaclient.yamlr � getLogger� replace_top_level_logger_namer� rA �MERGE_ID_KEY_MAPr� �UA_CONFIGURABLE_KEYSr� �get_event_loggerr\ r3 r� r= r> r� rH rH rH rI �<module> s. ((
�
|
?