~dZddlmZddlmZddlmZddlmZddlm Z ddlm Z ddl mZGd d eZd Zd Zy )z"Factory class for Session message.)absolute_import)division)unicode_literals)jupyter_config_factory)environment_config_factory)runtime_config_factory) labels_utilc(eZdZdZdZ ddZdZy)SessionMessageFactoryzFactory class for Session message. Factory class for configuring argument parser and creating a Session message from the parsed arguments. zInvalid session type: {}.NcT||_|jjjdi|_|xs!t j |jd|_|xstj|j|_ |xstj|j|_ y)a7Builder class for Session message. Session message factory. Only the flags added in AddArguments are handled. User need to provide session 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. jupyter_config_factory_override: Override the default JupyterConfigFactory instance. jupyterSessionTuse_config_propertyN) dataprocmessages JupyterConfig _session2keyrcfRuntimeConfigFactoryrecfEnvironmentConfigFactoryrjcfJupyterConfigFactoryr)selfrruntime_config_factory_override#environment_config_factory_overridejupyter_config_factory_overrides Klib/googlecloudsdk/command_lib/dataproc/sessions/session_message_factory.py__init__zSessionMessageFactory.__init__(s$DM//==?OPD ( J   DI  , 4 $$T]]3 # ( 0   / ci}|jj|}|||jt|<|jrAt j ||jjjj|d<|jj|}|r||d<|jj|}|r||d<|jjjj!|d<|j"r|j"|d<|sy|jjjdi|S)aCreates a Session message from given args. Create a Session message from given arguments. Only the arguments added in AddArguments are handled. Users need to provide session type specific message during message creation. Args: args: Parsed argument. Returns: A Session message instance. Raises: AttributeError: When session is invalid. labels runtimeConfigenvironmentConfignamesessionTemplateN)r GetMessagertyper"r ParseCreateArgsrrSession LabelsValuerrCONCEPTSsessionParse RelativeNamesession_template)rargskwargssession_configruntime_configenvironment_configs rr(z SessionMessageFactory.GetMessageKs F00;;DAN6DF4  T.1 23 {{$44  &&..::rJs> )&'#'#<T4FT4n$r