$+~dZddlmZddlmZddlmZddlmZddlmZddlm Z ddl m Z dd l m ZGd d eZy ) z+API helpers for interacting with attestors.)absolute_import)division)unicode_literals) list_pager)apis)util) exceptions)mapscveZdZdZddZdZddZdZdZdZ dd Z dd Z dd Z d Z dd ZddZdZdZy)Clientz(A client for interacting with attestors.Ncztj||_tj||_||_yN)rGetClientInstanceclientGetMessagesModulemessages api_version)selfrs :lib/googlecloudsdk/api_lib/container/binauthz/attestors.py__init__zClient.__init__ s.((5DK**;7DM"Dc|jjj|jj |j S)zGet the specified attestor.name)rprojects_attestorsGetr.BinaryauthorizationProjectsAttestorsGetRequest RelativeName)r attestor_refs rrz Client.Get%sC ;; ) ) - - DD**, E   rctj|jj|jj |j |xsd|ddS)z7List the attestors associated with the current project.)parentd attestorspageSize) batch_sizelimitfieldbatch_size_attribute)r YieldFromListrrr/BinaryauthorizationProjectsAttestorsListRequestr)r project_refr& page_sizes rListz Client.List,sW  # # && EE++- F 0#' ))rc|jtjk(r|jjS|jj Sr)rrV1rUserOwnedGrafeasNoteUserOwnedDrydockNoters r _GetNoteClasszClient._GetNoteClass7s:477* MM . .0 MM . .0rcD|jtjk(rdSdS)NuserOwnedGrafeasNoteuserOwnedDrydockNote)rrr/r2s rGetNotePropertyNamezClient.GetNotePropertyName<s&477* #$ "$rcl|jtjk(r |jS|jS)z7Return the Attestor's version-dependent Note attribute.)rrr/r5r6)rattestors r GetNoteAttrzClient.GetNoteAttrAs, 477"  * **  * **rc |jtj}|jjj |j j|j jd|j|d|j|j|ji|j|jS)z8Create an attestors associated with the current project.)r description) noteReference)r9 attestorIdr!) ParentrPROJECTS_COLLECTIONrrCreater1BinaryauthorizationProjectsAttestorsCreateRequestAttestorrr7r3Name)rrnote_refr<r+s rrBz Client.CreateHs%%d&>&>?K ;; ) ) 0 0 GG+T]]++!..0'++-/Ct/A/A/C"*"7"7"90$((*++- H   rcd|j|}td|j|jD}||vr.t j dj |jtd|j|jD|j|jj|jj|||jjj|}tfd|j|jDS)a|Add a PGP key to an attestor. Args: attestor_ref: ResourceSpec, The attestor to be updated. pgp_pubkey_content: The contents of the PGP public key file. comment: The comment on the public key. Returns: The added public key. Raises: AlreadyExistsError: If a public key with the same key content was found on the attestor. c34K|]}|jywr)asciiArmoredPgpPublicKey.0 public_keys r z#Client.AddPgpKey..isA?J ++?z4Provided public key already present on attestor [{}]c34K|]}|jywridrJs rrMz#Client.AddPgpKey..qA?J  ?rN)rIcommentc3>K|]}|jvr|ywrrPrKrL existing_idss rrMz#Client.AddPgpKey..|%.GJ == , G)rsetr: publicKeysr AlreadyExistsErrorformatrappendrAttestorPublicKeyrrUpdatenext)rrpgp_pubkey_contentrSr9existing_pub_keysupdated_attestorrVs @r AddPgpKeyzClient.AddPgpKeyXs&xx %HA**84??AA..  ) ) @ G Gmm  A**84??AAL X))00 ''%7 (  {{55<.rRrNz8Public key with ID [{}] already present on attestor [{}]) publicKeyPemsignatureAlgorithm)rQ pkixPublicKeyrSc3>K|]}|jvr|ywrrPrUs rrMz$Client.AddPkixKey..rWrX)rrYr:rZr r[r\rr]rr^ PkixPublicKeyrrr_r`) rrpkix_pubkey_contentpkix_sig_algorithm id_overriderSr9rcrVs @r AddPkixKeyzClient.AddPkixKeys&xx %HA**84??AAL;,#>  ) ) D K K8== * ++ X))00 ''--550#567 ( {{55<.rRrN-No matching public key found on attestor [{}]N) rrYr:rZr NotFoundErrorr\rrQrrr_)rr pubkey_idr9rVrLs r RemoveKeyzClient.RemoveKeysxx %HA**84??AAL $  $ $ 9 @ @mm   **84??-'?J ==I % ?-'DX)  KK""))(3 -'s C!c|j|}|j|jDcgc]}|jk(r|}}|s.t j dj |jt|dkDr.t jdj |j|d}|(|jst jd||_ |||_ |jjj|} t!fd|j| jDScc}w)a$Update a key on an attestor. Args: attestor_ref: ResourceSpec, The attestor to be updated. pubkey_id: The ID of the key to update. pgp_pubkey_content: The contents of the public key file. comment: The comment on the public key. Returns: The updated public key. Raises: NotFoundError: If an expected public key could not be located by ID. InvalidStateError: If multiple public keys matched the provided ID. InvalidArgumentError: If a non-PGP key is updated with pgp_pubkey_content. rrz4Multiple matching public keys found on attestor [{}]rz7Cannot update a non-PGP PublicKey with a PGP public keyc3@K|]}|jk(r|ywrrP)rKrLrts rrMz#Client.UpdateKey..s%'GJ ==I % Gs)rr:rZrQr rsr\rlenInvalidStateErrorrIInvalidArgumentErrorrSrrr_r`) rrrtrarSr9rL existing_keys existing_keyrcs ` r UpdateKeyzClient.UpdateKeysK$xx %H**84??'?J ==I % ?'   $ $ 9 @ @mm   =A  ( ( @ G Gmm  !#L%  2 2-- EG G.@l+$l{{55<rs0 2&''::D;U#VU#r