`dZddlmZddlmZddlmZddlmZddlmZddlm Z ddl m Z dd l m Z d Z d Zd Zd ZdZy)zBHelpers for writing commands interacting with jobs and their IDs. )absolute_import)division)unicode_literals)actions) arg_parsers) dataflow_util) properties) resourcescv|jddd|jdddtjzy) zRegister flags for specifying a single Job ID. Args: parser: The argparse.ArgParser to configure with job-filtering arguments. jobJOB_IDzJob ID to operate on.metavarhelp--region REGION_IDz*Region ID of the job's regional endpoint. N add_argumentrDEFAULT_REGION_MESSAGE)parsers 4lib/googlecloudsdk/command_lib/dataflow/job_utils.py ArgsForJobRefrsD  eX4KL 9  0 0 13c z|j dddd||jdddtjzy) zRegister flags for specifying jobs using positional job IDs. Args: parser: The argparse.ArgParser to configure with job ID arguments. **kwargs: Extra arguments to pass to the add_argument call. r zJob IDs to operate on.rrrz*Region ID of the jobs' regional endpoint. N)jobsr)rkwargss rArgsForJobRefsr,sY& I%=IAGI  9  0 0 13rc|j}tj|}tjj |t jjjj|ddS)zExtract the Job Ref for a command. Used with ArgsForJobRef. Args: args: The command line arguments. Returns: A Job resource.  projectIdlocation dataflow.projects.locations.jobsparams collection) r r GetRegionr REGISTRYParser VALUEScoreproject GetOrFail)argsr regions r ExtractJobRefr/=sf #  " "4 (&    ! ! !((--55??4 " 55rc  |j}tj|}|Dcgc]S}tjj |t jjjj|ddUc}Scc}w)zExtract the Job Refs for a command. Used with ArgsForJobRefs. Args: args: The command line arguments that were provided to this invocation. Returns: A list of job resources. rr"r#) rrr&r r'r(r r)r*r+r,)r-rr.r s rExtractJobRefsr1Ps $  " "4 (&EI EIS  %,,1199CC  8 9 EI  sABc "|jdd|jdtjtjj j d|jdtd |jd d |jd td |jdtjddd |jdd|jdd|jdd}|jdtjddd |jdtjddd |jd!tjddd tjd!d"d#$%y&)'zRegister flags applicable to all template launches. Args: parser: argparse.ArgumentParser to register arguments with. z--dataflow-kms-keyz/The Cloud KMS key to protect the job resources.)rz--disable-public-ipsz   ) )    $ $ 7 79 I K  C&MO    C&MO   & &|'K M 7 9         ,F  G%  & &|'D F . 0   & &'B D , .   & &'B D ,  & & @   rN)__doc__ __future__rrrgooglecloudsdk.callioperr#googlecloudsdk.command_lib.dataflowrgooglecloudsdk.corer r rrr/r1r@rrrGs>''+/=*) 33"5&*Br