U
    lHJe                     @   s  d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl	Z	d dl
mZmZ d dlmZmZmZmZmZmZmZ d dlZd dlmZ d dlmZmZmZmZmZmZ d dlm Z  dZ!dZ"d	Z#d
Z$dZ%dZ&dZ'dZ(dZ)dZ*dZ+dZ,dZ-dZ.dZ/dZ0dZ1ej23e dej23e dgej23e dej23e dgdZ4dddgZ5e6 Z7e8e9e:Z;ej<G d d! d!ej=Z>ed"d#e?fd$e?fd%e?fgZ@d&d' ZAeAe?e?d(d)d*ZBd+d, ZCe?ee? d-d.d/ZDdee? ee? eee?e?f  e?d0d1d2ZEedd3dee? eee?e?f  e?d4d5d6ZFG d7d8 d8ZGd9d: ZHd;d< ZIe?ee? d=d>d?ZJde?eKee? dAdBdCZLdeee?e?f  e?dDdEdFZMeNejOejPfe5e?dGdHdIZQdee? eee?  eee?e?f  e?dJdKdLZRe?eejS dMdNdOZTdejSee? eejU dPdQdRZVe?e?e?ee? e?ddSdTdUZWdVdW ZXdXdY ZYde?e?ee? ddZd[d\ZZd]d^ Z[d_d` Z\dadb Z]dcdd Z^ddedfdgZ_e?eKdhdidjZ`ee@ dkdldmZadeKee? dndodpZbdde>jcfee? ee? ee> ddqdrdsZdeee dkdtduZfeej dkdvdwZgdxdy Zhddkdzd{Ziee? e?d|d}d~Zjee? e?d|ddZkee? dddZldd Zmdd Znee? ee?ee?ee? f f dddZodS )    N)	lru_cachewraps)DictIterableList
NamedTupleOptionalSetUnion)AcquireProgress)event_logger
exceptionsgpgmessagessystemutil)ESM_APT_ROOTDIRg      N@z  # ubuntu-advantage-toolszDir::Etc::netrc/zDir::Etc::netrcparts/zDir::State::lists/a  /*
 * Autogenerated by ubuntu-advantage-tools
 * Do not edit this file directly
 *
 * To change what ubuntu-advantage-tools sets, use the `pro config set`
 * or the `pro config unset` commands to set/unset either:
 *      global_apt_http_proxy and global_apt_https_proxy
 * for a global apt proxy
 * or
 *      ua_apt_http_proxy and ua_apt_https_proxy
 * for an apt proxy that only applies to Ubuntu Pro related repos.
 */
z$Acquire::http::Proxy "{proxy_url}";
z%Acquire::https::Proxy "{proxy_url}";
z4Acquire::http::Proxy::esm.ubuntu.com "{proxy_url}";
z5Acquire::https::Proxy::esm.ubuntu.com "{proxy_url}";
z/etc/apt/trusted.gpg.d/z/usr/share/keyringsz/usr/lib/apt/methods/httpsz /usr/sbin/update-ca-certificatesz//etc/apt/apt.conf.d/90ubuntu-advantage-aptproxyz*/var/lib/apt/periodic/update-success-stampaL  # Written by ubuntu-advantage-tools

deb https://esm.ubuntu.com/{name}/ubuntu {series}-{name}-security main
# deb-src https://esm.ubuntu.com/{name}/ubuntu {series}-{name}-security main

deb https://esm.ubuntu.com/{name}/ubuntu {series}-{name}-updates main
# deb-src https://esm.ubuntu.com/{name}/ubuntu {series}-{name}-updates main
zetc/apt/sources.listzvar/lib/dpkg/statuszvar/cache/apt/archives/partialzvar/lib/apt/lists/partial)filesfoldersg      ?g      @g      $@c                   @   s   e Zd Ze Ze ZdS )AptProxyScopeN)__name__
__module____qualname__objectGLOBALUACLIENT r   r   ./usr/lib/python3/dist-packages/uaclient/apt.pyr   W   s   r   InstalledAptPackagenameversionarchc                    s   t   fdd}|S )z*Decorator ensuring apt_pkg is initialized.c                     s"   t jddkrt    | |S )NDir )apt_pkgconfiggetinit)argskwargsfr   r   new_fe   s    z"ensure_apt_pkg_init.<locals>.new_f)r   )r+   r,   r   r*   r   ensure_apt_pkg_initb   s    r-   abc                 C   s   t | |S N)r$   version_comparer.   r   r   r   r2   p   s    r2   c              
   C   s  |  d\}}tjdsdS zFt 4}tjddd||||tj	|dgt
