dZddlZddlZddlZddlmZddlmZddlZddl Z ddl Z ddl Z ddl m Z ddlmZddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZGddej0Zy)zThe BigQuery init CLI command.N)Optional)flags)client_project)bigquery_command)bq_cached_client)utils) utils_flags)utils_formatting) bq_id_utils) bq_logging)bq_processor_utilscpeZdZdZdedej ffd ZdefdZ de e fdZ de e fdZ xZS) Initz7Create a .bigqueryrc file and set up OAuth credentials.namefvcntt| ||d|_t j ddd|y)NFdelete_credentialszTIf specified, the credentials file associated with this .bigqueryrc file is deleted.) flag_values)superr__init__surface_in_shellrDEFINE_boolean)selfrr __class__s $platform/bq/frontend/command_init.pyrz Init.__init__!s8 $tR(!D    returncy)z/Init never needs to call itself before running.F)rs r _NeedsInitzInit._NeedsInit,s rctjtjjxstj j}t jj|std|dy dtjd|dk7rtd|dyt j|y #t$r}td |d |Yd }~y d }~wwxYw) z$Deletes this user's credential file.zCredential file z does not exist.ryzDelete credential file ? (y/N) z NOT deleting , exiting.Error removing : N)bq_utilsProcessBigqueryrc bq_auth_flagsSERVICE_ACCOUNT_CREDENTIAL_FILEvalueCREDENTIAL_FILEospathexistsprintfrontend_utilsPromptYNremoveOSError)rfilenamees rDeleteCredentialszInit.DeleteCredentials0s  55;; /  ( ( . .  77>>( # H >?   ''19 ;  X78ii  x 34 s;+B='B== C!CC!cZ d}tjdjtj|dddgt j tjj}tjtjtjj|jr|j!Stj"}t%t%dt%d t%t&j(j+|rt%d t%d |d t%d t%dt%dt-j.d|dk7rt%d|dyt% t'j0|t%dt%dt%t4j6j9}dt:j<ji}|rt%d|zt%||d<nIt?j@|jBd}t%d t%|s-t%d!t%d"t%d#t%nt%d$tEjF}|jId%tKjL|tNjPjRtU|D]A\} } tKjV| } | jYd%| dzi|j[| C|j]t_|dk(rAtajb|d} t%d&| d't%| jd|d<nt%d(t%d)t%d*t%d} tg| thset-jjd+t_|d,} | r dti| cxkrt_|krnnti| xsd} tg| thset%| r*tajb|| dz } | jd|d< to|d-5}|jqD]\}}t%|d.||/ dddt%d1t%y#t2$r}t%d|d|Yd}~yd}~wwxYw#tl$rYwxYw#1swYUxYw#tr$r}t%d0|d|Yd}~yd}~wwxYw)2z3Authenticate and create a default .bigqueryrc file.zBQ CLI will soon require all users to log in using `gcloud auth login`. `bq init` will no longer handle authentication after January 1, 2024.  P)widthredbold)colorattrsfilez;Welcome to BigQuery! This script will walk you through the zAre you sure you want to continue and overwrite your existing zconfiguration?r"z Overwrite r#zNOT overwriting r$rr%r&Nr'z9First, we need to set up your credentials if they do not zalready exist.credential_filez%Setting project_id %s as the default. project_idi) apiclient max_resultszCCredential creation complete. Now we will select a default project.z.No projects found for this user. Please go to z# https://console.cloud.google.com/zand create a project.zList of projects:#z Found only one project, setting z as the default.z6Found multiple projects. Please enter a selection for z3which should be the default, or leave blank to not zset a default.zEnter a selection (1 - z): wz = )rAzError writing z:BigQuery configuration complete! Type "bq" to get started.): termcolorcprintjointextwrapwrapsysstdoutbq_flags PROJECT_IDr,r(r)r ConfigureLoggingAPILOGrr8GetBigqueryRcFilenamer1r.r/r0r2r3r4r5rClientGetr*r-r list_projectsrEr get_formatter_from_flags AddColumnr configure_formatterr ApiClientHelperProjectReference enumerateformat_project_infoupdateAddDictPrintlenr ConstructObjectReference projectId isinstanceintPromptWithDefault ValueErroropenitemsIOError)rmessageproject_id_flag bigqueryrcr7cliententriesprojects formatterindexprojectresultproject_referenceresponsercfileflagr,s r RunWithArgszInit.RunWithArgsEs1 3  (--r23h ZZ ))//O   5 56   # # %%//1J G GH HI G ww~~j!  z KL LM  g ''*(NO O <= g *  EF  G  $ $ ( ( *F -"?"?"E"EFG 3o EF g-gl--$$$h O g  >? 34 %&  !"88: C ,, {22CC (1NE7#77@& --eai( )   F #2  x=A 0IIqk  "%  '"3"="=',  H I E F  ! '(x-%7725h-AH c(m!Ds8}!Dx}1- x- '  2 K KA&! %6$?$?GL !  C F"==?KD% T5) 7+ ! FG G o Q78H   !   j! 45 sZS51S+. T:+S:&T S( S##S(+ S76S7:T?T T*T%%T*)__name__ __module__ __qualname____doc__strr FlagValuesrboolr rrfr8rz __classcell__)rs@rrrsM? 3 E$4$4 $#* 8C= rr)r~r.rNrLtypingrabslrrIr*rPr(clientsrfrontendrrrr2r r r r r BigqueryCmdrrrrrsS$ "%%, %$f  ' 'f r