,rSrSSKJr SSKJr SSKJr SSKJr SSKJr SSK J r "SS \ 5r S rS rg ) z Factory class for Batch message.)absolute_import)division)unicode_literals)environment_config_factory)runtime_config_factory) labels_utilc4\rSrSrSrSrSrS SjrSrSr g) BatchMessageFactoryzFactory class for Batch message. Factory class for configuring argument parser and creating a Batch message from the parsed arguments. zInvalid batch job type: {}.zMissing batch job.Nc PXlURRRSURRRSURRRSURRR SURRR S0UlX lUR(d%[R"URSSS9UlX0l UR(d&[R"UR5Ul gg) aBuilder class for Batch message. Batch message factory. Only the flags added in AddArguments are handled. User need to provide batch job type specific message during message creation. Args: dataproc: A api_lib.dataproc.Dataproc instance. runtime_config_factory_override: Override the default RuntimeConfigFactory instance. environment_config_factory_override: Override the default EnvironmentConfigFactory instance. sparkBatch sparkRBatch sparkSqlBatch pysparkBatchrayBatchTinclude_autotuninginclude_cohortN)dataprocmessages SparkBatch SparkRBatch SparkSqlBatch PySparkBatchRayBatch _batch2keyrrcfRuntimeConfigFactoryrecfEnvironmentConfigFactory)selfrruntime_config_factory_override#environment_config_factory_overrides Hlib/googlecloudsdk/command_lib/dataproc/batches/batch_message_factory.py__init__BatchMessageFactory.__init__)s&M ))< **M ,,o ++^ '' DO#B  & &$'$<$< --D%d!'J#  * *(+(D(D --)d% +cU(d[[R5e[U[ UR R 555(d1[[RR[U555e0nX#UR [U5'UR(aA[R"XRRRR 5US'UR"R%U5nU(aXCS'UR&R%U5nU(aXSS'U(dgURRR"S0UD6$)aCreates a Batch message from given args. Create a Batch message from given arguments. Only the arguments added in AddAddArguments are handled. User need to provide bath job type specific message during message creation. Args: args: Parsed argument. batch_job: Batch type job instance. Returns: A Batch message instance. Raises: AttributeError: When batch_job is invalid. labels runtimeConfigenvironmentConfigN)AttributeErrorr MISSING_BATCH_ERR_MSG isinstancetuplerkeysINVALID_BATCH_TYPE_ERR_MSGformattyper)rParseCreateArgsrrBatch LabelsValuer GetMessager)r!args batch_jobkwargsruntime_configenvironment_configs r$r8BatchMessageFactory.GetMessageSs"  .DD EE it';';'=!> ? ?   8 8 ? ?Y P F/84??4 ? +, {{$44  &&,,88fX00;;DAN ._88CCDI$6 !   == ! ! ' ' 1& 11r')rrrr)NN) __name__ __module__ __qualname____firstlineno____doc__r2r.r%r8__static_attributes__r,r'r$r r s( =. '+*. (T-2r'r cF[R"U5 [U5 g)zAdds arguments related to Batch message. Add Batch arguments to the given parser. Job specific arguments are not handled, and need to be set during factory instantiation. Args: parser: A argument parser. N)rAddCreateLabelsFlags_AddDependencyparsers r$ AddArgumentsrJs""6*r'c\[R"USSS9 [R"U5 g)NTr)rrJrrHs r$rGrGs"6d4H6r'N)rC __future__rrr3googlecloudsdk.command_lib.dataproc.shared_messagesrrrr$googlecloudsdk.command_lib.util.argsrobjectr rJrGr,r'r$rPs; '&'=a2&a2H r'