s2|dZddlmZddlmZddlmZddlZddlZddlZddlm Z ddl m Z ddl m Z dd l mZdd l mZdd lmZdd lmZdd lmZddlm ZddlmZddlmZddlmZddlmZddlmZddlmZddlm Z ddlm!Z!dZ"dZ#dZ$dZ%dZ&dZ'ejPGddejRZ*y)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 tj|y#tj$r/t t j dj|wxYw)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 AAcBtj}tjj t j }tjj|t j|t j|fg}|j|t j|}t j|}|D]}|jjdds"tjj!|}t"j$j'||jjdd}|j)||ft j*|rt j,|}|j||} | D]} tjj!| } | jj/dr>t j0|} t"j$j'| | jjdd}|j)| |ft3d|Drt4j6j9dj;||D]\}}t4j6j9dj;|t"j$j|d} t"j$j=| st?j@| |jC||d y y ) 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/c3bK|]'\}}tjj| )yw)N)ospathexists).0_ file_paths r z(download_sample_files..ds%H<<1iRWW^^I & &,  ' ' 0(g <<  c "2 &**66w?Gh (:(:3(?(CDJ*-.  ''0??HK%00K0P' $44@@Mo    & &s +EEgNo77<< ?//55c:2>j?J78( H<HHJJ>EEgNO+ jj'..z:;''-- +A.i WW^^I & i $$WjD$I ,Ic  tj||||S#tj$r:t t j dj||jtj$r3}t tj|jddd}~wt$rt dj|wxYw)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_refrZ statementsrYexs r _create_db_oprhnsL   )  ++  . .M   F#1I1I  K  L MM  & &A TZZ +G4Y? @@ L 6==kJ KKLsAC9.B'''CcXtj|stdj|t j |}|j dD]R}|s|jdstjdjtj|d|gdd Ty ) z3Read and execute all insert statements in one file.{} cannot pre-populate data. INSERT) max_retrialsNORMALcy)NT)exc_typeargss r z0insert_sample_data_in_one_file..srWi)rsshould_retry_ifsleep_msN) rr?rrrReadFileContentsr; startswithrRetryerRetryOnExceptionr ExecuteSql)rH file_name session_refinsert_statementsinsert_statements r insert_sample_data_in_one_filers  + +G 4 3::7C DD,,Y7+11$7   & &x 0 MMq!22$$+ 64 38rWcRtj|stdj|t j dj|d5t jtj|}|D]}t||| dddy#1swYyxYw)zInsert sample data.rjzPopulating data into `{}`z#Aborting wait for data population. )aborted_messageN) rr?rrrProgressTrackerrGetDirectoryTreeListingrBr)rHrZr} data_files data_files r insert_sample_datars  + +G 4 3::7C DD''!((5<..55g>J $    s =BB&ctj|}tj|}tj|}|t j k(rdg}dj|jdDcgc]}|jdr|c}}|jdDcgc]}|s| } }ntj|}g} t||||} tj| dj|| rut j"j%||j&|j(dd} t j*| | } tj| dj|y y cc}wcc}w) z;Create the DB if it doesn't exist already, raise otherwise.rkz--;zCreating database '{}'r_ projectsId$spanner.projects.instances.databasesparams collectionzUpdating database '{}'N)rr6get_database_dialectrrwr DATABASE_DIALECT_POSTGRESQLr=r;rxrPreprocessDDLWithParserrhrAwaitrrREGISTRYParser_r UpdateDdl) rHrerZ schema_filerYschema create_ddllinestmt update_ddl create_op database_ref update_ops r check_create_dbrsi--g6+11':  ! !+ .& >>>J YY ,,t,J,$DOOD4I,JLF$*<<#4=#44$#4J=33F;JJL+z,.)I4;;KHJ%%++'33&11 : ,;L##L*=Ii6==kJL K>s3E9 E9(E>0E>cNeZdZdZdej diZedZdZ y)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 c|jdd|jddtd|jdtd y ) 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. rHz5The 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 Argsz Init.Argss_ O   @ B   rWc|j} tj||j }tjj|dtjjjjid}|j |j}ntj |}t"j$j'dj)| t+|t-|t"j$j'dj)||  t/|||tj0|rtjj||j2|j4d d }t7j8|}tjj;|j<d } t?||| tAjBdt7jD| Sdj)||} |j| dj)|z } tAjBdj)| S#t$r}t j d|d}~wwxYw#t$r}t j d|d}~wwxYw#t$r}t j d |d}~wwxYw#tF$r tIdwxYw#t7jD| wxYw)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}')rZrHrrrz-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})rHrz --database-id {}z Initialization done. Next, start the backend gRPC service with: $ gcloud spanner samples backend {} )%rHr check_appnamercalliope_exceptionsBadArgumentExceptionrrrrrVALUEScoreproject GetOrFailrZget_db_id_for_apprrDrErr!rVrr?r_rrr]ParseRelativeNamer:rrrDeleterd SystemError) selfrsrHrgrrerZrsessionr} backend_argss r RunzInit.RunsllGD G$""K%%++ *++0088BB 0 ,1L #$$k--g6kJJ-44[ABJ[! '"JJJ K 9;Jg|[9))'2''-- )55(33 < .l"((6g&&88 D9k .7K=    -<CC{Dl    %+22;?? __f\" $$K D  4 4Y CCD( J  4 4_b IIJ J  4 4_b IIJ&        -seI  I4# J? K K I1I,,I14 J=JJ J?$J::J?KKK1N) __name__ __module__ __qualname____doc__rr detailed_help staticmethodrrrqrWr rrs?/(// #   -*V$rWr)+r __future__rrrrar'rapitools.base.pyrrgooglecloudsdk.api_lib.spannerrrr r googlecloudsdk.api_lib.storager r googlecloudsdk.callioper r"googlecloudsdk.command_lib.spannerrrgooglecloudsdk.corerrrgooglecloudsdk.core.consolergooglecloudsdk.core.utilrrr!rVrhrrrDefaultUniverseOnlyCommandrrqrWr rs(&' >><4467(E96#*)8** .2JjL,,(%LPA$4<<A$A$rW