File: //usr/lib/python3/dist-packages/uaclient/__pycache__/config.cpython-38.pyc
U
��Jh�K � @ s d dl Z d dlZd dlZd dlmZmZ d dlmZmZm Z d dl
m Z m Z m
Z
mZmZmZmZmZ d dlmZmZmZmZmZmZmZmZm Z d dl m Z 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,e-d�dd�Z.ddd�Z/e-d�dd�Z0dS )� 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�BASE_VULNERABILITY_DATA_URL�CONFIG_DEFAULTS� CONFIG_FIELD_ENVVAR_ALLOWLIST�DEFAULT_CONFIG_FILE�DEFAULT_DATA_DIR)� state_files�user_config_file)� safe_load�name�type)ZavailableResourcesZresourceEntitlementsZ_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� vulnerability_data_url_prefix�lxd_guest_attach)
� contract_url�data_dir�features�log_file� log_level� security_urlr) �settings_overrides� ua_config�
livepatch_urlc @ s e Zd ZdZdZdZdGeeee f ee
j ee dd�dd�Z e
ed�d d
��Ze
ed�d d ��Ze
ed�d
d��Ze
ee d�dd��Zejed�dd��Ze
ee d�dd��Zejed�dd��Ze
ee d�dd��Zejed�dd��Ze
ee d�dd��Zejed�dd��Ze
ed�d d ��Zejed�d d ��Ze
edd �ee d�d d!���Zejed�d"d!��Ze
edd �ee d�d#d$���Zejed�d%d$��Ze
ed�d&d'��Zejed�d(d'��Ze
ed�d)d*��Z e jed�d+d*��Z e
e d�d,d-��Z e je d�d.d-��Z e
ed�d/d0��Z e jed�d1d0��Z e
e d�d2d3��Z e je d�d4d3��Z e
ed�d5d6��Z!e!jed�d7d6��Z!e
e
j"d�d8d9��Z#e#je
j"d�d:d9��Z#e
d;d<� �Z$e
d=d>� �Z%e
ed�d?d@��Z&e
dAdB� �Z'dCdD� Z(dEdF� Z)dS )H�UAConfig)r! r" )r# r$ )r r N)�cfg� user_config�series�returnc
C s� |rd| _ || _d| _nt� | _ t| j �\| _| _|r>|| _nVztj�� | _W nD tk
r� } z&t j
d|d� t �
d� t� � | _W 5 d}~X Y nX d| jkr�tj j | j�
� | jd �dd�| _|| _dS )� NzError loading user config)�exc_infozUsing default config valuesr2 T)Z optional_type_errors_become_null)Zcfg_pathr5 � invalid_keys�get_config_path� parse_configr6 r �read� Exception�LOG�warning�UserConfigDataZ from_dictZto_dictr7 )�selfr5 r6 r7 �e� rE �1/usr/lib/python3/dist-packages/uaclient/config.py�__init__V s(
�zUAConfig.__init__�r8 c C s | j �dt�S )Nr+ )r5 �getr �rC rE rE rF r+ y s zUAConfig.contract_urlc C s | j �dt�S )Nr0 )r5 rI r rJ rE rE rF r0 } s zUAConfig.security_urlc C s | j �dt�S )Nr3 )r5 rI r rJ rE rE rF r3 � s zUAConfig.livepatch_urlc C s | j jS �N)r6 r rJ rE rE rF r � s zUAConfig.http_proxy)�valuec C s || j _tj �| j � d S rK )r6 r r �write�rC rL rE rE rF r � s c C s | j jS rK )r6 r rJ rE rE rF r � s zUAConfig.https_proxyc C s || j _tj �| j � d S rK )r6 r r rM rN rE rE rF r � s c C s | j jS rK )r6 r" rJ rE rE rF r" � s zUAConfig.ua_apt_https_proxyc C s || j _tj �| j � d S rK )r6 r" r rM rN rE rE rF r" � s c C s | j jS rK )r6 r! rJ rE rE rF r! � s zUAConfig.ua_apt_http_proxyc C s || j _tj �| j � d S rK )r6 r! r rM rN rE rE rF r! � s c C s | j j}|d krtS |S rK )r6 r) r �rC �valrE rE rF r) � s z&UAConfig.vulnerability_data_url_prefixc C s || j _tj �| j � d S rK )r6 r) r rM rN rE rE rF r) � s )�maxsizec C s: | j j}|r|S | j j}|r6t�tjjddd�� |S d S )Nr r# ��old�new)r6 r# r �event�infor �WARNING_DEPRECATED_FIELD�format�rC Z
global_valZ old_apt_valrE rE rF r# � s ��z UAConfig.global_apt_http_proxyc C s. || j _d | j _tjj�� tj �| j � d S rK )r6 r# r r4 �fget� cache_clearr rM rN rE rE rF r# � s c C s: | j j}|r|S | j j}|r6t�tjjddd�� |S d S )Nr r$ rR )r6 r$ r rU rV r rW rX rY rE rE rF r$ � s ��z UAConfig.global_apt_https_proxyc C s. || j _d | j _tjj�� tj �| j � d S rK )r6 r$ r r4 rZ r[ r rM rN rE rE rF r$ � s c C s | j j}|d krdS |S )Ni`T )r6 r% rO rE rE rF r% � s z UAConfig.update_messaging_timerc C s || j _tj �| j � d S rK )r6 r% r rM rN rE rE rF r% � s c C s | j j}|d krdS |S )Ni@8 )r6 r& rO rE rE rF r& � s zUAConfig.metering_timerc C s || j _tj �| j � d S rK )r6 r& r rM rN rE rE rF r& � s c C s | j j}|d krdS |S )NF)r6 �poll_for_pro_licenserO rE rE rF r\ � s z UAConfig.poll_for_pro_licensec C s || j _tj �| j � d S rK )r6 r\ r rM rN rE rE rF r\ s c C s | j j}|d krdS |S )NiX )r6 �polling_error_retry_delayrO rE rE rF r] s z"UAConfig.polling_error_retry_delayc C s || j _tj �| j � d S rK )r6 r] r rM rN rE rE rF r] s c C s | j j}|d krdS |S )NT)r6 r' rO rE rE rF r' s zUAConfig.apt_newsc C s || j _tj �| j � d S rK )r6 r' r rM rN rE rE rF r' ! s c C s | j j}|d krtS |S rK )r6 r( r rO rE rE rF r( & s zUAConfig.apt_news_urlc C s || j _tj �| j � d S rK )r6 r( r rM rN rE rE rF r( - s c C s | j j}|d krtjjS |S rK )r6 r* r �LXDGuestAttachEnum�OFFrO rE rE rF r* 2 s zUAConfig.lxd_guest_attachc C s* |d krt jj}|| j_t j�| j� d S rK )r r^ r_ r6 r* rM rN rE rE rF r* 9 s c C s | j �dt�S )Nr, )r5 rI r rJ rE rE rF r, @ s zUAConfig.data_dirc C s> | j �dd�}ztt|�� �W S tk
r8 tj Y S X d S )Nr/ �DEBUG)r5 rI �getattr�logging�upper�AttributeErrorr` )rC r/ rE rE rF r/ D s
zUAConfig.log_levelc C s | j �dtd �S )Nr. )r5 rI r rJ rE rE rF r. L s zUAConfig.log_filec C s. | j �d�}|r*t|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)r5 rI �
isinstance�dictr@ rA )rC r- rE rE rF r- P s
�zUAConfig.featuresc
C sl dD ]D}t | |�}|d kr(t�d|� qt|t�r:|dk rtj||d��q| jsV| jrj| j sb| j
rjt� � �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� | js�| jr�t�| j| jtjj� n&| j �s
| j
�r t�| j | j
tjj� g }t�� �rtt�| j| j� | j�sTt� tj ��sj| j�stt� tj ��rt|� d� ddl m!} dd l"m#} || �j$}t%d
d � |D ��}|�r�|�&| j| j� | j�s�|� |j ��s�| j�s�|� |j ��r�|� d � t'|�dk�r d
�(|�}t)t*j+j,|d�� ddl-m.} | j/t0j1j2k�rR| | �j3�sRt)t*j4� nt5j6�7t5j8| j/d�� d S )N)r% r& z1No config set for %s, default value will be used.r ��keyrL r
Zhttpsr )� livepatch)�_enabled_servicesc s s | ]}|j d kr|V qdS )ri N)r )�.0ZentrE rE rF � <genexpr>� s
z*UAConfig.process_config.<locals>.<genexpr>ri z, )�services)� _is_attached)Z guest_attach)9ra r@ �debugre �intr ZInvalidPosIntConfigValuer# r$ r! r" Z InvalidProxyCombinationConfigr
Zvalidate_proxyZ PROXY_VALIDATION_APT_HTTP_URLZ PROXY_VALIDATION_APT_HTTPS_URLr Z PROXY_VALIDATION_SNAP_HTTP_URLr Z PROXY_VALIDATION_SNAP_HTTPS_URLr Zsetup_apt_proxyZ
AptProxyScopeZGLOBALZUACLIENTr Zis_snapd_installedZconfigure_snap_proxyZget_config_option_valueZHTTP_PROXY_OPTIONZHTTPS_PROXY_OPTION�append�uaclientri Z-uaclient.api.u.pro.status.enabled_services.v1rj �enabled_services�anyZconfigure_livepatch_proxy�len�join�printr Z!PROXY_DETECTED_BUT_NOT_CONFIGUREDrX Z(uaclient.api.u.pro.status.is_attached.v1rn r* r r^ r_ Z is_attachedZ-WARNING_LXD_GUEST_ATTACH_SET_BUT_NOT_ATTACHEDr Zlxd_pro_config_filerM Z LXDProConfig)
rC ZproprL Zservices_with_proxiesri rj rs Zlivepatch_enabledrm rn rE rE rF �process_config_ s�
� � ���� �� � � ��
�
��
�
� �������
�� �� �zUAConfig.process_configc C s\ | j d k r&t| j �D ]}t�d|� qd| jkrXt�d� t�d� t�d� t�d� d S )Nz&Ignoring invalid uaclient.conf key: %sr2 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; �sortedr@ rA r5 )rC � invalid_keyrE rE rF �warn_about_invalid_keys� s
�
��z UAConfig.warn_about_invalid_keys)NNN)*�__name__�
__module__� __qualname__Zua_scoped_proxy_optionsZglobal_scoped_proxy_optionsZ&deprecated_global_scoped_proxy_optionsr r �strr r rB rG �propertyr+ r0 r3 r �setterr r" r! r) r r# r$ rp r% r& �boolr\ r] r' r( r^ r* r, r/ r. r- rx r{ rE rE rE rF r4 K s� �� #
vr4 rH c C s t j�d�} | r| S tS )z4Get config path to be used when loading config dict.ZUA_CONFIG_FILE)�os�environrI r )Z config_filerE rE rF r< � s r< c C sx t � t�}| st� } t�d| � tj�| �r@|�t t
� | ��� i }tj �
� D ]�\}}|�� }|�d�rN|dd� }|�d�r�|dd� }|�d�r�tj�|�r�t t
� |��}n tj|d��d |kr�||i|d < q�||d |< qN|tkrN|||< qN|�|� d
|k�rtj�|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 %sZua_� NZ features_� Zyaml)�filepathr- r, )r+ r0 rg ) �copyr r< r@ ro r� �path�exists�updater r
Z load_filer� �items�lower�
startswith�endswithr Z InvalidFeatureYamlConfigValuer �
expanduserr
Zis_service_urlZInvalidURLConfigValue�set�keys�
difference�VALID_UA_CONFIG_KEYS�pop) Z config_pathr5 Zenv_keysrh rL Z
field_nameZfeature_field_namer; rz rE rE rF r= � sD
�
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:�dkr6|d fS |S � � S )Nr1 Z
cloud_type)r= rI �UNSET_SETTINGS_OVERRIDE_KEY)r5 �_Zvalue_override)�fr� rE rF �new_fA s
�z>apply_config_settings_override.<locals>.wrapper.<locals>.new_f)r )r� r� r� )r� rF �wrapper@ s
z/apply_config_settings_override.<locals>.wrapperrE )r� r� rE r� rF � apply_config_settings_override3 s
r� )N)1r� rb r� � functoolsr r �typingr r r rr r r r r
r r r
r Zuaclient.defaultsr r r r r r r r r Zuaclient.filesr r Z
uaclient.yamlr Z getLoggerZ replace_top_level_logger_namer| r@ ZMERGE_ID_KEY_MAPr� ZUA_CONFIGURABLE_KEYSr� Zget_event_loggerrU r4 r r< r= r� rE rE rE rF �<module> s. (
, � #
?