td W 5 Q R X W n tjk
r } z^|jdkrt|j }td	|rtj| d
ntd|rtj| d
tjt|dW 5 d}~X Y n& tjk
r   tjt
|dY nX dS )a  Validate apt credentials for a PPA.

    @param repo_url: private-ppa url path
    @param username: PPA login username.
    @param password: PPA login password or resource token.

    @raises: UbuntuProError for invalid credentials, timeout or unexpected
        errors.
    ://z/usr/lib/apt/apt-helperNzdownload-filez{}://{}:{}@{}/pool/zapt-helper-output)Ztimeoutretry_sleepsd   z401\s+unauthorized|httperror401)repozconnection timed outZdetail)Zsecondsr6   )splitospathexiststempfileZTemporaryDirectoryr   subpformatjoinAPT_HELPER_TIMEOUTAPT_RETRIESr   ProcessExecutionErrorZ	exit_codestrstderrlowerresearchZAPTInvalidCredentialsZ
APTTimeoutAPTUnexpectedError
subprocessZTimeoutExpiredZAPTCommandTimeout)repo_urlusernamepasswordZprotocol	repo_pathZtmpderD   r   r   r   assert_valid_apt_credentialsu   s@    

   
  rO   )	apt_errorreturnc                 C   sV   t  }|  dD ]<}|rtd|}|rd| d dd  }|| q|S )aA  Parse apt update errors for invalid apt config in user machine.

    This functions parses apt update errors regarding the presence of
    invalid apt config in the system, for example, a ppa that cannot be
    reached, for example.

    In that scenario, apt will output a message in the following formats:

    The repository 'ppa 404 Release' ...
    Failed to fetch ppa 404 ...

    On some releases, both of these errors will be present in the apt error
    message.

    :param apt_error: The apt error string
    :return: a NamedMessage containing the error message
    
z2(Failed to fetch |The repository .)(?P<url>[^\s]+)z- urlz/distsr   )setstripr8   rF   rG   	groupdictadd)rP   failed_reposlineZpattern_matchZrepo_url_matchr   r   r   (_parse_apt_update_for_invalid_apt_config   s     rZ   )cmd	error_msgoverride_env_varsrQ   c              
   C   s   zt j| dt|d\}}W n| tjk
r } z\dt|jkrFt n$t|j}|rjtj	d
t|d|rr|nt|}tj|dW 5 d}~X Y nX |S )a  Run an apt command, retrying upon failure APT_RETRIES times.

    :param cmd: List containing the apt command to run, passed to subp.
    :param error_msg: The string to raise as UbuntuProError when all retries
       are exhausted in failure.
    :param override_env_vars: Passed directly as subp's override_env_vars arg

    :return: stdout from successful run of the apt command.
    :raise UbuntuProError: on issues running apt-cache policy.
    T)Zcapturer4   r]   z%Could not get lock /var/lib/dpkg/lockrR   )rX   r7   N)r   r=   rA   r   rB   rC   rD   APTProcessConflictErrorrZ   APTInvalidRepoErrorr?   sortedrH   )r[   r\   r]   out_errrN   rX   msgr   r   r   run_apt_command   s$    

