fdZddlmZddlmZddlmZddlmZddlmZddl m Z Gdd e Z y ) z+API Utilities for gcloud diagnose commands.)absolute_import)division)unicode_literals)metadata_utils)apis)iam_utilcDeZdZdZd dZdZdZdZdZd dZ d Z d Z y) DiagnoseClientz4Client for calling apis needed by diagnose commands.Nc|xstjdd|_|xstjdd|_|xstjdd|_y)z?Initializes DiagnoseClient with the common api clients it uses.computev1iamstorageN)rGetClientInstance_compute_client _iam_client_storage_client)selfcompute_client iam_clientstorage_clients =lib/googlecloudsdk/api_lib/compute/diagnose/diagnose_utils.py__init__zDiagnoseClient.__init__sX A$00DA !HT%;%;E4%HDA$00DA c|jj}|jjj|j t j ||j|}|jS)a\Signs a string with the private key of the provided service account. Args: service_account: The string email of a service account that has permissions to sign a blob. bytes_to_sign: The byte-string to sign. Returns: A byte-string signature of the provided blob, signed by the provided service account. ) bytesToSign)namesignBlobRequest) rMESSAGES_MODULEprojects_serviceAccountsSignBlob)IamProjectsServiceAccountsSignBlobRequestrEmailToAccountResourceNameSignBlobRequest signature)rservice_account bytes_to_signmessagesresponses rr!zDiagnoseClient.SignBlob%sv//H88AA::44_E$44)5+ ; ,-H   rc|jj}|jjj|j t j |}|jS)zLists all service accounts within a particular project. Args: project: The project string to search for the service account in. Returns: A list of service account message objects. )r)rrr List%IamProjectsServiceAccountsListRequestrProjectToProjectResourceNameaccounts)rprojectr(r)s rListServiceAccountsz"DiagnoseClient.ListServiceAccounts;sb//H88==6666w? 7 ABH   rc|jj}|jjj|j t j ||j|}|jS)a Creates a service account within the provided project. Args: project: The project string to create a service account within. account_id: The string id to create the service account with. Returns: A string email of the service account. ) accountId)rcreateServiceAccountRequest) rrr Create'IamProjectsServiceAccountsCreateRequestrr-CreateServiceAccountRequestemail)rr/ account_idr(r)s rCreateServiceAccountz#DiagnoseClient.CreateServiceAccountLsu//H88??8866w?(0(L(L$)M)& 9 '(H >>rc|jj}|jjj|j ||}|j D]}|cSy)aDGets the first bucket the project has access to with a matching prefix. Args: project: The id string of the project the bucket is associated with. prefix: The string literal prefix of the bucket being searched for. Returns: The first bucket message object found matching the prefix, or none. )prefixr/N)rrbucketsr+StorageBucketsListRequestitems)rr/r;r(r)buckets r FindBucketzDiagnoseClient.FindBucket^s_##33H##++00**&'*JLH.. m! rc|jj}|jj}|jjj }|jjjj |_||j _|jjjjd|_ |jj||j|S)a?Creates a bucket object that deletes its contents after a number of days. Args: days_to_live: The number of days to wait before deleting an item within this bucket. Count starts when the item is created. Returns: A bucket message object that has not yet been created in Cloud Storage. Delete)type) lifecycle) rrBucketLifecycleValueRuleValueListEntryConditionValue conditionage ActionValueactionruleappend)r days_to_liver(rDlifecycle_rules rCreateBucketWithLifecyclez(DiagnoseClient.CreateBucketWithLifecycleqs##33H..0I__33FFHN&&99HHJ#/N &&99EE F NN.) ??Y? //rc|jj}|jjj|j ||y)aInserts the bucket object as a GCS bucket associated with the project. Args: project_id: The project string to save the bucket to. bucket: The bucket message object to insert. Raises: HttpError: with status_code 409 if the bucket already exists. )r?r/N)rrr<InsertStorageBucketsInsertRequest)r project_idr?r(s r InsertBucketzDiagnoseClient.InsertBucketsC##33H  '',,FJ,OQrc |jj}|jjj|jdi|j }|j }||i}|jjj|j|jtj|||||jy)aWrites a key value pair to the metadata server. Args: project: The project string the instance is in. instance_ref: The instance the metadata server relates to. key: The string key to enter the data in. val: The string value to be written at the key. )metadataexisting_metadata)instancerXr/zoneN) rr instancesGetComputeInstancesGetRequestAsDictrX SetMetadata"ComputeInstancesSetMetadataRequestrrConstructMetadataMessager[) rr/ instance_refkeyvalr(rZrY new_metadatas rUpdateMetadatazDiagnoseClient.UpdateMetadatas##33H##--11+++Dl.A.A.CDFH )):L""..33]]#<<%"35"" 4 $%r)NNN) ) __name__ __module__ __qualname____doc__rr!r0r9r@rQrVrhr\rrr r s/<C,"$&0, Q%rr N) rm __future__rrrgooglecloudsdk.api_lib.computergooglecloudsdk.api_lib.utilrgooglecloudsdk.command_lib.iamrobjectr r\rrrss*2&'9,3Q%VQ%r