_dZddlmZddlmZddlmZddlZddlZddlZddlm Z ddl m Z ddl m Zdd l mZdd lmZdd lmZdd lmZdd lmZddlmZddlmZddlZddiZdZej>ej@ejBjDejBjFejBjHejJGddejLZ'y)zGCommand to query metadata against Dataproc Metastore services database.)absolute_import)division)unicode_literalsN) exceptions)operations_util) services_util)util) storage_api) storage_util)base) resource_args)log)resource_printerEXAMPLESa) To query metadata against a Dataproc Metastore service with the name `my-metastore-service` in location `us-central1`, and the sql query "show tables;", run: $ {command} my-metastore-service --location=us-central1 --query="show tables;" c`tj|dddd|jdddy) z|Parses provided arguments to add base arguments used for Alpha/Beta/GA. Args: parser: an argparse argument parser. zto query metadataFT)pluralrequired positionalz--queryz Use Google Standard SQL query for Cloud Spanner and MySQL query syntax for Cloud SQL. Cloud Spanner SQL is described at https://cloud.google.com/spanner/docs/query-syntax)" )rhelpN)r AddServiceResourceArg add_argumentparsers 0lib/surface/metastore/services/query_metadata.py AddBaseArgsr3s= %% !%$4Q c0eZdZdZeZedZdZdZ y)QueryzDExecute a SQL query against a Dataproc Metastore Service's metadata.cXt|tjj|y)zSee base class.N)rr FORMAT_FLAG AddToParserrs rArgsz Query.ArgsPs   (rc 6|jjj}d} tj|j |j |j}tjjdj|jd} t#j$|dj|j |j}|$|j.rt1|j.d kryd}|j.D](}|j2d k(s|j4j6}*|yt9j:}t=j>tAjB|jEtFjHjK|d d } tjj| dd|| dsyt1| ddkDrtjjd| dd} t=j>tAjB|jEtFjHjKtMjNtMjP|| d d S#tj$rtj dwxYw#tj&$rE}tj(|j ddt+j,|Yd}~/d}~wwxYw)zRuns 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. N) release_trackzwith operation [{}]zQuery did not succeed.zWaiting for [{}] to queryserviceF)kindis_asyncfailedresultManifestUriTzutf-8)encodingstatusmessage filenameszcThe number of rows exceeds 1000 to display. Please find more results at the cloud storage location.r))CONCEPTSr%Parseservices_api_util QueryMetadata RelativeNamequery ReleaseTrackroutPrintformatnameapitools_exceptions HttpErrorapi_utilQueryMetadataErrorrPollAndReturnOperationOperationErrorUpdatedResourcesix text_typeadditionalPropertieslenkeyvalue string_valuer StorageClientjsonloadio TextIOWrapper ReadObjectr ObjectReferenceFromUrl posixpathjoindirname) selfargsenv_ref operationoperation_resulteresult_manifest_urir- gcs_clientresult_manifest_jsonquery_result_file_names rRunz Query.RunVsmm##))+GIB#11    $**DqA 99   ! !,,44NN!))*=>.    S  ( (B  ' '(@ AAB  " "#     q! ###s%A6J!AK)J=L:LLcH|rd|vs d|dvsd|vry|ddDcgc] }|dxsd }}djdt|D}|dDcgc]}d |i}}tj|d j |t j ycc}wcc}w) aDDisplays the server response to a query. This is called higher up the stack to over-write default display behavior. What gets displayed depends on the mode in which the query was run. Args: args: The arguments originally passed to the command. result: The output of the command before display. metadatacolumnsrowsNr:z (Unspecified),c3FK|]\}}dj||yw)z!row.slice({0}).join():label="{1}"N)r9).0ifs r z Query.Display..s*;(91@FFq!L(9s!rowz table({0}))r7)rR enumeraterr8r9rr7)rTrUresultfieldfields table_formatrirbs rDisplayz Query.Displays Zv-&C2V+ J' 22E f ((2  88;(1&(9;;L$*6N 3NSUCLND 34!4!4\!BP 4s B BN) __name__ __module__ __qualname____doc__ DETAILED_HELP detailed_help staticmethodr"r^rorrrrFs, M-)) GRQrr)(rs __future__rrrrLrJrQapitools.base.pyrr; googlecloudsdk.api_lib.metastorerrr2r r=googlecloudsdk.api_lib.storager r googlecloudsdk.callioper $googlecloudsdk.command_lib.metastorer googlecloudsdk.corergooglecloudsdk.core.resourcerrBrtrUnicodeIsSupported ReleaseTracksr6ALPHABETAGADefaultUniverseOnlyCommandrrwrrrsN&' ><O=67(>#9    &T..33T5F5F5I5IjQDLLjQ jQr