rd   )maxsize)r\   r]   rQ   c                 C   s   t ddg| |dS )N	apt-cachepolicy)r[   r\   r]   )rd   )r\   r]   r   r   r   get_apt_cache_policy   s
    rh   c                   @   s$   e Zd Zdd Zdd Zdd ZdS )PreserveAptCfgc                 C   s   || _ i | _d S r1   )apt_funccurrent_apt_cfg)selfrj   r   r   r   __init__   s    zPreserveAptCfg.__init__c                    s&   t j  fdd  D | _|  S )Nc                    s   i | ]}|t  |qS r   )copydeepcopyr&   ).0keycfgr   r   
<dictcomp>  s     z,PreserveAptCfg.__enter__.<locals>.<dictcomp>)r$   r%   keysrk   rj   )rl   r   rr   r   	__enter__   s
    
zPreserveAptCfg.__enter__c                 C   s4   t j}| j D ]}||| j|  qt   d S r1   )r$   r%   rk   ru   rT   Zinit_system)rl   typevalue	tracebackrs   rq   r   r   r   __exit__  s    zPreserveAptCfg.__exit__N)r   r   r   rm   rv   rz   r   r   r   r   ri      s   ri   c                  C   s.   t j D ]} t j|  q
t   t d S r1   )r$   r%   ru   clearr'   Cacherq   r   r   r   get_apt_pkg_cache  s    r~   c                  C   sh   zJt j D ]} td| st j|  qt jdt t   t 	d W S  t
k
rb   i  Y S X d S )Nz^Acquirer"   )r$   r%   ru   rF   rG   r{   rT   r   r'   r|   	Exceptionr}   r   r   r   get_esm_apt_pkg_cache  s    r   )pkg_namerQ   c              	   C   sT   t t4}z||  }W n  tk
r6   Y W 5 Q R  d S X W 5 Q R X |jrP|jjS d S r1   )ri   r~   KeyErrorcurrent_verver_str)r   cachepackager   r   r   get_pkg_version+  s    
r   F)r   check_esm_cacherQ   c              
   C   s  t t`}z||  }W n  tk
r6   Y W 5 Q R  d S X t|}||}|s^W 5 Q R  d S |j}W 5 Q R X |sv|S t t}|rz||  }W n$ tk
r   | Y W  5 Q R  S X t|}	|	|}
|
s|W  5 Q R  S |
j}t||dkr|W  5 Q R  S W 5 Q R X |S )Nr   )	ri   r~   r   r$   ZDepCacheZget_candidate_verr   r   r2   )r   r   r   r   Z	dep_cache	candidateZcandidate_versionZ	esm_cacheZesm_packageZesm_dep_cacheZesm_candidateZesm_candidate_versionr   r   r   get_pkg_candidate_version8  s@    





 r   )r]   rQ   c              
   C   s   zztddg| d}W n~ tjk
r4   t Y nb tjk
rd } ztj|jdW 5 d }~X Y n2 tj	k
r } ztj
|jdW 5 d }~X Y nX W 5 t   X |S )Napt-getupdater[   r]   Zrepo_msgr7   )rh   cache_clearrd   r   r^   ZAPTUpdateProcessConflictErrorr_   ZAPTUpdateInvalidRepoErrorrc   ZUbuntuProErrorAPTUpdateFailed)r]   ra   rN   r   r   r   run_apt_update_commandd  s     
$
r   )sources_list_pathc                 C   s  t t}tjdtj|  tjdd tjdd t }|	  t
 }tjtjdd}t|}z|z"| |||d W 5 Q R X W nT ttd	d
k
r   t Y n2 tk
r } ztjt|dW 5 d }~X Y nX W 5 t  X W 5 Q R X d S )NzDir::Etc::sourcelistzDir::Etc::sourcepartszN.O.F.O.L.D.E.RzAPT::List-Cleanup0zDir::State::Listslockr   Errorr   r7   )ri   r~   r$   r%   rT   r9   r:   abspath
SourceListread_main_listr   r?   Zfind_dirZFileLockrh   r   r   getattrr   r^   SystemErrorr   rC   )r   r   sources_listfetch_progressZ	lock_filer   rN   r   r   r   update_sources_listz  s.    
 

 
&r   )packagesapt_optionsr]   rQ   c              
   C   s~   |d krg }zt dddg| |  |d}W nN tjk
rH   t Y n2 tjk
rx } ztj|jdW 5 d }~X Y nX |S )Nr   install--assume-yesr   r   )rd   r   r^   ZAPTInstallProcessConflictErrorr_   ZAPTInstallInvalidRepoErrorrc   )r   r   r]   ra   rN   r   r   r   run_apt_install_command  s     
 r   )originrQ   c              	   C   s\   t  }tt@}|jD ]2}|j}|r|jD ]\}}|j| kr*|| q*qW 5 Q R X t|S r1   )	rT   ri   r~   r   r   	file_listr   rW   list)r   resultr   r   Zinstalled_versionfile_r   r   r    get_installed_packages_by_origin  s    


r   )r   exclude_originrQ   c                    s6   g }| j D ]&} fdd|jD }|r
|| q
|S )Nc                    s(   g | ] \}}|j d kr|j kr|qS )Znow)Z	componentr   )rp   r   r   r   r   r   
<listcomp>  s   
 
z3get_remote_versions_for_package.<locals>.<listcomp>)Zversion_listr   append)r   r   Zvalid_versionsr    Zvalid_originsr   r   r   get_remote_versions_for_package  s    

r   )repo_filenamerJ   credentialssuiteskeyring_filerQ   c                 C   s2  z| d\}}W n tk
r.   d}|}Y nX t j}|drP|dd }t||| d}tddgtj	}	|	
 D ]&}
d	||
krqxd
|
krqxd} qqxd}|D ]D}||krqd}d|kr|std|| d}|dj|||d7 }qt| | t||| tjt|}tjt|}t|| dS )zAdd an authenticated apt repo and credentials to the system.

    @raises: InvalidAPTCredentialsError when the token provided can't access
        the repo PPA.
    :Zbearer/NFrf   rg   za={}-updatesz	o=Ubuntu,Tr#   z-updatesz?Not enabling apt suite "%s" because "%s-updates" is not enabledz# zC{maybe_comment}deb {url} {suite} main
# deb-src {url} {suite} main
)maybe_commentrS   suite)r8   
ValueErrorr   Zget_release_infoseriesendswithrO   rd   r   ZAPT_POLICY_FAILED
splitlinesr>   LOGwarning
write_fileadd_apt_auth_conf_entryr9   r:   r?   KEYRINGS_DIRAPT_KEYS_DIRr   Zexport_gpg_key)r   rJ   r   r   r   rK   rL   r   Zupdates_enabledrg   rY   contentr   r   Zsource_keyring_fileZdestination_keyring_filer   r   r   add_auth_apt_repo  sZ    


   r   c                 C   s   t  }| d\}}|ds&|d7 }tj|r>t|}nd}dj|||t	d}d}g }	|
 D ]\}
|s|
 }t|dkr|d }||kr|	| d	}qd||kr|	| d	}|	|
 qd|s|	| |	d tj|d
|	dd dS )zBAdd or replace an apt auth line in apt's auth.conf file or conf.d.r3   r   r#   z:machine {repo_path} login {login} password {password}{cmt})rM   loginrL   ZcmtF      TrR     modeN)!get_apt_auth_file_from_apt_configr8   r   r9   r:   r;   r   	load_filer>   APT_AUTH_COMMENTr   lenr   r   r?   )rJ   r   rL   apt_auth_file	_protocolrM   Zorig_contentZrepo_auth_lineZadded_new_authZ	new_linesrY   Z
split_lineZcurr_line_repor   r   r   r     sB    
	



r   c                    s   |  d\}}|dr$|dd }t }tj|rt|}dj|d d	 fdd	|
 D }|sxt| ntj||d
d dS )z+Remove a repo from the shared apt auth filer3   r   Nr   zmachine {repo_path}/ login)rM   rR   c                    s   g | ]} |kr|qS r   r   )rp   rY   Zauth_prefixr   r   r   P  s      z2remove_repo_from_apt_auth_file.<locals>.<listcomp>r   r   )r8   r   r   r9   r:   r;   r   r   r>   r?   r   ensure_file_absentr   )rJ   r   rM   r   Zapt_authr   r   r   r   remove_repo_from_apt_auth_fileF  s    

r   )r   rJ   r   rQ   c                 C   s2   t |  |r&tjt|}t | t| dS )z>Remove an authenticated apt repo and credentials to the systemN)r   r   r9   r:   r?   r   r   )r   rJ   r   r   r   r   remove_auth_apt_repoX  s
    

r   c                 C   sB   | d\}}|dr$|dd }dj||d}t| | dS )z.Add an apt preferences file and pin for a PPA.r3   r   Nr   z<Package: *
Pin: release o={origin}
Pin-Priority: {priority}
)r   priority)r8   r   r>   r   r   )Zapt_preference_filerJ   r   r   r   rM   r   r   r   r   add_ppa_pinningc  s    
 r   c                  C   sZ   t dddtg\} }| r,| dd d S t dddtg\} }| dd dS dS )	z7Return to patch to the system configured APT auth file.
apt-configshellrq   'r   z90ubuntu-advantager   N)r   r=   APT_CONFIG_AUTH_PARTS_DIRr8   APT_CONFIG_AUTH_FILErstrip)ra   rb   r   r   r   r   p  s    

r   c              
   C   s~   |  d\}}|dr$|dd }d}tdddtg\}}|rP| d	d
 }|dd}tttj	
||d| S )zEList any apt files in APT_CONFIG_LISTS_DIR given repo_url and series.r3   r   Nr   z/var/lib/apt/listsr   r   rq   r   r   r   z
_dists_{}*)r8   r   r   r=   APT_CONFIG_LISTS_DIRreplacer`   globr9   r:   r?   r>   )rJ   r   r   rM   Z	lists_dirra   rb   Zaptlist_filenamer   r   r   find_apt_list_files~  s$    

 r   c                 C   s   t | |D ]}t| q
