dZddlmZddlmZddlmZddlmZddlmZddl m Z ddl m Z dd l Z d Z d Zd Zd ZdZdZdZy )zHHelpers for writing commands interacting with Cloud Dataflow snapshots. )absolute_import)division)unicode_literals) arg_parsers) dataflow_util) properties) resourcesNcV|jddd|jdddd y ) zRegister flags for specifying a single Snapshot ID. Args: parser: The argparse.ArgParser to configure with snapshot arguments. snapshot SNAPSHOT_IDz"ID of the Cloud Dataflow snapshot.)metavarhelp--regionT REGION_IDz,Region ID of the snapshot regional endpoint.requiredr rN add_argumentparsers 9lib/googlecloudsdk/command_lib/dataflow/snapshot_utils.pyArgsForSnapshotRefrsA   /1  9 ;cX|jdddd|jddddy ) zRegister flags for specifying a single Job ID. Args: parser: The argparse.ArgParser to configure with job-filtering arguments. --job-idTJOB_IDzThe job ID to snapshot.rrr:The region ID of the snapshot and job's regional endpoint.Nrrs rArgsForSnapshotJobRefr0sE   $ &   H JrcX|jdddd|jdddd y ) zRegister flags for listing Cloud Dataflow snapshots. Args: parser: The argparse.ArgParser to configure with job-filtering arguments. rFrz/The job ID to use to filter the snapshots list.rrTrrNrrs rArgsForListSnapshotr BsE   < >   H Jrc Z|jdddtjdddy ) z~Register flags for specifying a snapshot ttl. Args: parser: the argparse.ArgParser to configure with a ttl argument. z--snapshot-ttl7dDURATION1h30d) lower_bound upper_boundzTime to live for the snapshot.)defaultr typerN)rrDurationrs rArgsForSnapshotTtlr+Ts3     De D + -rc|j}tj|}tjj |t jjjj|ddS)zExtract the Snapshot Ref for a command. Used with ArgsForSnapshotRef. Args: args: The command line arguments. Returns: A Snapshot resource.  projectIdlocationz%dataflow.projects.locations.snapshotsparams collection) r r GetRegionr REGISTRYParserVALUEScoreproject GetOrFail)argsr regions rExtractSnapshotRefr<bsf]](  " "4 (&    ! !!((--55??9 " ::rc|j}tj|}tjj |t jjjj|ddS)zExtract the Job Ref for a command. Used with ArgsForSnapshotJobRef. Args: args: The command line arguments. Returns: A Job resource. r-z dataflow.projects.locations.jobsr0) job_idrr3r r4r5rr6r7r8r9)r:jobr;s rExtractSnapshotJobRefr@usf #  " "4 (&    ! ! !((--55??4 " 55rcFtj|jdzS)zExtract the Duration string for the Snapshot ttl. Args: args: The command line arguments. Returns: A duration string for the snapshot ttl. s)six text_type snapshot_ttl)r:s rExtractSnapshotTtlDurationrFs t(( )C //r)__doc__ __future__rrrgooglecloudsdk.callioper#googlecloudsdk.command_lib.dataflowrgooglecloudsdk.corerr rCrrr r+r<r@rFrrrMsI''/=*) ;"J$J$ -:&5&0r