,dZddlmZddlmZddlmZddlmZddlmZ ddl m Z ddl m Z dd l mZd Zd Zd Zd ZdZy)z%Utils for Fleet memberships commands.)absolute_import)division)unicode_literals)base) exceptions)api_util) resources)errorsc2~~|jdz|_|S)zSet the appropriate request path in project attribute for initializeHub requests. Args: ref: reference to the membership object. args: command line arguments. request: API request to be issued Returns: modified request z/locations/global/memberships)projectrefargsrequests Blib/googlecloudsdk/command_lib/container/fleet/memberships/util.pySetInitProjectPathrs 4OO&EE'/ .c~|jjd}|dk(rtd|jz|jdr&|jd|dz|jz|_|S|jd|dz|_|S)zSet parent collection with location for created resources. Args: ref: reference to the membership object. args: command line arguments. request: API request to be issued Returns: modified request /zInvalid parent collection: %slocationNz/-)parentrfind ValueErrorIsKnownAndSpecifiedr)rrrindexs rSetParentCollectionr+s ..  s #% b[ 4w~~E FF j)^^FU+c1DMMAGN .^^FU+d2GN .rc~|jdrMtj|rtj||_|Stj ||_|St jdd)zSet membership location for requested resource. Args: ref: reference to the membership object. args: command line arguments. request: API request to be issued Returns: modified request membership MEMBERSHIPz(membership is required for this command.)rr MembershipLocationSpecifiedMembershipResourceNamenameSearchMembershipResourcecalliope_exceptionsRequiredArgumentExceptionr s rSetMembershipLocationr'Cst  l+,,T255d;gl . 77=gl .  7 7@ BBrc`~tj|rtj|}ntj|}|jj }t j||}g}dx}}|jdr"|jd|jd}|tj jk7r3|jdr"|jd|jd}|jds"|jds|jdr|jd d j|}t j|||||||jd|jd|jdddd|jd  } | S) zExecute update membership request. Args: ref: API response from update membership call args: command line arguments. Returns: response N external_id externalId infra_typeinfrastructureType clear_labels update_labels remove_labelslabels,async) r)r+r-r.r/ issuer_url oidc_jwksapi_server_version async_flag)r r!r"r$calliope_command ReleaseTrackr GetMembershipGetValueappend calliope_baseGAjoinUpdateMembership) rrr# release_trackobj update_fieldsr)r+ update_maskresponses rExecuteUpdateMembershipRequestrE\sw **40  + +D 1D  - -d 3D''446-t]3#-!!+  ]]=!&-- .Km00333 9-.|,J ]]>"dmm'--8"'+  & &  ==0MM/2MM/2w' )( /rc|dvrdndj|}|r|dk(rdj|Sd|vrdj|Sd|vrd j|Stjd ) aGet the appropriate Connect Gateway endpoint. This function checks the environment endpoint overide configuration for Fleet and uses it to determine which Connect Gateway endpoint to use. The overridden Fleet value will look like https://autopush-gkehub.sandbox.googleapis.com/. When there is no override set, this command will return a Connect Gateway prod endpoint. When an override is set, an appropriate non-prod endpoint will be provided instead. For example, when the overridden value looks like https://autopush-gkehub.sandbox.googleapis.com/, the function will return autopush-connectgateway.sandbox.googleapis.com. Regional prefixes are supported via the location argument. For example, when the overridden value looks like https://autopush-gkehub.sandbox.googleapis.com/ and location is passed in as "us-west1", the function will return us-west1-autopush-connectgateway.sandbox.googleapis.com. Args: endpoint_override: The URL set as the API endpoint override for 'gkehub'. None if the override is not set. location: The location against which the command is supposed to run. This will be used to dynamically modify the service name to a location-specific value. If this is the value 'global' or None, a global service name is returned. Returns: The service name to use for this command invocation, optionally modified to target a specific location. Raises: UnknownApiEndpointOverrideError if the Fleet API endpoint override is not one of the standard values. )globalNz{}-zhttps://gkehub.googleapis.com/z{}connectgateway.googleapis.comzautopush-gkehubz0{}autopush-connectgateway.sandbox.googleapis.comzstaging-gkehubz/{}staging-connectgateway.sandbox.googleapis.comgkehub)formatr UnknownApiEndpointOverrideError)endpoint_overriderprefixs rGetConnectGatewayServiceNamerNsR--25<<3I&  > > - 3 3F ;;-- = D DV LL,, < C CF KK  0 0 ::rN)__doc__ __future__rrrgooglecloudsdk.callioperr<rr%*googlecloudsdk.command_lib.container.fleetrr 6googlecloudsdk.command_lib.container.fleet.membershipsr rrr'rErNrrrUs<,&'9E?@I  020f5;r