dS )z?Remove any apt list files present for this repo_url and series.N)r   r   r   )rJ   r   r:   r   r   r   remove_apt_list_files  s    r   )_entitlementsc                 C   s   ddl m} | dkr$ddlm} |} | jD ]}t||s:q*|jj|jd}|j	j|jd}t
j|rtjtjj|dtjd t| t
j|r*tjtjj|dtjd t| q*dS )a  
    Clean apt files written by uaclient

    :param _entitlements:
        The uaclient.entitlements module to use, defaults to
        uaclient.entitlements. (This is only present for testing, because the
        import happens within the function to avoid circular imports.)
    r   )RepoEntitlementN)entitlementsr   )filename)Z	file_type)Zuaclient.entitlements.repor   uaclientr   ZENTITLEMENT_CLASSES
issubclassZrepo_list_file_tmplr>   r   Zrepo_pref_file_tmplr9   r:   r;   eventinfor   ZAPT_REMOVING_SOURCE_FILEsysrD   r   r   ZAPT_REMOVING_PREFERENCES_FILE)r   r   Z__entitlementsZent_clsZ	repo_fileZ	pref_filer   r   r   clean_apt_files  s.    	


r   )pkgrQ   c                 C   s
   | t  kS r1   )get_installed_packages_names)r   r   r   r   is_installed  s    r   )rQ   c                  C   s2   t dddg\} }|  dd  }dd |D S )NZaptr   z--installedr   c                 S   s8   g | ]0}t |d d |dd |dd dqS )r   r    r   r   )r   r    r!   )r   r8   )rp   entryr   r   r   r     s   z*get_installed_packages.<locals>.<listcomp>)r   r=   r   )ra   r   package_listr   r   r   get_installed_packages  s
    r   )include_versionsrQ   c                 C   s   t  }dd |D }|S )Nc                 S   s   g | ]
}|j qS r   r   )rp   r   r   r   r   r     s     z0get_installed_packages_names.<locals>.<listcomp>)r   )r   r   Z	pkg_namesr   r   r   r     s    r   )
http_proxyhttps_proxyproxy_scoperQ   c                 C   s   | s|rB|rBd}|t jkr d}n|t jkr.d}ttjj|d d}| r|t jkrf|tj| d7 }n|t jkr|t	j| d7 }|r|t jkr|t
j|d7 }n|t jkr|tj|d7 }|dkrt| }|dkrtt ntt| dS )a  
    Writes an apt conf file that configures apt to use the proxies provided as
    args.
    If both args are None, then no apt conf file is written. If this function
    previously wrote a conf file, and was run again with both args as None,
    the existing file is removed.

    :param http_proxy: the url of the http proxy apt should use, or None
    :param https_proxy: the url of the https proxy apt should use, or None
    :return: None
    r#   z	UA-scopedglobal)Zscope)Z	proxy_urlN)r   r   r   r   r   r   ZSETTING_SERVICE_PROXY_SCOPEr>   APT_CONFIG_UA_PROXY_HTTPAPT_CONFIG_GLOBAL_PROXY_HTTPAPT_CONFIG_UA_PROXY_HTTPSAPT_CONFIG_GLOBAL_PROXY_HTTPSAPT_PROXY_CONFIG_HEADERr   r   APT_PROXY_CONF_FILEr   )r   r   r   messageZapt_proxy_configr   r   r   setup_apt_proxy  sD    







