o پi@sXdZddlZddlZddlmZddlmZmZmZddl Z e e Z GdddZ dS)z Configuration argument parser for command-line applications. Handles merging of YAML configuration files with command-line arguments. N)Path)AnyDictListc@seZdZdZ  ddejdeefddZdeedeefd d Z d eedefd d Z dede ee ffddZ deddfddZde ee fdeefddZd eedededdfddZd eededee ddfddZd eedede ddfddZdS)ConfigArgumentMergerzLHandles merging of configuration file arguments with command-line arguments.Nparserboolean_actionscCs^|dur||_dd|jD|_dd|jD|_dS|dur'||_i|_dSg|_i|_dS)z0Initialize with list of store_true action names.NcSsg|] }t|tjr|jqS) isinstanceargparse_StoreTrueActiondest).0actionr r X/home/ubuntu/.local/lib/python3.10/site-packages/sglang/srt/server_args_config_parser.py s  z1ConfigArgumentMerger.__init__..cSsPi|]$}|jr&t|tjs&t|tjs&d|jvr&d|jvr&d|jvr|j|qS)--configz--helpz-h)option_stringsr r r _StoreActionr )rar r r !s     z1ConfigArgumentMerger.__init__..)r_actionsstore_true_actionsunsupported_actions)selfrrr r r__init__s    zConfigArgumentMerger.__init__cli_argsreturncCsX||}|s |S||}||}|d}|d|}||dd}|||S)a Merge configuration file arguments with command-line arguments. Configuration arguments are inserted after the subcommand to maintain proper precedence: CLI > Config > Defaults Args: cli_args: List of command-line arguments Returns: Merged argument list with config values inserted Raises: ValueError: If multiple config files specified or no config file provided rN)_extract_config_file_path_parse_yaml_config_convert_config_to_argsindex)rrconfig_file_path config_data config_args config_index before_config after_configr r rmerge_config_with_args3s      z+ConfigArgumentMerger.merge_config_with_argsargscCsZddt|D}t|dkrtd|sdS|d}|t|dkr'td||dS)z,Extract the config file path from arguments.cSsg|] \}}|dkr|qS)rr )riargr r rrVszBConfigArgumentMerger._extract_config_file_path..z2Multiple config files specified! Only one allowed.Nrz-No config file specified after --config flag!) enumeratelen ValueError)rr*config_indicesr&r r rrTs  z.ConfigArgumentMerger._extract_config_file_path file_pathc Cs||zt|d }t|}Wdn1swYWnty9}z td|d|d}~ww|dur@i}t|tsIt d|S)a8 Parse YAML configuration file and convert to argument list. Args: file_path: Path to the YAML configuration file Returns: List of arguments in format ['--key', 'value', ...] Raises: ValueError: If file is not YAML or cannot be read rNzFailed to read config file z: z3Config file must contain a dictionary at root level) _validate_yaml_fileopenyaml safe_load Exceptionloggererrorr dictr0)rr2filer$er r rr ds    z'ConfigArgumentMerger._parse_yaml_configcCs@t|}|jdvrtd|j|std|dS)z&Validate that the file is a YAML file.)z.yamlz.ymlz&Config file must be YAML format, got: zConfig file not found: N)rsuffixlowerr0exists)rr2pathr r rr4s z(ConfigArgumentMerger._validate_yaml_fileconfigcCsg}|D]D\}}|dd}||jvr)|j|}d|d|jjd}t|t|tr6||||qt|t rC| |||q| |||q|S)z2Convert configuration dictionary to argument list.-_zUnsupported config option 'z' with action '') itemsreplacer __class____name__r0r bool_add_boolean_arglist _add_list_arg_add_scalar_arg)rrBr*keyvaluekey_normrmsgr r rr!s     z,ConfigArgumentMerger._convert_config_to_argsrOrPcCsR|dd}||jvr|r|d|dSdS|d|t|gdS)z Add boolean argument to the list. Only store_true flags: - value True -> add flag - value False -> skip Regular booleans: - always add --key true/false rCrD--N)rGrappendextendstrr?)rr*rOrPrQr r rrKs  z%ConfigArgumentMerger._add_boolean_argcCs0|r|d||dd|DdSdS)zAdd list argument to the list.rScss|]}t|VqdS)N)rV)ritemr r r sz5ConfigArgumentMerger._add_list_arg..N)rTrUrr*rOrPr r rrMsz"ConfigArgumentMerger._add_list_argcCs|d|t|gdS)z Add scalar argument to the list.rSN)rUrVrYr r rrNsz$ConfigArgumentMerger._add_scalar_arg)NN)rI __module__ __qualname____doc__r ArgumentParserrrVrr)rrrr r4r!rJrKrMrNr r r rrs"  ! ""r)r\r loggingpathlibrtypingrrrr6 getLoggerrIr9rr r r rs