k5SrSSKJr SSKJr SSKJr SSKrSSKJrJrJ r J r J r SSK J r SSKrSSKJr SS KJr SS KJr SS KJr \R(rS rS rSrSS/rSrSrS\ R6S\R8R:4SjrS2S\ R6S\S\S\S\S\ \S\ \S\ \S\ \S\ \S\ \4S jjr S3S\ R6S\R8R:S\ \S\ \S\ \S\ \S\ \S\ \S\ \4S!jjr!S"\S#\\\4S$\ \4S%jr"S&\S'\S$\4S(jr#S&\S)\\\4S$\ \4S*jr$S\ R6S\R8R:4S+jr%S\ R6S\S\S,\&S-\ \4 S.jr'S\ R6S\R8R:S/\4S0jr(S\ R6S\R8R:4S1jr)g)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 referencecUR5R5R5RUR 5S9R 5$)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 V0n U(aXkS'U(aX{S'U (aXS'[RRU5n U (a^[R"U5X'U(a>[ X[ 5(a[R"U5XS'O=[ S5eO1U (a[R"U 5U S'OSU-n [ U 5eSU<S U<3nUR5R5R5RXU S 9R5$) 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@*!-~ +}+|$DDHH- 2 < ">#J  & '  U )3X >& ooy{{} vVjvIwy r c U(dU(aU(d[R"S5e0n /n U(aXYS'U RS5 U(aXiS'U RS5 UbU RS5 U(aXyS'US:XaU(aD[R"U5n XS'U R [ UR5U 55 O0U S'U(a-[R"U5U SS'U RS 5 GOUS :XaU(a[R"U5n XS 'U RS 5(a*U S RS 5(aU RS5 U RS5(a*U SRS 5(aU RS5 O0U S 'U(a-[R"U5U S S'U RS5 GOUS:XaqU(ah[R"U5n XS'U RS5(aU RS5 U RS5(aU RS5 GOUS:XaU(aD[R"U5nXS'U R [ UR5U55 O0U S'U(a-[R"U5U SS'U RS5 GOUS:XaQU(aD[R"U5nXS'U R [ UR5U55 O0U S'OUS:XaWU(aJ[R"U5nUU S'SU;aU RS 5 S!U;aU RS"5 Om0U S'OgU(a`[R"U5U S#'U R [S#U S#55 [U ;a[U ;aU R[5 UR5R5R5RUR!5S$R#U 5U S%9R%5$)&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 rUpdateConnectionres%Dz?  & &  *+!-~~& +}}%~&+| #.88D2 ..02E F "j-?-I-I .j\*./%)33J?n(   . / /N 5 K5 ;<   L ) )n\.J.N.N // 89j(:(D(D )j %)*'!+55jA,   0 1 156   < = =BC++#5#?#? #K $>! ..02L M %'j!2D2N2N 3j!,/34/) 2 < ? !%5 5;<j"4">">#J!/:o3NO + +0M$ ooy{{} u~~XXk*  wy r base_pathjson_propertiesreturncfUVs/sHnUS-[R"U5-PM! sn$s snf)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&.prefixrcU[;a*[R"[U5(aU$SU-S-$[R "U5$)a@Escapes 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@sP ~ <<0$ 7 7 Dj3   t $$r json_valuec[U[5(aU(dU/$/nUH?nUS-[X5-nURU5nUR [ XE55 MA U$)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 rrYrYYsd J % %Z 8O &d# 1& ??J^^D)N MM+JGH -r cUR5R5R5RUR 5S9R 5 g)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_tokencSU<SU<3nUR5R5R5RXTUS9R 5$)a<List 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 rListConnectionsrs@&*4X >& ooy{{} t6+tFwy r policyc[R"U[RRSS9 UR 5R 5R 5RUR5SU0S9R5$)asSets 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 c[R"U[RRSS9 UR 5R 5R 5RUR5S9R5$)a?Gets 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