&SrSSKrSSKrSSKrSSKJrJrJrJr SSK r SSK J r SSK J r SSK Jr \ RrSrSrS \S \4S jr\R(\SS S S 4S \S\S\\S\S\S\S \4SjjrS\S \4SjrS \S\"SS\ R0\\4S\ R0\\4S\ R0\\4S\ R0\4S\ R0\4S\ R0\4S\ R0\4/5S \4SjrS,S \S\\S\"SS!\ R0\\4/5S"\\S#\\S \4 S$jjrS-S%\S&\S \4S'jjrS(\\\4S \\\44S)jrS\\S \4S*jrS\\S \4S+jrg).zAPI utils for the BQ CLI.N)Dict NamedTupleOptionalUnion)flags) bq_consts)bq_errorz_https?://(www|(staging-www|test-bigquery).sandbox|bigquery-sq|bigquery).(google|googleapis).comzgoogleapis.comservicereturnc6U[RLagU[RLagU[RLagU[RLagU[R LagU[R LagU[RLag[SU35e)z4Returns the service name for the given service enum.bigquerybigqueryconnectionbigqueryreservationbigquerydatatransfer analyticshubbiglakezUnsupported service: ) ServiceBIGQUERY CONNECTIONS RESERVATIONSDTS ANALYTICS_HUBBIGLAKEBQ_IAM ValueError)r s !platform/bq/utils/bq_api_utils.py_get_service_name_from_enumrs    '%%% '&&& '++ !'''' '//! '..  ,WI6 77F service_enumuniverse_domainregionis_mtlsis_repis_lepc x[R"SUUUU(aSOSU(aSOSU(aSOS5 [U5nU(aU(aU(a USUSU3$U(dU(aU(a USUSU3$U(aU(dUSU3$U(dU(aU(a USUSU3$USU3$) z*Returns the TPC service endpoint hostname.zgBuilding a root URL for the %s service in the "%s" universe for region "%s" %s mTLS, %s REP, and %s LEPwithwithout.z .rep.mtls.z.rep.z.mtls.-)logginginfor)rr r!r"r#r$r s r"_get_tpc_service_endpoint_hostnamer,,s ,,* fYfIfI  ( 5'  FYaxz/): ;; VYaxu_$5 66 VYf_- .. VXQwiq 1 22 9Ao& ''rurlc<URS5(dUS-$U$)N/)endswithr-s radd_trailing_slash_if_missingr2Vs c   9 *rinputted_flags InputtedFlagsAPIUNIVERSE_DOMAINLOCATIONUSE_LEPUSE_REPUSE_REGIONAL_ENDPOINTSMTLSc URRURRURR/R S5nUS:a[ R "S5eUS:Xa1URR(d[ R "S5eURR(aJ[R"SURR5 [URR5$US:XaURR(aSnOURRnURR(a[R"S5 SnOURRnURR(aURRnO[n[UUUUR RURRUS 9n[["R$R'["R$R)S US S S S 955n[R"S U5 U$)a3Takes BQ CLI flags to build a root URL to make requests to. If the `api` flag is set, and is a http/https URL then it will be used otherwise the result is built up from the different options for a TPC service endpoint. Args: service: The service that this request will be made to. Usually the API that is being hit. inputted_flags: The flags set, usually straight from bq_flags. Returns: The root URL to be used for BQ requests. This is built from the service being targeted and a number of flags as arguments. It's intended to be used both for building the URL to request the discovery doc from, and to override the rootUrl and servicePath values of the discovery doc when they're incorrect. It always ends with a trailing slash. Raises: BigqueryClientError: If the flags are used incorrectly. TzLOnly one of use_lep, use_rep or use_regional_endpoints can be used at a timezVA region is needed when the use_lep, use_rep or use_regional_endpoints flags are used.zCLooking for a root URL and an `api` value was found, using that: %srNzHBuilding a root URL and `use_regional_endpoints` is present, forcing LEP)rr r!r"r#r$https)schemenetlocpathqueryfragmentzFinal root URL built as: %s)r8valuer9r:countr BigqueryClientErrorr7r5presentr*r+r2r6 _GDU_DOMAINr,r;urllibparse urlunsplit SplitResult)r r3)number_of_flags_requesting_a_regional_apir!r$r hostnameroot_urls rget_tpc_root_url_from_flagsrQ\sL""""++11/ E$K , /2  & &   014%%++  & &      LLM   )););)A)A BB014  ! ! ' ' F  $ $ * *F **00 LL F  # # ) )F##))$44::O!O /% !!''  # # ) )  (+ ll ,, " "XBb2 # ( ,,,h7 /r discovery_urlBIGQUERY_DISCOVERY_API_KEY_FLAGkeylabelsc~U(a"SU3U;a[R"SU5 U$SU;a[R"S5 U$U(d-URRn[R"SU5 U(a:SU;aSnOSnXSU3- nU(aUSU3- n[R"SU5 U$) zAdds an API key to the URL.zkey=zHAPI key %s has already been added, presumably from --discovery_url_extrazfAn API key already exists in the URL, presumably from --discovery_url_extra, so not adding any new keyzrNo API key has been set, using value from the `bigquery_discovery_api_key` flag targeting the universe_domain (%s)?&z&labels=z#Discovery URL has been updated (%s))r*r+rSrE)rRr r3rTrU delimiters radd_api_key_to_discovery_urlrZs$ tC5\] * LL !   } LL <   8 8 > >C LL     mii{$se,,M &**m LL6 F rrP api_versionc[RRU5n[RRSU05nUR SUS9n[RR U5$)z.Returns the discovery doc URL from a root URL.versionz/$discovery/rest)rBrC)rJrKurlsplit urlencode_replacerL)rPr[partsrCs rget_discovery_url_from_root_urlrb sY ,,   )% ,, )[!9 :% ..0. >%   ''rdiscovery_documentc[U[5(a[R"U5$[U[5(a%[R"UR S55$[ S[U535e)zTakes a downloaded discovery document and parses it. Args: discovery_document: The discovery doc to parse. Returns: The parsed api doc. zutf-8z%Unsupported discovery document type: ) isinstancestrjsonloadsbytesdecodertype)rcs rparse_discovery_docrlsi"C(( ::( ))$e,, ::(//8 99-d3E.F-GH rc$U(dgU[:H$)z+Returns whether the universe domain is GDU.F)rI)r s ris_gdu_universern,s   K ''rcU(dg[U;=(d+ [R"[5R U5SL$)zReturns whether the url is GDU.FN)rIrecompileSTATIC_HTTP_ENDPOINT_RE_PREFIXmatchr1s r is_gdu_urlrt3s7    jj/066s;4Gr)NN)v2) __doc__rgr*rptypingrrrrrJabslrutilsrr rrrrIrfrrboolr,r2 FlagHolderrQrZrbrirlrnrtrrr}s~  44      88S8*$,,&  '('('( SM'( '(  '(  '( '(T s s x x E$$Xc]3 4  0 0# ? @ ))(3-8 9 ((. / ((. / %u'7'7'= > U%%d+ ,  x xN 22c]22  #/  2 #2 SM2 2l'+(( #((c5j) #s(^((Xc](t(HSMdr