. dZddlZddlZddlZddlZddlZddlZddlZddlm Z m Z m Z m Z m Z ddlmZddlmZddlmZddlmZddlZddlZddlZddlmZej8Zd Zd Zd Zd Z d Z!de"fdZ#e#Z$de%fdZ&e&Z'de e"fdZ(de"ddfdZ)d3dZ* d4de"de e"de e"e"ffdZ+de de"de e"e"ffdZ,de e"ddfdZ-de e"de e"de e"fdZ.de e"de e"de e"fdZ/de"de"d e%d!d"de e"f d#Z0de"fd$Z1de"fd%Z2 d5d&e3d'e d(fd)Z4de e"fd*Z5de e"fd+Z6d,e"de e"e"ffd-Z7d.e"de e"fd/Z8de"fd0Z9d1e"de%fd2Z:y)6z7A library of functions to handle bq flags consistently.N)DictListLiteralOptionalTextIO)app)flags)version) credentials) stringutilz%https://www.googleapis.com/auth/drivez(https://www.googleapis.com/auth/bigqueryz.https://www.googleapis.com/auth/cloud-platformz/https://www.googleapis.com/auth/accounts.reauthVERSIONreturnct} tj|t}sd}t j |j}d|vsJd|S#t$rY?wxYw)zDReturns content of VERSION file found in same dir as the cli binary.zunknown-version z0New lines are not allowed in the version string.)__name__pkgutilget_data_VERSION_FILENAMEFileNotFoundErrorr ensure_strstrip)root version_strs platform/bq/bq_utils.py _GetVersionr#st $ ""4):;K #K%%k288:+ +878    sA A"!A"c,tjdS)z4Returns true if the current binary is targeting TPC.ztpc-)VERSION_NUMBER startswithr _IsTpcBinaryr!7s  " "6 **r ctdjxrtjxs1tjj dxstjS)zReturn the name of the bigqueryrc file to use. In order, we look for a flag the user specified, an environment variable, and finally the default value for the flag. Returns: bigqueryrc filename as a string. bigqueryrc BIGQUERYRC)FLAGSpresentr#osenvirongetrr rGetBigqueryRcFilenamer*?sE \""7u'7'7  %   r flagcZ||j|t||t||yN)parsesetattrgetattr) flag_valuesr+values r UpdateFlagr3Os( d%  +tW[$78r c$tdty)z7Updates FLAGS with values found in the bigqueryrc file.N)ProcessBigqueryrcSectionr%rr rProcessBigqueryrcr6Vs 4'r filename section_nameci}tjj|s|S t|5}t ||}ddd|S#1swY|SxYw#t $rY|SwxYw)a Read a configuration file section returned as a dictionary. Args: filename: The filename of the configuration file. section_name: if None, read the global flag settings. Returns: A dictionary of flag names and values from that section of the file. N)r'pathexistsopen_ProcessSingleConfigSectionIOError)r7r8 dictionaryrcfiles r_ProcessConfigSectionrA[sm*  !  h6.v|Dj        s- A AAA AA A! A!fileci}| }|D]}|jjdr8|jjdr|j dd}||k(}Z|s]|jjds|j s|j d\}}}|sd}|j }|j }|jdr|dd }|jdr|||<|S) zRead a configuration file section returned as a dictionary. Args: file: The opened configuration file object. section_name: Name of the section to read. Returns: A dictionary of flag names and values from that section of the file. []#=true-N)lstriprrstripendswithr partition) rBr8r? in_sectionline next_sectionr+ equalsignr2s rr=r=ts**d {{}$)?)?)DZZ\!B'l</j   ! !# &djjl!^^C0D)U e :: !C '*C CCr accountc2tjd|duS)znReturns whether the account may be a service account based on the user-created or system-created account name.z!^.+@(.+)(\.gserviceaccount\.com)$N)re fullmatch)rs rIsServiceAccountrs :G DD PPr )rNr-)r);__doc__rr'rrnrrrtypingrrrrrabslrr google.authr ry google.oauth2r google_oauth2rrrpyglibr r%rrrrrstrrrboolr! IS_TPC_BINARYr*r3r6rAr=r5r`r_rjrqrobjectrrrrrrrrr rrsb=  8866  7 <HA S"+d+   x}  9#99( 26!)# #s(^2    #  #s(^ FK8C=K$KB sm !# c] sm!#c](    -  c] 3/s/fBH !()=!>6$s)!tCy! CDcN>3498 Dc DQcQdQr