k5dZddlmZddlmZddlmZddlZddlmZmZm Z m Z m Z ddl m Z ddlZddlmZdd lmZdd lmZdd lmZej(Zd Zd ZdZddgZdZdZde j6dej8j:fdZ d2de j6dedededede ede ede ede ede ede efd Z d3de j6dej8j:de ede ede ede ede ede ede efd!Z!d"ed#eeefd$e efd%Z"d&ed'ed$efd(Z#d&ed)eeefd$e efd*Z$de j6dej8j:fd+Z%de j6deded,e&d-e ef d.Z'de j6dej8j:d/efd0Z(de j6dej8j:fd1Z)y)4z+The BigQuery CLI connection client library.)absolute_import)division)print_functionN)AnyDictListMappingOptional) discovery) bq_api_utils)bq_error) bq_id_utils)bq_processor_utilsauthorization_code version_infoz [0-9A-Za-z_]+zconfiguration.parametersz'configuration.authentication.parametersz'configuration.authentication.profile_idzconfiguration.authenticationclient referencec|jjjj|j j S)zGets connection with the given connection reference. Arguments: client: the client used to make the request. reference: Connection to get. Returns: Connection object with the given id. name)projects locations connectionsgetpathexecuterrs (platform/bq/clients/client_connection.py GetConnectionr&s; ooy{{} s  s!wy  project_idlocationconnection_type propertiesconnection_credential display_name description connection_id kms_key_nameconnector_configurationc i} |r|| d<|r|| d<| r| | d<tjj|} | rTtj|| | <|ret | | t rtj|| | d<n6t d| rtj| | d<nd|z} t | d|d |}|jjjj||| jS) aCreate a connection with the given connection reference. Arguments: client: the client used to make the request. project_id: Project ID. location: Location of connection. connection_type: Type of connection, allowed values: ['CLOUD_SQL'] properties: Connection properties in JSON format. connection_credential: Connection credentials in JSON format. display_name: Friendly name for the connection. description: Description of the connection. connection_id: Optional connection ID. kms_key_name: Optional KMS key name. connector_configuration: Optional configuration for connector. Returns: Connection object that was created. friendlyNamer' kmsKeyName credentialz'The `properties` were not a dictionary. configurationzMconnection_type %s is unsupported or connector_configuration is not specified projects/ /locations/)parent connectionIdbody) rCONNECTION_TYPE_TO_PROPERTY_MAPr ParseJson isinstancer ValueErrorrrrcreater)rr!r"r#r$r%r&r'r(r)r* connection property_nameerrorr2s rCreateConnectionr=<s@*!-J~ +J}+J|$DDHH- 2 < ">#J  & '  U )3X >& ooy{{} vV-jvIwy r c |s|r|stjdi} g} |r|| d<| jd|r|| d<| jd|| jd|r|| d<|dk(r||rDtj|} | | d<| j t |j| ni| d<|rtj|| dd<| jd n|d k(r|rtj|} | | d <| jd r%| d jd r| jd| jdr+| djd r| jdni| d <|rtj|| d d<| jdn|dk(rd|rtj|} | | d<| jdr| jd| jdr| jdn|dk(r||rDtj|}|| d<| j t |j|ni| d<|r.tj|| dd<| jdn|dk(rL|rDtj|}|| d<| j t |j|ni| d<n|dk(rM|rEtj|}|| d<d|vr| jd d!|vru| jd"nci| d<n]|r[tj|| d#<| j td#| d#t| vrt| vr| jt|jjjj|j!d$j#| | %j%S)&aUpdate connection with the given connection reference. Arguments: client: the client used to make the request. reference: Connection to update connection_type: Type of connection, allowed values: ['CLOUD_SQL'] properties: Connection properties connection_credential: Connection credentials in JSON format. display_name: Friendly name for the connection description: Description of the connection kms_key_name: Optional KMS key name. connector_configuration: Optional configuration for connector Raises: bq_error.BigqueryClientError: The connection type is not defined when updating connection_credential or properties. Returns: Connection object that was created. zMconnection_type is required when updating connection_credential or propertiesr,r'r)r- CLOUD_SQLcloudSqlr.zcloudSql.credentialAWSawscrossAccountRole iamRoleIdzaws.crossAccountRole.iamRoleId accessRolezaws.access_role.iam_role_idzaws.credentialAzureazurecustomerTenantIdzazure.customer_tenant_idfederatedApplicationClientIdz%azure.federated_application_client_idSQL_DATA_SOURCE sqlDataSourcezsqlDataSource.credential CLOUD_SPANNER cloudSpannerSPARKsparksparkHistoryServerConfigz!spark.spark_history_server_configmetastoreServiceConfigzspark.metastore_service_configr/,)r updateMaskr4)r BigqueryClientErrorappendrr6extend_GetUpdateMasklowerr_GetUpdateMaskRecursively_AUTH_PROFILE_ID_PATH _AUTH_PATHrrrpatchrjoinr)rrr#r$r%r&r'r)r*r: update_maskcloudsql_propertiesaws_propertiesazure_propertiessql_data_source_propertiescloudspanner_propertiesspark_propertiess rUpdateConnectionresDz?  & &  *+!-J~~& +J}}%~&+J| #.88D2j ..02E F "j-?-I-I .j\*./%)33J?n(j   . /N 5 K5 ;<   L )n\.J.N.N / 89j(:(D(D )j %)*'!+55jA,j   0 156   < =BC++#5#?#? #K $>j! ..02L M %'j!2D2N2N 3j!,/34/) 2 < ? !%5 5;<j"4">">#J!/:o3NO + +0M$ ooy{{} u~~XXk*  wy r base_pathjson_propertiesreturnc^|Dcgc]}|dztj|zc}Scc}w)zCreates an update mask from json_properties. Arguments: base_path: 'cloud_sql' json_properties: { 'host': ... , 'instanceId': ... } Returns: list of paths in snake case: mask = ['cloud_sql.host', 'cloud_sql.instance_id'] .) inflection underscore)rfrg json_propertys rrWrW-s=+ *-#o --m<<*  s"*prefixrc|tvr$tjt|r|Sd|zdzSt j |S)aREscapes name if it points to a map key or converts it to snake case. If name points to a map key: 1. Do not change the name. 2. Escape name with backticks if it is not a valid proto field name. Args: prefix: field mask prefix to check if name points to a map key. name: name of the field. Returns: escaped name `)_MAP_KEY_PATHSre fullmatch_VALID_FIELD_NAME_REGEXPrkrl)rnrs r_EscapeIfRequiredru@sM ~ <<0$ 7 Dj3   t $$r json_valuect|tr|s|gSg}|D]@}|dzt||z}|j|}|j t ||B|S)zRecursively traverses json_value and returns a list of update mask paths. Args: prefix: current prefix of the json value. json_value: value to traverse. Returns: a field mask containing all the set paths in the json value. rj)r7dictrurrVrY)rnrvresultr new_prefixnew_json_values rrYrYYse J %Z 8O &d# 1&$ ??J^^D)N MM+JGH -r c|jjjj|j j y)zDelete a connection with the given connection reference. Arguments: client: the client used to make the request. reference: Connection to delete. rN)rrrdeleterrrs rDeleteConnectionr~qsA //++-44 >> 5 GIr max_results page_tokencd|d|}|jjjj|||j S)aNList connections in the project and location for the given reference. Arguments: client: the client used to make the request. project_id: Project ID. location: Location. max_results: Number of results to show. page_token: Token to retrieve the next page of results. Returns: List of connection objects r0r1)r2 pageTokenpageSize)rrrlistr)rr!r"rrr2s rListConnectionsrsB&*4X >& ooy{{} t6Z+tFwy r policyctj|tjjd|j j j j|jd|ijS)aSets IAM policy for the given connection resource. Arguments: client: the client used to make the request. reference: the ConnectionReference for the connection resource. policy: The policy string in JSON format. Returns: The updated IAM policy attached to the given connection resource. Raises: BigqueryTypeError: if reference is not a ConnectionReference. SetConnectionIAMPolicymethodr)resourcer4) r typecheckApiClientHelperConnectionReferencerrr setIamPolicyrr)rrrs rrrsg$!!55 %  ooy{{}|Y^^-Xv4F|Gwy r ctj|tjjd|j j j j|jjS)aOGets IAM policy for the given connection resource. Arguments: client: the client used to make the request. reference: the ConnectionReference for the connection resource. Returns: The IAM policy attached to the given connection resource. Raises: BigqueryTypeError: if reference is not a ConnectionReference. GetConnectionIAMPolicyr)r) rrrrrrr getIamPolicyrrrs rrrs` !!55 %  ooy{{}|Y^^-|.wy r )NNNNNN)NNNNNNN)*__doc__ __future__rrrrrtypingrrrr r googleapiclientr rkutilsr r rrServiceAUTHORIZATION_CODE VERSION_INFOrtrqrZr[Resourcerrrstrr=rerWrurYr~intrrrr rrs1&% 55%$   * ,- B +    **>>8,0"&!%#'"&-1I   III I  I $C= I3-I#IC=I3-I&c]Ib $+/"&!%"&-1b   b**>>b b  b$C=b3-b#b3-b&c]bJ%)#s(^ #Y&%c%%%2 !#s(^ #Y0     **>>       :   **>> @   **>>r