U 2d@slddlZddlZddlZddlZddlZddlmZmZm Z ddl m Z m Z m Z mZmZmZdZdddgZdZd Zd Zd Ze ZeeeZed d efdefdefdefdefgZedddZ d+e ee ee ee!ddddZ"d,ee ee!ddddZ#ee edddZ$eedddZ%d d!Z&d"d#Z'd-ee eed%d&d'Z(eed(d)d*Z)dS).N)List NamedTupleOptional)apt event_logger exceptionsmessagessystemutilz /usr/bin/snapg?g?g@z proxy.httpz proxy.httpsz/run/snapd.socketz /v2/snaps/{} SnapPackagenameversionrevisionchannel publisher)returncCs dtkS)z(Returns whether or not snap is installedsnapd)rZget_installed_packages_namesrr//usr/lib/python3/dist-packages/uaclient/snap.pyis_snapd_installed!sr) http_proxy https_proxy retry_sleepsrcCsttstddS|s|r0ttjjdd|rPtj dddd|g|d|rptj dddd |g|ddS) a Configure snap to use http and https proxies. :param http_proxy: http proxy to be used by snap. If None, it will not be configured :param https_proxy: https proxy to be used by snap. If None, it will not be configured :param retry_sleeps: Optional 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. z6Skipping configure snap proxy. snapd is not installed.Nsnap)servicesetr z proxy.http={}rzproxy.https={}) rLOGdebugeventinforZSETTING_SERVICE_PROXYformatr subp)rrrrrrconfigure_snap_proxy&s r#) protocol_typerrcCs4tstddStjdddd|g|ddS)a Unset snap configuration settings for http and https proxies. :param protocol_type: String either http or https :param retry_sleeps: Optional 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. z8Skipping unconfigure snap proxy. snapd is not installed.NrZunsetr zproxy.{}r)rrrr r"r!)r$rrrrunconfigure_snap_proxyKs  r%)keyrcCs>z tddd|g\}}|WStjk r8YdSXdS)z Gets the config value from snap. :param protocol: can be any valid snap config option :return: the value of the snap config option, or None if not set rgetr N)r r"striprProcessExecutionError)r&out_rrrget_config_option_value`s  r,cCsZtddddddg\}}|}|dd}g}|D]}|d}|t|q6|S)Nrlistz--colorneverz --unicoder)r r" splitlinessplitappend get_snap_info)r*r+ZappsZsnapslinerrrrget_installed_snapsms  r5c Csttjz tWn4tjk rL}zt dt |W5d}~XYnXzt j ddddgtj dWntjk rtYnXdS)NzrMtypingrrrZuaclientrrrrr r r=rDZHTTP_PROXY_OPTIONZHTTPS_PROXY_OPTIONrNrPZget_event_loggerrZ getLoggerZreplace_top_level_logger_name__name__rr7r boolrfloatr#r%r,r5r:rBrFr3rrrrsf     &