o ;@sFdZddlmZddlmZddlmZddlmZddlmZddl m Z ddl m Z dd l m Z dd lZdd lmZdd lmZd dZddZddZddZddZddZddZ  d4ddZejjfddZd ejjdd d fd d!Zd"d#Zd5d$d%Zd&d'Z d(d)Z!d*d+Z"d5d,d-Z#d.d/Z$d0d1Z%d2d3Z&d S)6zLibrary for obtaining API clients and messages. This should only be called by api_lib.util.apis, core.resources, gcloud meta commands, and module tests. )absolute_import)division)unicode_literals) apis_util)resource) properties) transport)apis_mapN)urljoin)urlparsecCstj|||fSN)r_API_NAME_ALIASESget)api_namerF/tmp/google-cloud-sdk/lib/googlecloudsdk/api_lib/util/apis_internal.py_GetApiNameAndAlias#srcCs@t|\}}tj|i}t|D] \}}|jr|SqdSr )rr MAPrsix iteritemsdefault_version)r_api_versverapi_defrrr_GetDefaultVersion(s rcCsttjS)z6Returns list of avaiblable apis, ignoring the version.)sortedr rkeysrrrr _GetApiNames1srcCs8t|\}}tj|d}|durt|t|S)zReturn available versions for given api. Args: api_name: str, The API name (or the command surface name, if different). Raises: apis_util.UnknownAPIError: If api_name does not exist in the APIs map. Returns: list, of version names. N)rr rrrUnknownAPIErrorlistr)rr version_maprrr _GetVersions6s   r"cCst|\}}|tjvrt|tjj}| d ||}|s'| |d}|p*|}tj|}|dus8||vr>t ||||}|S)aReturns the APIDef for the specified API and version. Args: api_name: str, The API name (or the command surface name, if different). api_version: str, The version of the API. Raises: apis_util.UnknownAPIError: If api_name does not exist in the APIs map. apis_util.UnknownVersionError: If api_version does not exist for given api_name in the APIs map. Returns: APIDef, The APIDef for the specified API and version. z{}/{}N) rr rrrrVALUESapi_client_overrides AllValuesrformatUnknownVersionError)r api_versionapi_name_aliasversion_overridesversion_override api_versionsrrrr GetApiDefIs       r-cCst||}t|S)aReturns the client class for the API specified in the args. Args: api_name: str, The API name (or the command surface name, if different). api_version: str, The version of the API. Returns: base_api.BaseApiClient, Client class for the specified API. )r-_GetClientClassFromDefrr(rrrr_GetClientClassos r0cCs0|jj}|dd\}}t||gd}t||S)zReturns the apitools client class for the API definition specified in args. Args: api_def: apis_map.APIDef, The definition of the API. Returns: base_api.BaseApiClient, Client class for the specified API. .fromlist)apitoolsclient_full_classpathrsplit __import__getattr)rr6 module_pathclient_class_name module_objrrrr.}s  r.Fc Cs|r |dusJn|durddlm}|jtj|r|ndd}t||}|t|||d|d}|dur6||_tj j j } | rL| d| d } d |j| <|S) aGReturns an instance of the API client specified in the args. Args: api_name: str, The API name (or the command surface name, if different). api_version: str, The version of the API. no_http: bool, True to not create an http object for this client. http_client: bring your own http client to use. Incompatible with no_http=True. check_response_func: error handling callback to give to apitools. http_timeout_sec: int, seconds of http timeout to set, defaults if None. Returns: base_api.BaseApiClient, An instance of the specified API client. Nr) transportsunset)response_encodingtimeoutF)urlget_credentialshttpkeyzX-Google-Project-Overrideapikey)googlecloudsdk.core.credentialsr=GetApitoolsTransportrENCODINGr0_GetEffectiveApiEndpointcheck_response_funcrr#coreapi_keyGetAddGlobalParamadditional_http_headers) rr(no_http http_clientrJhttp_timeout_secr= client_classclient_instancerLheaderrrr_GetClientInstances,      rVcCsft||}|tjjkr|jj}n|tjjkr|jj}n|jj}| dd\}}t ||gd}t ||S)a*Returns the GAPIC client class for the API def specified by the args. Args: api_name: str, The API name (or the command surface name, if different). api_version: str, The version of the API. transport_choice: apis_util.GapicTransport, The transport to be used by the client. r1r2r3) r-rGapicTransport GRPC_ASYNCIOgapicasync_client_full_classpathRESTrest_client_full_classpathr6r7r8r9)rr(transport_choicerr6r:r;r<rrr_GetGapicClientClasss     r^c s6fdd}t||d} | ||t||||dS)aReturns an instance of the GAPIC API client specified in the args. For apitools API clients, the API endpoint override is something like http://compute.googleapis.com/v1/. For GAPIC API clients, the DEFAULT_ENDPOINT is something like compute.googleapis.com. To use the same endpoint override property for both, we use the netloc of the API endpoint override. Args: api_name: str, The API name (or the command surface name, if different). api_version: str, The version of the API. credentials: google.auth.credentials.Credentials, the credentials to use. address_override_func: function, function to call to override the client host. It takes a single argument which is the original host. transport_choice: apis_util.GapicTransport, The transport to be used by the client. attempt_direct_path: bool, True if we want to attempt direct path gRPC where possible. redact_request_body_reason: str, the reason why the request body must be redacted if --log-http is used. If None, the body is not redacted. custom_interceptors: list[grpc interceptor], a list of custom interceptors to add to the channel. Returns: An instance of the specified GAPIC API client. cs^z tjj}Wn tjyd}Ynw|rt|j}r%|}|dur+|St|Sr ) rr#api_endpoint_overridesPropertyrMNoSuchPropertyErrorr netlocUniversifyAddress)addressendpoint_overrideaddress_override_funcrrrAddressOverrides  z0_GetGapicClientInstance..AddressOverride)r])rg mtls_enabledattempt_direct_pathredact_request_body_reasoncustom_interceptors)r^ _MtlsEnabled) rr( credentialsrgr]rjrkrlrhrSrrfr_GetGapicClientInstances#rocCs8tjjj}|}|dur|j|kr||j|d}|S)z2Update a URL based on the current universe domain.Nr2)rr#rKuniverse_domainrMdefaultreplace)rduniverse_domain_propertyrprrrrcs  rccCs:t||}|jr|pt||}n|pt||}|jp|jS)zReturns mtls endpoint.)r-r5r0r^mtls_endpoint_override MTLS_BASE_URL)rr(rSrrrr_GetMtlsEndpoint!s  rvcCs4tjjjr dStjjjsdSt||}|jS)aChecks if the API of the given version should use mTLS. If context_aware/always_use_mtls_endpoint is True, then mTLS will always be used. If context_aware/use_client_certificate is True, then mTLS will be used only if the API version is in the mTLS allowlist. gcloud maintains a client-side allowlist for the mTLS feature (go/gcloud-rollout-mtls). Args: api_name: str, The API name. api_version: str, The version of the API. Returns: True if the given service and version is in the mTLS allowlist. TF)rr# context_awarealways_use_mtls_endpointGetBooluse_client_certificater- enable_mtlsr/rrrrm+s  rmcCs@t|}t|}|jdks|jdkr|Std|j|j|jS)zFConstructs a normalized endpoint URI depending on the client base_url./z{}://{})r pathr r&schemerb)rebase_urlurl_baseurl_endpoint_overriderrr_BuildEndpointOverrideHsrcCsvt||jr t||}nt||}t|dr|j}t |Sz t||j}Wt |Sty:d||}Yt |Sw)zReturns base url for given api.BASE_URLzhttps://{}.googleapis.com/{}) r-r5r0r^hasattrr_GetResourceModuleAttributeErrorr&rc)rr(rSclient_base_urlrrr_GetBaseUrlFromApiSs     rcCsz tjj|}Wn tjyd}Ynwt||}|jr(|p&t||}n|p.t ||}t ||}|r=t ||}|St ||rLt t|||}|S|}|S)z)Returns effective endpoint for given api.N)rr#r_r`rMrar-r5r0r^rrrmrcrv)rr(rSrerrrdrrrrIfs0     rIcCst||}t|jjdgdS)aReturns the messages module for the API specified in the args. Args: api_name: str, The API name (or the command surface name, if different). api_version: str, The version of the API. Returns: Module containing the definitions of messages for the specified API. somethingr3)r-r8r5messages_full_modulepathr/rrr_GetMessagesModules  rcCsDt||}|jrt|jjdddgdSt|jjdddgdS)z/Imports and returns given api resources module.r1 resourcesrr3)r-r5r8 class_pathrYr/rrrrs rc cs`zt||}Wn tyYdSw|jD]}t|||j|j|j|j|j |j |j VqdS)z)Yields all collections for for given api.N) r ImportError Collections resource_utilCollectionInforDOCS_URLcollection_namer} flat_pathsparamsenable_uri_parsing)rr(resources_module collectionrrr_GetApiCollectionss&  r)FNNNr )'__doc__ __future__rrrgooglecloudsdk.api_lib.utilrrrgooglecloudsdk.corerr%googlecloudsdk.generated_clients.apisr rsix.moves.urllib.parser r rrrr"r-r0r.rVrWGRPCr^rorcrvrmrrrIrrrrrrrsR           & 2  B