dZddlmZddlmZddlmZddlmZddlmZddlm Z ddl m Z dd l m Z dd l mZdd l mZdd l mZe j"e j$e j&j(Gd de j*Ze j$e j&j.GddeZe j$e j&j2GddeZy)z?Imports data into an AlloyDB cluster from Google Cloud Storage.)absolute_import)division)unicode_literals)api_util)cluster_operations)base)cluster_helper)flags)log) properties) resourcesc6eZdZdZdddZedZdZdZy) Importz>Import data into an AlloyDB cluster from Google Cloud Storage.z {description}z To import data into a cluster, run: $ {command} my-cluster --region=us-central1 --database=my-database --gcs-uri=gs://my-bucket/source-file-path --sql --user=my-user" ) DESCRIPTIONEXAMPLESc@tjj|tj|tj |tj |dtj|tj|tj|y)zySpecifies additional command flags. Args: parser: argparse.Parser: Parser object for command line inputs. FN) r ASYNC_FLAG AddToParserr AddRegion AddCluster AddDatabase AddSourceURI AddImportUserAddImportOptions)parsers &lib/surface/alloydb/clusters/import.pyArgsz Import.Args0sj OO' OOF V fe$ v  6"c0tj|||S)N)r ConstructImportRequestFromArgs)selfalloydb_messages cluster_refargss rr z%Import.ConstructImportRequestFromArgs?s  8 8+t rctj|j}|j}|j}|j j dtjjjj|j|j}|j|||}|jj!|}t"j$j'|j(d}|j*s&t-j.|d|jdt0j2j5dj7|j9|S)zConstructs and sends request. Args: args: argparse.Namespace, An object that contains the values for the arguments specified in the .Args() method. Returns: ProcessHttpResponse of the request made. z#alloydb.projects.locations.clusters) projectsId locationsId clustersIdz%alloydb.projects.locations.operations) collectionzImporting data from clusterFzOperation ID: {})r AlloyDBClient ReleaseTrackalloydb_clientr"resource_parserCreater VALUEScoreproject GetOrFailregionclusterr projects_locations_clustersrr REGISTRYParseRelativeNamenameasync_rAwaitr statusPrintformatName) r!r$clientr,r"r#reqopop_refs rRunz Import.RunDs& # #D$5$5$7 8F**N..((//-$$))11;;KK<< 0K  - -+t C  3 3 : :3 ?B    1 1 C2F ;; /1B1B1DeJJ'..v{{}=> IrN) __name__ __module__ __qualname____doc__ detailed_help staticmethodrr rCrrrr"s7G% - # # rrceZdZdZy) ImportAlphar]sF&'3=(=4#*) D%%(()@T  @*@FD%%++,E&E-ED%%**+EE,Er