o K&b@sjddlmZddlmZddlmZmZmZmZm Z ddl m Z ddl m Z ddlmZddlmZedZd Zd d efd d efddefdZddeDZGdddeZde jdeefddZdee jde eeffddZdedeeefdefddZd ee jdee jfd!d"Zd#edeeeffd$d%Z d#edeeeffd&d'Z!d(S))) defaultdict)Enum)Any DefaultDictDictListTuple)Cache)package)UAConfig)get_platform_infoseries) esm-infraesm-appsZUbuntuz {}-securityZ UbuntuESMAppsz{}-apps-securityZ UbuntuESMz{}-infra-security)standard-securityrrcCsi|]\}}||qSr).0kvrr:/usr/lib/python3/dist-packages/uaclient/security_status.py s rc@s eZdZdZdZdZdZdZdS) UpdateStatusz2Represents the availability of a security package.Zupgrade_availableZpending_attachZpending_enableZupgrade_unavailableN)__name__ __module__ __qualname____doc__ AVAILABLE UNATTACHED NOT_ENABLED UNAVAILABLErrrrrs rr returncCsRg}|jjD] }|j|jftdkr|d|j|jftdkr&|dq|S)Nrr) installedoriginsoriginarchiveSERVICE_TO_ORIGIN_INFORMATIONappend)r esm_servicesr#rrrlist_esm_for_package"s     r(r"cCs2|D]}t|j|jf}|r||jfSqdS)zGTranslates the archive name in the version origin to a UA service name.)r))ORIGIN_INFORMATION_TO_SERVICEgetr#r$Zsite)r"r#servicerrrget_service_name0s r- service_nameua_infocCsP|dks|dr||dvrtjjS|dstjjS||dvr$tjjStjjS)zDefines the update status for a package based on the service name. For ESM-[Infra|Apps] packages, first checks if UA is attached. If this is the case, also check for availability of the service. rattachedenabled_servicesentitled_services)rrvaluerrr)r.r/rrrget_update_status;s  r4packagescCsdd|DS)zFilters a list of packages looking for available security updates. Checks if the package has a greater version available, and if the origin of this version matches any of the series' security repositories. cSs:g|]}|jD]}||jkrtdd|jDr|qqS)css |] }|j|jftvVqdS)N)r#r$r*)rr#rrr Ys  z5filter_security_updates...)Zversionsr!anyr")rr versionrrr Tsz+filter_security_updates..r)r5rrrfilter_security_updatesLsr:cfgcCsdggd}|jdd}|dr?d|d<|dD]&}|dtvr>|dd kr/|d |d|d d kr>|d |dq|S)z6Returns the UA information based on the config object.F)r0r1r2T)Z show_betar0servicesnameZentitledZyesr2statusenabledr1)r> ESM_SERVICESr&)r;r/r>r,rrr get_ua_info`s     rAc Cst|}d|i}g}t}dd|D}t||d<tt}tt}|D]}t|} | D] } || d7<q,q$t|} | D]$} t| j\} }t | |}|| d7<| | j j | j | ||dq>|d|d<|d |d <|d|d <|d |d <|d |d<d||dS)aZReturns the status of security updates on a system. The returned dict has a 'packages' key with a list of all installed packages which can receive security updates, with or without ESM, reflecting the availability of the update based on the UA status. There is also a summary with the UA information and the package counts. ZuacSsg|]}|jr|qSr)Z is_installed)rr rrrr9sz#security_status..Znum_installed_packages)r r8r.r>r#rZnum_esm_infra_packagesrZnum_esm_apps_packagesZnum_esm_infra_updatesZnum_esm_apps_updatesrZnum_standard_security_updatesz0.1)Z_schema_versionsummaryr5)rAr lenrintr(r:r-r"r4r&r r=r8)r;r/rCr5cacheZinstalled_packagesZ package_countZ update_countr r'r,Zsecurity_upgradable_versions candidater.Z origin_siter>rrrsecurity_statususD       rHN)" collectionsrenumrtypingrrrrrZaptr r Z apt_packageZuaclient.configr Z uaclient.utilr r r@formatr%itemsr*rZPackagestrr(ZOriginr-r4r:rArHrrrrs6