2XSrSSKrSSKrSSKrSSKJr SSKJr SSKJ r SSKJ r SSKJ r SSK J r SS K Jr SS KJr SSKJr SS KJr SS KJr SS KJr SSKJr SSKJr SSKJr SSKJr SSKJr SrSrSr Sr!Sr"Sr#\RH"SS\RJ55r&g)z!Command for spanner samples init.N) exceptions)database_operations)database_sessions) databases) instances) storage_api) storage_util)base) ddl_parser)samples)log) properties) resources)progress_tracker)files)retryc [R"U5 g![Ra- [ [ R "SRUS955ef=f)z*Raise if the given instance doesn't exist.z Instance '{instance_id}' does not exist. Create it with: $ gcloud spanner instances create {instance_id}  instance_idN)rGetapitools_exceptionsHttpNotFoundError ValueErrortextwrapdedentformatrs #lib/surface/spanner/samples/init.pycheck_instancer)sT. MM+  . ..   F{F +  - ...s AAcv[R"5n[RR [ R 5n[RRU[ R"U55[ R"U54/nURU[ R"U5S9n[ R"U5nUHnURRS5S(dM([RR!U5n["R$R'XVRRS5S5nUR)Xg45 M [ R*"U5(a[ R,"U5nURX(S9n U Hn [RR!U 5n U RR/S5(aMD[ R0"U5n ["R$R'XRRS5S5nUR)X45 M [3SU55(a[4R6R9SR;U55 UHupg[4R6R9SR;U55 ["R$RU5Sn ["R$R=U 5(d[>R@"U 5 URCXgSS 9 M g g ) zDownload schema and binaries for the given sample app. If the schema and all binaries exist already, skip download. If any file doesn't exist, download them all. Args: appname: The name of the sample app, should exist in samples.APP_NAMES )prefix/c3n# UH+up[RRU5(+v M- g7f)N)ospathexists).0_ file_paths r (download_sample_files..as%H<<1RWW^^I & & &,  ' ' 0(g <<  c "2 &**66w?Gh (:(:3(?(CDJ-.  ''00??HK%000P' $44@@Mo    & &s + +EEgNo77<< //55c:2>j?78( H<HHHJJ>EEgNO+ jj'..z:;''-- +A.i WW^^I & & i $$WD$I ,Ic [R"UUUUS9$![Ra7 [ [ R "SRXRS955e[Ra4n[ [R"UR5SS5eSnAf[a [ SRU55ef=f)z2Wrapper over databases.Create with error handling.)database_dialectz Database '{database_id}' exists already. Delete it with: $ gcloud spanner databases delete {database_id} --instance={instance_id} ) database_idrerrormessageNzFailed to create database '{}'.)rCreaterHttpConflictErrorrrrr instancesId HttpErrorjsonloadscontent Exception) instance_refrX statementsrWexs r _create_db_oprfksL   )  ++  . .M   F#1I1I  K  L MM  & &A TZZ +G4Y? @@ L 6==kJ KKLsAC 6/B%%(C cx[R"U5(d[SRU55e[R "U5nUR S5HZnU(dM URS5(dM$[R"SS9R[RUSU/SSS 9 M\ g ) z3Read and execute all insert statements in one file.{} cannot pre-populate data. INSERT) max_retrialsNORMALcg)NT)exc_typeargss r0insert_sample_data_in_one_file..srUi)rqshould_retry_ifsleep_msN) r r=rrrReadFileContentsr9 startswithrRetryerRetryOnExceptionr ExecuteSql)rF file_name session_refinsert_statementsinsert_statements rinsert_sample_data_in_one_filers  + +G 4 4 3::7C DD,,Y7+11$7   & &x 0 0 MMq!22$$+ 64 38rUcj[R"U5(d[SRU55e[R "SRU5SS9 [ R"[R"U55nUHn[UUU5 M SSS5 g!,(df  g=f)zInsert sample data.rhzPopulating data into `{}`z#Aborting wait for data population. )aborted_messageN) r r=rrrProgressTrackerrGetDirectoryTreeListingr@r)rFrXr| data_files data_files rinsert_sample_datars  + +G 4 4 3::7C DD''!((5<..55g>J $    s AB$$ B2c6[R"U5n[R"U5n[R"U5nU[ R :Xat/nSRURS5Vs/sHowRS5(aMUPM sn5nURS5Vs/sH o(dM UPM n nO[R"U5n/n [XUU5n [R"U SRU55 U (as[ R"R%UUR&UR(S.SS9n [ R*"X5n [R"U SRU55 g g s snfs snf) z;Create the DB if it doesn't exist already, raise otherwise.riz--;zCreating database '{}'r] projectsId$spanner.projects.instances.databasesparams collectionzUpdating database '{}'N)r r4get_database_dialectrrvrDATABASE_DIALECT_POSTGRESQLr;r9rwr PreprocessDDLWithParserrfrAwaitrrREGISTRYParser]r UpdateDdl) rFrcrX schema_filerWschema create_ddllinestmt update_ddl create_op database_ref update_ops rcheck_create_dbrse--g6+11':  ! !+ .& >>>J YY ,,t,J,$OOD4I,JLF$*<<#4=#44$#4J=J33F;JJLz,.)I4;;KHJ%%++'33&11 : ,;L##L=Ii6==kJL K>s7FF4 FFcV\rSrSrSrS\R "S50r\S5r Sr Sr g) InitzInitialize a Cloud Spanner sample app. This command creates a Cloud Spanner database in the given instance for the sample app and loads any initial data required by the application. EXAMPLESaC To initialize the 'finance' sample app using instance 'my-instance', run: $ {command} finance --instance-id=my-instance To initialize the 'finance-graph' sample app using instance 'my-instance', run: $ {command} finance-graph --instance-id=my-instance czURSSS9 URSS[SS9 URS[S S 9 g ) zArgs is called by calliope to gather arguments for this command. Args: parser: An argparse parser that you can use to add arguments that go on the command line after this command. Positional arguments are allowed. rFz5The sample app name, e.g. "finance", "finance-graph".)help --instance-idTz1The Cloud Spanner instance ID for the sample app.)requiredtyper --database-idzBID of the new Cloud Spanner database to create for the sample app.)rrN) add_argumentstr)parsers rArgs Init.Argss_ O   @ B   rUcURn[R"U5 UR n[RRUS[RRRR0SS9nURb URnO[R "U5n["R$R'SR)U55 [+U5 [-U5 ["R$R'SR)XbS 95 [/X%U5 [R0"U5(a[RRUUR2UR4S .S S9n[6R8"U5n[RR;UR<S S9n [?X&U 5 [@RB"S5[6RD"U 5 $SR)X$S9n URbU SR)U5- n [@RB"SR)U 55$![an[R "SU5eSnAff=f![an[R "SU5eSnAff=f![an[R "S U5eSnAff=f![Fa [IS5ef=f![6RD"U 5 f=f)zThis is what gets called when the user runs this command. Args: args: an argparse namespace. All the arguments that were provided to this command invocation. Returns: Some value that we want to have printed later. APPNAMENrzspanner.projects.instancesrzChecking instance '{}'rz@Initializing database '{database_id}' for sample app '{appname}')rXrFrrrz-spanner.projects.instances.databases.sessions) relative_namerzG Initialization done for your Spanner database. zKFailed to insert data for the database. Please fallback to manually insert.z%{appname} --instance-id={instance_id})rFrz --database-id {}z Initialization done. Next, start the backend gRPC service with: $ gcloud spanner samples backend {} )%rFr check_appnamercalliope_exceptionsBadArgumentExceptionrrrrrVALUEScoreproject GetOrFailrXget_db_id_for_appr rBrCrrrTrr=r]rrr[ParseRelativeNamer8rrrDeleterb SystemError) selfrqrFrerrcrXrsessionr| backend_argss rRunInit.RunsllGD G$""K%%++ *++0088BB 0 ,1L #$$k--g6kJJ-44[ABJ[! '"JJJ K 9;Jg[9))'22''-- )55(33 < .l"((6g&&88 D9k .7=    -<CCDl    %+22;?? __f\" $$K D  4 4Y CCD( J  4 4_b IIJ J  4 4_b IIJ&        -seI I1 J; KK I.I))I.1 J;JJ K$J;;KKKK4roN) __name__ __module__ __qualname____firstlineno____doc__rr detailed_help staticmethodrr__static_attributes__rorUrrrs=(// #   -*V$rUr)'rr_r$rapitools.base.pyrrgooglecloudsdk.api_lib.spannerrrrrgooglecloudsdk.api_lib.storagerr googlecloudsdk.callioper r"googlecloudsdk.command_lib.spannerr r googlecloudsdk.corer rrgooglecloudsdk.core.consolergooglecloudsdk.core.utilrrrrTrfrrrDefaultUniverseOnlyCommandrrorUrrs( >><4467(E96#*)8** .2JjL,,(%LPA$4<<A$A$rU