s2|SrSSKJr SSKJr SSKJr SSKrSSKrSSKrSSKJ r SSK J r SSK J r SS K Jr SS K Jr SS KJr SS KJr SS KJr SSKJ r SSKJr SSKJr SSKJr SSKJr SSKJr SSKJr SSKJ r SSKJ!r! Sr"Sr#Sr$Sr%Sr&Sr'\RP"SS\RR55r*g)z!Command for spanner samples init.)absolute_import)division)unicode_literalsN) 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)r Getapitools_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..ds%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 '{}'.)r CreaterHttpConflictErrorrrrr instancesId HttpErrorjsonloadscontent Exception) instance_refr[ statementsrZexs r _create_db_oprinsL   )  ++  . .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 r 0insert_sample_data_in_one_file..srXi)rtshould_retry_ifsleep_msN) rr@rrrReadFileContentsr< startswithrRetryerRetryOnExceptionr ExecuteSql)rI file_name session_refinsert_statementsinsert_statements r insert_sample_data_in_one_filers  + +G 4 4 3::7C DD,,Y7+11$7   & &x 0 0 MMq!22$$+ 64 38rXcj[R"U5(d[SRU55e[R "SRU5SS9 [ R"[R"U55nUHn[UUU5 M SSS5 g!,(df  g=f)zInsert sample data.rkzPopulating data into `{}`z#Aborting wait for data population. )aborted_messageN) rr@rrrProgressTrackerrGetDirectoryTreeListingrCr)rIr[r data_files data_files r insert_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.rlz--;zCreating database '{}'r` projectsId$spanner.projects.instances.databasesparams collectionzUpdating database '{}'N)rr7get_database_dialectrryr DATABASE_DIALECT_POSTGRESQLr>r<rzrPreprocessDDLWithParserrirAwaitrrREGISTRYParser`r UpdateDdl) rIrfr[ schema_filerZschema create_ddllinestmt update_ddl create_op database_ref update_ops r check_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. rIz5The 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 r Args Init.Argss_ O   @ B   rXcURn[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}')r[rIrrrz-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})rIrz --database-id {}z Initialization done. Next, start the backend gRPC service with: $ gcloud spanner samples backend {} )%rIr check_appnamercalliope_exceptionsBadArgumentExceptionrrrrrVALUEScoreproject GetOrFailr[get_db_id_for_apprrErFrr!rWrr@r`rrr^ParseRelativeNamer;rrrDeletere SystemError) selfrtrIrhrrfr[rsessionr backend_argss r RunInit.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;;KKKK4rrN) __name__ __module__ __qualname____firstlineno____doc__rr detailed_help staticmethodrr__static_attributes__rrrXr rrs=(// #   -*V$rXr)+r __future__rrrrbr'rapitools.base.pyrrgooglecloudsdk.api_lib.spannerrrr r googlecloudsdk.api_lib.storager r googlecloudsdk.callioper r"googlecloudsdk.command_lib.spannerrrgooglecloudsdk.corerrrgooglecloudsdk.core.consolergooglecloudsdk.core.utilrrr!rWrirrrDefaultUniverseOnlyCommandrrrrXr rs(&' >><4467(E96#*)8** .2JjL,,(%LPA$4<<A$A$rX