o K&b @sddlZddlmZmZddlmZmZmZmZm Z dZ gdZ dZ dZ eZdefd d Z   dd eed eed eeeddfddZ dded eeeddfddZdedeefddZdS)N)ListOptional)apt event_logger exceptionsmessagesutilz /usr/bin/snap)g?g?g@z proxy.httpz proxy.httpsreturncCs dtvS)z(Returns whether or not snap is installedZsnapd)rZget_installed_packagesr r //usr/lib/python3/dist-packages/uaclient/snap.py is_installeds r http_proxy https_proxy retry_sleepscCsttstdtdS|s|rttj jdd|r-tj dddd|g|d|r?tj dddd |g|ddSdS) 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. z1Skipping configure snap proxy. {} does not exist.Nsnap)servicesetsystemz proxy.http={}rzproxy.https={}) rwhichSNAP_CMDloggingdebugformateventinforZSETTING_SERVICE_PROXYsubp)r rrr r r configure_snap_proxys(  r protocol_typecCs>ttstdtdStjdddd|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. z3Skipping unconfigure snap proxy. {} does not exist.NrZunsetrzproxy.{}r)rrrrrrr)rrr r r unconfigure_snap_proxy<s  rkeycCs8ztddd|g\}}|WStjyYdSw)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 rgetrN)rrstriprZProcessExecutionError)r out_r r r get_config_option_valueUs  r%)NNN)N)rtypingrrZuaclientrrrrrrZSNAP_INSTALL_RETRIESZHTTP_PROXY_OPTIONZHTTPS_PROXY_OPTIONZget_event_loggerrboolr strfloatrrr%r r r r s<  *