r   c                  C   s    d } t jtrt tj} | S r1   )r9   r:   r;   APT_UPDATE_SUCCESS_STAMP_PATHstatst_mtimeZ
cache_timer   r   r   get_apt_cache_time  s    r  c                  C   s$   t  } | d krd S tj| tjjS r1   )r  datetimeZfromtimestamptimezoneZutcr  r   r   r   get_apt_cache_datetime  s    r  c                     s   t j tjtddd td td  } t fdd| D rBd S tt td D ]}t	| qTtd D ]}tj
|ddd	 qld S )
Nz**/*T)	recursiver   r   c                 3   s   | ]}| kV  qd S r1   r   )rp   r   Zexisting_filesr   r   	<genexpr>$  s     z._ensure_esm_cache_structure.<locals>.<genexpr>i  )exist_okr   )r   r9   r:   r?   r   ESM_BASIC_FILE_STRUCTUREallr   Zensure_folder_absentZcreate_filemakedirs)Zdesired_filesr   Zfolderr   r	  r   _ensure_esm_cache_structure  s     
r  c                 C   s  t  sd S t  ddlm} ddlm} ddlm}m	} d}d}| 
d}|d kr`|| d }|dg D ]D}|dd	d
kr|dddk}|dd	dkrl|dddk}ql|| }	|r|	 d |jkr|	  n|	  t  r || }
|r|
 d |jkr|
  n|
  ttx}t }|  G dd dt}| }z|||d W n8 tk
r } ztdt| W 5 d }~X Y nX W 5 Q R X d S )Nr   )status)ApplicationStatus)ESMAppsEntitlementESMInfraEntitlementFzstatus-cacheservicesr   r#   zesm-appsZ	availableZnoZyesz	esm-infrac                   @   s,   e Zd ZejdddZejdddZdS )z-update_esm_caches.<locals>.EsmAcquireProgress)itemc                 S   s   t d|j d S )NzFetched ESM Apt Cache item: {})r   debugr>   urirl   r  r   r   r   doneg  s    z2update_esm_caches.<locals>.EsmAcquireProgress.donec                 S   s   t d|j d S )Nz&Failed to fetch ESM Apt Cache item: {})r   r   r>   r  r  r   r   r   failj  s    
z2update_esm_caches.<locals>.EsmAcquireProgress.failN)r   r   r   r$   ZAcquireItemDescr  r  r   r   r   r   EsmAcquireProgressf  s   r  z%Failed to fetch the ESM Apt Cache: {})r   Zis_current_series_ltsr  Zuaclient.actionsr  Z(uaclient.entitlements.entitlement_statusr  Zuaclient.entitlements.esmr  r  Z
read_cacher&   Zapplication_statusZDISABLEDZsetup_local_esm_repoZdisable_local_esm_repoZis_current_series_active_esmri   r   r$   r   r   r   r   r   r   r   r>   rC   )rs   r  r  r  r  Zapps_availableZinfra_availableZcurrent_statusserviceZappsZinfrar   r   r  r   rN   r   r   r   update_esm_caches1  sP    




	r  package_namesZerror_messagec                 C   s(   t dddddgt|  |ddid d	S )
a  
    Remove APT packages from the system.

    Setting DEBIAN_FRONTEND to noninteractive makes sure no prompts will
    appear during the operation. In this case, --force-confdef will
    automatically pick the default option when some debconf should appear.
    In the absence of a default option, --force-confold will automatically
    choose to keep the old configuration file.
    r   remover   $-o Dpkg::Options::="--force-confdef"$-o Dpkg::Options::="--force-confold"DEBIAN_FRONTENDnoninteractiver]   Nrd   r   r  r   r   r   remove_packagesv  s    
r'  c                 C   s(   t dddddgt|  |ddid d	S )
a  
    Purge APT packages from the system - remove everything.

    Setting DEBIAN_FRONTEND to noninteractive makes sure no prompts will
    appear during the operation. In this case, --force-confdef will
    automatically pick the default option when some debconf should appear.
    In the absence of a default option, --force-confold will automatically
    choose to keep the old configuration file.
    r   purger   r!  r"  r#  r$  r%  Nr&  r  r   r   r   purge_packages  s    
r)  r  c                 C   s   t | dddgddid dS )a6  
    Install packages, allowing downgrades.

    The --allow downgrades flag is needed because sometimes we need to
    reinstall the packages to a lower version (passed in the package_name
    string, as package=version).

    Setting DEBIAN_FRONTEND to noninteractive makes sure no prompts will
    appear during the operation. In this case, --force-confdef will
    automatically pick the default option when some debconf should appear.
    In the absence of a default option, --force-confold will automatically
    choose to keep the old configuration file.
    z--allow-downgradesr!  r"  r#  r$  )r   r]   N)r   r*  r   r   r   reinstall_packages  s    r+  c                  C   s*   t j D ]} t j|  q
t   t jS r1   )r$   r%   ru   r{   Zinit_configr}   r   r   r   _get_apt_config  s    r,  c              	   C   s"   t t}|| }W 5 Q R X |S r1   )ri   r,  r   )Zbase_keyapt_cfgZapt_cfg_keysr   r   r   get_apt_config_keys  s    
r.  )	cfg_namesrQ   c              	   C   sN   i }t t8}| D ],}||}t|s6||p4d}|||< qW 5 Q R X |S )z
    Get all APT configuration values for the given config names. If
    one of the config names is not present on the APT config, that
    config name will have a value of None
    N)ri   r,  r&   rC   Z
value_list)r/  Zapt_cfg_dictr-  Zcfg_nameZ	cfg_valuer   r   r   get_apt_config_values  s    

r0  )NN)NN)F)N)NN)N)N)F)prn   r  enumr   Zloggingr9   rF   rI   r   r<   	functoolsr   r   typingr   r   r   r   r   r	   r
   r$   Zapt.progress.baser   r   r   r   r   r   r   r   Zuaclient.defaultsr   r@   r   r   r   r   r   r   r   r   r   r   r   ZAPT_METHOD_HTTPS_FILEZCA_CERTIFICATES_FILEr   r   ZESM_REPO_FILE_CONTENTr:   r?   r  rA   Zget_event_loggerr   Z	getLoggerZreplace_top_level_logger_namer   r   uniqueEnumr   rC   r   r-   r2   rO   rZ   rd   rh   ri   r~   r   r   boolr   r   Zretryr^   r   r   r   ZPackager   ZVersionr   r   r   r   r   r   r   r   r   r   r   r   r   r   r   floatr  r  r  r  r'  r)  r+  r,  r.  r0  r   r   r   r   <module>   s2  $ 
 *(  *    - 
'  
  ?+   %8E
