LdZddlZddlZddlZddlmZmZmZmZm Z ddl m Z ddl m Z ddl Z ddlZddlZddlZddlZddlZddlmZddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZe j>Zde ej@fdZ!Gdde"Z#GddZ$y)z,BigQuery client that exists for some reason.N)AnyDictListOptionalType)app)flags)main_credential_loader)bigquery_client)bigquery_client_extended) wait_printer)utils) bq_api_utils)bq_gcloud_utils) bq_loggingreturnctjjrtjStj jrtj StjS)zGReturns the default wait_printer_factory to use while waiting for jobs.)bq_flagsQUIETvaluer QuietWaitPrinterHEADLESSTransitionWaitPrinterVerboseWaitPrinter(platform/bq/frontend/bq_cached_client.py_GetWaitPrinterFactoryFromFlagsr!sB ^^  ( ((   - --  ( ((rc eZdZdZiZed dedeee ffdZ e d de jde jfdZ e d dede jdejfdZedefd Zedejfd Zy)Clientz@Class caching bigquery_client.BigqueryClient based on arguments.config_loggingrc Jdtffd }tj|r-tjt j jtjtt jjr\tjt jjs/tj jst#j$dt j&jr`t j&jt(_t-t(dr-t j&jt(j._i}d}|D] }||||<t#j0dtj2t#j4d|t7|d <t j8jr?t;t j8j5}|j=|d <d d d t j>jd nt j>j|d <t j@jrt j@j|d<t#j0d||S#1swYxYw)zCollect and combine FLAGS and kwds to create BQ Client. Args: config_logging: if True, set python logging according to --apilog. **kwds: keyword arguments for creating BigqueryClient. namec6|vr|Stt|SN)getattrFLAGS)r#kwdss r KwdsOrFlagsz(Client._CollectArgs..KwdsOrFlags8s4>h 9 %&.&B&B&H&H#K K%d+k$ LL0(2I2I MM-{;*I*KK&'$$ ''-- .!,-FFH () /  , , 2 2 :  . . 4 4*+ $$**,4,I,I,O,Ok() LL,k:  / .s 5JJ"N credentialsc|tjdnmtjjr*tjdt j }n)tjdtj }|J|S)zA function to lookup the credentials to use for this BQ CLI invocation. Args: credentials: bypass the credential lookup and use these instead. Returns: The credentials ot use for this BQ CLI invocation. zCredentials passed in directlyz$Credentials loaded using Google Authz%Credentials loaded using oauth2client)rFrLrDrErr GetCredentialsFromFlagscredential_loader)rXs rGetCredentialszClient.GetCredentialswsn ll34  & & , , ll9:*BBDk ll:;%==?k  "" " rc tjd|tj|fi|}tj }|ddtj |i|S)aKBuild a new BigqueryClient configured from kwds and FLAGS. Args: config_logging: if True, set python logging according to --apilog. credentials: bypass the credential lookup and use these instead. **kwds: keyword arguments for creating BigqueryClient. Returns: A BigqueryClient from the kwds. z4Collecting args before creating a BigqueryClient: %srXr)rFrNr rWFactoryGetBigqueryClientFactoryr\)r!rXr(rTbigquery_client_factorys rCreatez Client.Creates_" MMH$O%%n==K%>>@ " ))+6 :E rc tjd|tjdi|}dj |t j jtjjS)Nz In Client._GetClientCacheKey: %szlclient_args={client_args},service_account_credential_file={service_account_credential_file},apilog={apilog},)rTservice_account_credential_fileapilogr) rFrNr rWformatrDSERVICE_ACCOUNT_CREDENTIAL_FILErrr?)clsr(rTs r_GetClientCacheKeyzClient._GetClientCacheKeys\ MM4d;%%--K !6#,9,Y,Y,_,_??((" rcvtj}||jvrtjd|n7 tj |j|<tjd|j|S#t $r/}tjdtj|d}~wwxYw)z/Return a BigqueryClient initialized from flags.zRUsing a cached client with previous auth and discovery docs from the cache_key: %sz"Successfully created a new client.zFailed to create a new client.N) r rh client_cacherFrLra ValueErrorr UsageError)rg cache_keyes rGetz Client.Gets))+IC$$$ ll    &,mmo# 9:   I &&   56nnQ s6B B8 *B33B8)Tr%)TN)__name__ __module__ __qualname____doc__rj staticmethodboolrr;rrWr 'LegacyAndGoogleAuthCredentialsUnionTyper\r BigqueryClientExtendedra classmethodrhrorrrr r *sH,E4E4S>EENMQ"JJ>>,!MQ"JJ 66 2   '*AA''rr cXeZdZdZdZGddZedeejfdZ y)r^z7Class encapsulating factory creation of BigqueryClient.Nc>eZdZdZdZedejfdZy)Factory.ClientTablePrinterz5Class encapsulating factory creation of TablePrinter.Nrcd|jtj|_|jSr%)_TABLE_PRINTERbq_frontend_utils TablePrinterrgs rGetTablePrinterz*Factory.ClientTablePrinter.GetTablePrinters+   #.;;=   r) rprqrrrsr}rxr~rrrrrClientTablePrinterr{s*?N 1 > >  rrrc\|jtj|_|jSr%)_BIGQUERY_CLIENT_FACTORYr rwrs rr_z Factory.GetBigqueryClientFactorys/ ##+ " 9 9 "  ' ''r) rprqrrrsrrrxrr rwr_rrrr^r^s=?!   ( $ ; ;<((rr^)%rsrFsystextwraptypingrrrrrabslrr rI termcolorrDrr<r[authr clientsr r r frontendrr~rrrr' WaitPrinterrobjectr r^rrrrs2 22'#, /!   )l.F.F)G)_'V_'D((r