w%DdZddlZddlmZddlmZddlmZddlmZddd ZGd d ejZ Gd d ejZ Gdde Z de dfiZ dZdZdZdZdZdZdZdZdZdZdZd&dZdZ d'dZd(d Zd!Zd"Zd#Zd$Z d%Z!y))z,Flags and helpers for the Managed Flink CLI.N) arg_parsers)base) completers)parameter_info_libz$The number of taskmanagers is fixed.zIThe number of taskmanagers is scaled automatically according to workload.)fixedelasticc*eZdZdZ dfd ZxZS)ListCommandParameterInfoz&Helper class for ListCommandCompleter.c2tt| ||||S)N)check_properties for_update)superr GetFlag)selfparameter_nameparameter_valuer r __class__s 5lib/googlecloudsdk/command_lib/managed_flink/flags.pyrz ListCommandParameterInfo.GetFlag%s+ )4 8) 9 )NTF)__name__ __module__ __qualname____doc__r __classcell__rs@rr r "s.   rr ceZdZdZdZy)ListCommandCompleterz%Helper class for DeploymentCompleter.c<t|||jtS)N)updaters)r collectionCOMPLETERS_BY_CONVENTION)r parsed_args argumentss r ParameterInfoz"ListCommandCompleter.ParameterInfo7s # )  rN)rrrrr$rrrr4s -rrc"eZdZdZfdZxZS)DeploymentCompleterz"Completer for listing deployments.c 2tt| dddd|y)Nz+managedflink.projects.locations.deploymentszmanaged-flink deployments list)r list_commandr%)rr'__init__)rkwargsrs rr*zDeploymentCompleter.__init__Cs( t-@5 r)rrrrr*rrs@rr'r'@s*rr' deploymentFcX|jdddd|jdddd y ) zAdds network config arguments.z--network-config-vpcNETWORKnetworkzThe network to use for the job.metavardesthelpz--network-config-subnetwork SUBNETWORK subnetworkz"The subnetwork to use for the job.N add_argumentparsers rAddNetworkConfigArgsr:OsB  ,   #  / rc.|jddddy)z Adds workload identity argument.z--workload-identityWORKLOAD_IDENTITYworkload_identityzpThe workload identity to use for the job. Managed Flink Default Workload Identity will be used if not specified.r0Nr6r8s rAddWorkloadIdentityArgumentr>_s$!  F rcZtjdddddgdj|y) zCreates location argument.z --locationLOCATIONTlocationz--regionzThe location to run the job in.)r1requiredr2suggestion_aliasesr3NrArgument AddToParserr8s rAddLocationArgumentrGls.-- $ , KrcZtjddgdddj|y)zJob type arguments.z --job-typeJOB_TYPE)autojarpythonsqlrJzdThe type of job to run. If "auto" will be selected based on the file extension for the job argument.)r1choicesdefaultr3NrDr8s rAddJobTypeArgumentrPxs+--. - KrcRtjdddj|y)zCreates the job argument.jobz JAR|PY|SQLzLThe file containing the Flink job to run. Can be a jar, python, or sql file.r1r3NrDr8s rAddJobJarArgumentrTs%--   Krcztjddtjddj |y)z Creates the extra jars argument.z--jarsJAR extra_jarszEThe extra jars to pass to the job. Can be a jar, python, or sql file.r1typer2r3NrrErArgListrFr8s rAddExtraJarsArgumentr\s4--      KrcXtjddddddj|y)zCreates dry run argument.z --dry-run store_truedry_runFz9Return command used to submit a job without invoking API.actionr2rOrBr3NrDr8s rAddDryRunArgumentrbs,--   F KrcTtjddddj|y)Creates main class argument.z--classCLASS main_classzaThe main class of the Flink job. Required if the jar file manifest does not contain a main class.r0NrDr8s rAddMainClassArgumentrgs(--  , KrcH|jdtjdy)z0Collects extra arguments into the job_args list.job_argszThe job arguments to pass.)nargsr3N)r7argparse REMAINDERr8s rAddJobArgsCollectorrms$    'rcVtjdddddj|y)zCreates name argument.z--nameNAMEnameFz@The name of the job. The Flink job name will be used if not set.r1r2rBr3NrDr8s rAddNameArgumentrrs)--  M  KrcRtjdddj|y)zCreates job id argument.job_idJOBIDzThe id of the job.rSNrDr8s rAddJobIdArgumentrvs#-- KrcXtjddd|ddj|y)zCreates async argument.z--asyncr^ async_submitFz(Return immediately after job submission.r`NrD)r9rOs rAddAsyncArgumentrys,--   5 KrcVtjdddddj|y)z"Creates staging location argument.z--staging-locationSTAGING_LOCATIONstaging_locationTzLThe Google Cloud Storage staging location for the job. Must start with gs://rqNrDr8s rAddStagingLocationArgumentr}s+--   Krc|r|}nd}|rdj||f}tjdd|dt|j |y)zCreates deployment argument.z: The Flink Deployment to use for this invocation. z z --deploymentDEPLOYMENT_NAMEr,)r1rBr2 completerr3N)joinrrEr'rF)r9help_text_to_prependhelp_text_to_overwriterB help_texts rAddDeploymentArgumentrsV&II 19=>I-- #  Krc btjddt||ddj|y)z!Creates autotuning mode argument.z--autotuning-modeAUTOTUNING_MODEautotuning_modez%Selects the autotuning mode for jobs.)r1rNrOrBr2r3N)rrE_AUTOTUNING_MODESrF)r9rOrBs rAddAutotuningModeArgumentrs1--  2KrcV|jdtjdddy)z!Adds fixed parallelism arguments.z --parallelism' lower_bound upper_boundz;The parallelism of the job when in "fixed" autotuning mode.rYr3Nr7r BoundedIntr8s rAddFixedParallelismArgsr's+  ! !aU C Hrc|jdtjddd|jdtjdddy ) z#Adds elastic parallelism arguments.z--min-parallelismrrrz{The minimum parallelism of the job when in "elastic" autotuning mode. This will also be the initial parallelism of the job.rz--max-parallelismzEThe maximum parallelism of the job when in "elastic" autotuning mode.Nrr8s rAddElasticParallelismArgsr0s[  ! !aU C I    ! !aU C  rcXtjddddddj|y)zCreates show output argument.z--enable-outputr^ show_outputFz%Shows the output of the Flink client.r`NrDr8s rAddShowOutputArgumentrDs,--   2 Krcztjddtjddj |y)z$Creates the extra archives argument.z --archivesZIParchivesz_The extra archives to pass to the job. Can be a zip file containing resource files for the job.rXNrZr8s rAddExtraArchivesArgumentrPs4--     ) KrcTtjddddj|y)rdz --python-venvr python_venvzThe path to the zip file to manage the virtualenv for Python dependencies. Required if the job type is python. Must start with gs://.r0NrDr8s rAddPythonVirtualEnvArgumentr^s(--   Kr)F)NNF)rF)"rrkgooglecloudsdk.callioperrgooglecloudsdk.command_lib.utilrrrParameterInfoByConventionr rr'r!r:r>rGrPrTr\rbrgrmrrrvryr}rrrrrrrr%rrrs3/(6>4 1KK$ ::: .)+>*FG       2      8 (   r