?(SrSSKJr SSKJr SSKJr SSKrSSKrSSKJr SSK J r SSK J r SS K J r SS K Jr S r\R R#\R R%S 5\5r\R("S \5rSr\R R#\\5rSr\R R#\\5rSr\R R#\\5rSr\R R#\\5rSrSrSr Sr!Sr"Sr#Sr$\RJ"S/SQ5r&\ \&"SSSSSS\\\RNS9 \!\&"SSSSS\S S!\RNS"9 \"\&"S#SSSS$S\\\RNS9 \#\&"S%SSS&S\S S!\RPS"9 \$\&"S'SSSS(S\\\RNS9 0r)Sr*S)r+Sr,"S*S+\ RZ5r.S,r/S-r0S.r1S/r2S0r3S1r4S2r5S3r6S4r7S5r8S7S6jr9g)8zSpanner samples API helper.)absolute_import)division)unicode_literalsN) databases) exceptions)execution_utils)log)filesz.gcloud-spanner-samples~GCLOUD_SPANNER_SAMPLES_HOMEbinr etczdata-insert-statementszgs://cloud-spanner-samplesz not-existbankingfinancez finance-graphz finance-pggamingAppAttrs) db_idbin_pathetc_path gcs_prefix schema_file backend_bin workload_bindatabase_dialectdata_insert_statements_pathz banking-dbzbanking-schema.sdl) rrrrrrrrrz finance-dbzfinance-schema.sdlz-server-1.0-SNAPSHOT-jar-with-dependencies.jarz/workload-1.0-SNAPSHOT-jar-with-dependencies.jar) rrrrrrrrrzfinance-graph-dbzfinance-graph-schema.sdlz finance-pg-dbzfinance-schema-pg.sdlz gaming-dbzgaming-schema.sdlschemac\rSrSrSrSrg)SpannerSamplesErrorz5User error running Cloud Spanner sample app commands.N)__name__ __module__ __qualname____firstlineno____doc____static_attributes__r 1lib/googlecloudsdk/command_lib/spanner/samples.pyrrs=r'rcLU[;a[SRU55eg)zRaise if the given sample app doesn't exist. Args: appname: str, Name of the sample app. Raises: ValueError: if the given sample app doesn't exist. zUnknown sample app '{}'N)APPS ValueErrorformatappnames r( check_appnamer/s( D .55g> ??r'c>[U5 [UR$)zGet the database ID for the given sample app. Args: appname: str, Name of the sample app. Returns: str, The database ID, e.g. "finance-db". Raises: ValueError: if the given sample app doesn't exist. )r/r*rr-s r(get_db_id_for_appr1s g  r'c[U5 [Un[RR [ UR UR5$)aGet the local path of the schema file for the given sample app. Note that the file and parent dirs may not exist. Args: appname: str, Name of the sample app. Returns: str, The local path of the schema file. Raises: ValueError: if the given sample app doesn't exist. )r/r*ospathjoinSAMPLES_ETC_PATHrrr. app_attrss r(get_local_schema_pathr9s>7m) & (:(:++ --r'c[U5 [RR[[ UR 5$)aYGet the local path to binaries for the given sample app. This typically includes server and workload binaries and any required dependencies. Note that the path may not exist. Args: appname: str, Name of the sample app. Returns: str, The local path of the sample app binaries. Raises: ValueError: if the given sample app doesn't exist. )r/r3r4r5SAMPLES_BIN_PATHr*rr-s r(get_local_bin_pathr<s, &W (>(> ??r'c[U5 [UR[:Xa[ SR U55e[ RR[[UR5$)a!Get the local path to data insert statements for the given sample app. Args: appname: str, Name of the sample app. Returns: str, The local path of the sample app data insert statements. Raises: ValueError: if the given sample app or the data_insert_statements_path don't exist. .Unknown sample app data insert statements '{}') r/r*r NOT_EXISTr+r,r3r4r5SAMPLES_DATA_INSERT_PATHr-s r(%get_local_data_insert_statements_pathrAs_ ']..); 8??H  W I I r'c[U5 [UnSRUR[UR /5$)a9Get the GCS file path for the schema for the given sample app. Doesn't include the bucket name. Use to download the sample app schema file from GCS. Args: appname: str, Name of the sample app. Returns: str, The sample app schema GCS file path. Raises: ValueError: if the given sample app doesn't exist. /)r/r*r5r_GCS_SCHEMA_PREFIXrr7s r(get_gcs_schema_namerEs?7m) /1F1FG IIr'cj[U5 SR[UR[S/5$)aoGet the GCS prefix for binaries for the given sample app. Doesn't include the bucket name. Different sample apps have different numbers and types of binaries, list the bucket contents before downloading. Args: appname: str, Name of the sample app. Returns: str, The sample app binaries GCS prefix. Raises: ValueError: if the given sample app doesn't exist. rC)r/r5r*r_GCS_BIN_PREFIXr-s r(get_gcs_bin_prefixrIs, 4=++_bA BBr'c[U5 [UR[:Xa[ SR U55eSR [UR[S/5$)zGet the GCS prefix for data insert statements for the given sample app. Args: appname: str, Name of the sample app. Returns: str, The sample app binaries GCS prefix. Raises: ValueError: if the given sample app or the gcs_prefix don't exist. r>rCrG)r/r*rr?r+r,r5"_GCS_DATA_INSERT_STATEMENTS_PREFIXr-s r(%get_gcs_data_insert_statements_prefixrL sa ']* 8??H   G}!CRH r'c>[U5 [UR$)zGet the database dialect for the given sample app. Args: appname: str, Name of the sample app. Returns: str, The database dialect. Raises: ValueError: if the given sample app doesn't exist. )r/r*rr-s r(get_database_dialectrN"s g ' ''r'c[U5 [UR[:g=(a [UR[:g$)aCheck if the sample app has both gcs_prefix and data_insert_statements_path. Args: appname: str, Name of the sample app. Returns: bool, both gcs_prefix and data_insert_statements_path exist. Raises: ValueError: if the given sample app doesn't exist. )r/r*rr?rr-s r(has_sample_data_statementsrP2s; 7m)+A w- 3 3y @r'cU(aN[R"USSS9n[RR SR U55 [ X"S9nO0n[R"U40UD6$)aEWrapper for execution_utils.Subprocess that optionally captures logs. Args: args: [str], The arguments to execute. The first argument is the command. capture_logs_fn: str, If set, save logs to the specified filename. Returns: subprocess.Popen or execution_utils.SubprocessTimeoutWrapper, The running subprocess. T)append create_pathzWriting logs to {})stdoutstderr) r FileWriterr statusPrintr,dictr Subprocess)argscapture_logs_fnlogfile popen_argss r(run_procr_Es^tNGJJ)00ABW5JJ  # #D 7J 77r')N):r% __future__rrr collectionsr3googlecloudsdk.api_lib.spannerrgooglecloudsdk.corerrr googlecloudsdk.core.utilr _SAMPLES_DEFAULT_DIR_NAMEr4r5 expanduser_SAMPLES_DEFAULT_DIR_PATHgetenvSAMPLES_DIR_PATH _BIN_RELPATHr; _LOG_RELPATHSAMPLES_LOG_PATH _ETC_RELPATHr6_DATA_INSERT_RELPATHr@ GCS_BUCKETr?BANKING_APP_NAMEFINANCE_APP_NAMEFINANCE_GRAPH_APP_NAMEFINANCE_PG_APP_NAMEGAMING_APP_NAME namedtuplerDATABASE_DIALECT_GOOGLESQLDATABASE_DIALECT_POSTGRESQLr*rHrDrKErrorrr/r1r9r<rArErIrLrNrPr_r r'r(rysh"&' 4*/#*6GGLLGGs6899:68 77<< 0,? 77<< 0,? 77<< 0,?/77<<(8:NO*  ("  ! !   h$-("== h($-CF"== H   $3.""== +$-CF">> X$,'"== [8t%=">***> @  -(@&.I*C&, ( &8r'