o si@s\ddlZddlZddlmZdddZddZddZd d Zd d Z d dZ dddZ dS)N)parsecCsrdd}|dur t}|D]&}||}||D]}||||}|jd|||||dqq|S)aPrepare an argparser from a dictionary. Args: dic (dict): Two-level config dictionary with unique bottom-level keys. parser (argparse.ArgumentParser, optional): If a parser already exists, add the keys from the dictionary on the top of it. Returns: argparse.ArgumentParser: Parser instance with groups corresponding to the first level keys and arguments corresponding to the second level keys with default values given by the values. cSs&|durtStt|trtSt|S)zIf the default value is None, replace NoneType by str_int_float. If the default value is boolean, look for boolean strings.N) str_int_float isinstancestr2boolbool str2bool_argtypevaluer O/home/ubuntu/.local/lib/python3.10/site-packages/asteroid/utils/parser_utils.pystandardized_entry_types z9prepare_parser_from_dict..standardized_entry_typeNz--)defaultr)argparseArgumentParserkeysadd_argument_group add_argument)dicparserr kgroupkk entry_typer r r prepare_parser_from_dicts   rcCs2t|rt|St|rt|St|tr|SdS)zType to convert strings to int, float (in this order) if possible. Args: value (str): Value to convert. Returns: int, float, str: Converted value. N)isintintisfloatfloatrstrr r r r r(s  rcCs2t|ts|S|dvrdS|dvrdS|S)zAType to convert strings to Boolean (returns input if not boolean))yestruey1T)nofalsen0F)rrlowerr r r r r9s   rcCs t|}t|tr |Std)z+Argparse type to convert strings to BooleanzBoolean value expected.)rrrrArgumentTypeErrorr r r r rEs  rcC$zt|WdStyYdSw)zComputes whether `value` can be cast to a float. Args: value (str): Value to check. Returns: bool: Whether `value` can be cast to a float. TF)r ValueErrorr r r r rM   rcCr*)zComputes whether `value` can be cast to an int Args: value (str): Value to check. Returns: bool: Whether `value` can be cast to an int. TF)rr+r r r r r^r,rFcsx|jdi}|jD]}fdd|jD}|||j<q tttdkr)dnd}|||d<||=|r:|fS|S)a4Get a dict of dicts out of process `parser.parse_args()` Top-level keys corresponding to groups and bottom-level keys corresponding to arguments. Under `'main_args'`, the arguments which don't belong to a argparse group (i.e main arguments defined before parsing from a dict) can be found. Args: parser (argparse.ArgumentParser): ArgumentParser instance containing groups. Output of `prepare_parser_from_dict`. return_plain_args (bool): Whether to return the output or `parser.parse_args()`. args (list): List of arguments as read from the command line. Used for unit testing. Returns: dict: Dictionary of dictionaries containing the arguments. Optionally the direct output `parser.parse_args()`. argscsi|] }|jt|jdqSN)destgetattr).0ar-r r sz&parse_args_as_dict..z3.10zoptional argumentsoptions main_args) parse_args_action_groups_group_actionstitle parse_versionplatformpython_version)rreturn_plain_argsr.args_dicr group_dict opt_args_keyr r-r parse_args_as_dictos    rBr/)FN) rr<packaging.versionrr;rrrrrrrBr r r r s  "