o a@sddlmZddlZddlmZddlZddlmZddlZe ddddl m Z m Z ddl Z ddlZddlZddlZeeZddlZdZd d Zd d Zd dZddZddZddZdddZdddZdS))print_functionN)datetimewrapsGtkz3.0)Giorz%/var/lib/ubuntu-advantage/status.jsoncCszt|_|j||j||j||jD]}tt|tj r3tj |}t |||qt d|qdS)Nz can not get name for object '%s')rZBuilderZbuilderZset_translation_domainZ add_from_fileZconnect_signalsZ get_objects issubclasstypeZ BuildableZget_namesetattrloggingdebug)selfpathdomainonamer>/usr/lib/python3/dist-packages/softwareproperties/gtk/utils.pysetup_ui's     rcCs,z tjd}|dkWStyYdSw)Nz#gnome-online-accounts-panel.desktopF)rZDesktopAppInfonew Exception)drrrhas_gnome_online_accounts4s    rcCstj}t}||jSN) aptsourcesdistro get_distro distro_infoUbuntuDistroInfoZis_ltscodenamerdirrris_current_distro_lts;s  r"cCs*tj}t}|j|t vSr) rrrrrrZ supportedrZnowdater rrris_current_distro_supported@s r$cCs@tj}t}|jdd}|D] }|j|jkr|SqdS)Nobject)result)rrrrrZget_allZseriesr)rr!Zreleasesreleaserrrcurrent_distroEs   r(c Cs,d}tjtr!tt }|}Wdn1swYn8z tjgdtjd}Wnt yG}zt d|iWYd}~Sd}~ww|j dkrVt d|j iS|j }|sat diSzt |}Wnt jy}zt d |iWYd}~Sd}~ww|d d d krt d |d |S) zBReturn a dict of all UA status information or empty dict on error.N)Zuastatusz --format=json)stdoutzFailed to run `ua status`: %srz(Ubuntu Advantage client returned code %dzOWarning: no Ubuntu Advantage status found. Is ubuntu-advantage-tools installed?z0Failed to parse ubuntu advantage client JSON: %sZ_schema_versionz0.1z#UA status schema version change: %s)osrexistsUA_STATUS_JSONopenread subprocessrunPIPErprint returncoder+jsonloadsZJSONDecodeErrorget)Z status_jsonstreamr&er*rrr get_ua_statusNsL         r; esm-infracCsd|st}d}d}|dgD]}|d|krqd|vr%td|dk}d|vr-|d}q||fS)akGet service availability and status for a specific UA service. Return a tuple (available, service_status). :boolean available: set True when either: - attached contract is entitled to the service - unattached machine reports service "availability" as "yes" :str service_status: will be one of the following: - "disabled" when the service is available and applicable but not active - "enabled" when the service is available and active - "n/a" when the service is not applicable to the environment or not entitled for the attached contract Fzn/aservicesr availableZyesr*)r;r8bool)Z service_namer*r>Zservice_statusservicerrrget_ua_service_status{srA 皙?csfdd}|S)a Retry calling the decorated function using an exponential backoff. Args: exceptions: The exception to check. may be a tuple of exceptions to check. tries: Number of times to try (not retry) before giving up. delay: Initial delay between retries in seconds. backoff: Backoff multiplier (e.g. value of 2 will double the delay each retry). cs tfdd}|S)Nc s}}|dkr@z|i|WSy;}zd||}t|t||d8}|9}WYd}~nd}~ww|dks |i|S)Nz{}, Retrying in {} seconds...)formatr Zwarningtimesleep)argskwargsZmtriesZmdelayr:msg)backoffdelay exceptionsftriesrrf_retrys     z*retry..deco_retry..f_retryr)rOrQrLrMrNrP)rOr deco_retrys zretry..deco_retryr)rNrPrMrLrSrrRrretrys rT)r<N)rBrCrD)Z __future__rZaptsources.distrorrr functoolsrZgiZrequire_versionZ gi.repositoryrrr6r,r1r Z getLogger__name__ZLOGrGr.rrr"r$r(r;rArTrrrrs.       -