o K&b7@s~ddlZddlZddlZddlmZmZmZddlmZddl m Z ddl m Z dZ dZdZd d d d d ZGddde ZdS)N)AnyDictList) HTTPError)util)AutoAttachCloudInstancezhttp://metadata/computeMetadata/v1/instance/service-accounts/default/identity?audience=contracts.canonical.com&format=full&licenses=TRUEz/sys/class/dmi/id/product_namezGoogle Compute EngineZ8045211386737108299Z6022427724719891830Z599959289349842382Z2592866803419978320)ZxenialZbionicZfocalZjammyc@sneZdZeejegdddeee ffddZ edefddZ ede fdd Z deefd d Zd S) UAAutoAttachGCPInstance))Z retry_sleepsreturncCstjtddid\}}d|iS)NzMetadata-FlavorZGoogle)Zheaders identityToken)rZreadurl TOKEN_URL)selfZ url_responseZ_headersr5/usr/lib/python3/dist-packages/uaclient/clouds/gcp.py identity_docs z$UAAutoAttachGCPInstance.identity_doccCsdS)NZgcpr)rrrr cloud_type'sz"UAAutoAttachGCPInstance.cloud_typecCs*tjtrtt}t|krdSdS)z$This machine is a viable GCPInstanceTF)ospathexistsDMI_PRODUCT_NAMErZ load_fileGCP_PRODUCT_NAMEstrip)rZ product_namerrr is_viable+s   z!UAAutoAttachGCPInstance.is_viablecCsN|jd}t|ddd}t|d}|dididgS) zGet a list of licenses from the GCP metadata. Instance identity token (jwt) carries a list of licenses associated with the instance itself. Returns an empty list if licenses are not present in the metadata. r .r z===zutf-8ZgoogleZcompute_engineZ license_id)rbase64Zurlsafe_b64decodesplitjsonloadsdecodeget)rtokenidentityZ identity_dictrrrget_licenses_from_identity5s  z2UAAutoAttachGCPInstance.get_licenses_from_identityN)__name__ __module__ __qualname__propertyrZretryrrstrrrrboolrrr$rrrrrs r)rrrtypingrrrZ urllib.errorrZuaclientrZuaclient.cloudsrrrrZ GCP_LICENSESrrrrrs"