o )Tib@s^dZddlZddlZddlZddlmZmZddlmZm Z m Z m Z ddl m Z ddlmZmZmZmZmZmZmZddlmZdd lmZmZmZmZdd lmZmZm Z dd l!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)dd l*m+Z+m,Z,m-Z-m.Z.m/Z/dd lm0Z0dgZ1ej2Z3ej4Z5GdddeZ6de7fddZ8ddZ9ddZ:ddZ;ddZdS)zCMethods to add arguments based on class/method/function signatures.N)SUPPRESSArgumentParser)AnyCallableOptionalUnion)_ActionConfigLoad)LoggerPropertyget_class_instantiatorget_generic_originget_unaliased_typeis_final_class is_subclassis_subclasses_disabled) Namespace) attrs_supportget_doc_short_descriptionis_attrs_classis_pydantic_model) ParamDataget_parameter_originsget_signature_parameters)ActionTypeHintLazyInitBaseClasscallable_instancesget_subclass_namesis_list_pathlike is_optionalnot_required_typessequence_origin_types)NoneTypeget_import_pathget_private_kwargsget_typehint_originiter_to_set_str)register_pydantic_typeSignatureArgumentsc@s@eZdZdZ        d(dedeededed eee e e efd ee eee fd ed ed edeefddZ      d)dededeededed ee eee fd ed edeefddZ      d)dedeededed ee eee fd ed edeefddZ        d*deededed ee eee fd ed ed edee edeedeefddZddddddefdeedeed ee ed eded ed edee ed efddZ      d+d eeeed!ffdeded ee ed ed"ed#edefd$d%Z    d,d&d'ZdS)-r'zTMethods to add arguments based on signatures to an :class:`ArgumentParser` instance.NTFtheclass nested_keyas_group as_positionaldefaultskip instantiate fail_untyped sub_configsreturnc  sXt|} tt| std|t|tttfs:t|t r$t|| s:t |j s3t |j r3t|| s:td|t | dddd\} } }|j|d||||| || | d }|rp[t|rb|dnd|}t|t rr|}n"t|j rt|}fdd |Dfd d |D}|r||rfd d |D}|jdi||S)aZAdds arguments from a class based on its type hints and docstrings. Note: Keyword arguments without at least one valid type are ignored. Args: theclass: Class from which to add arguments. nested_key: Key for nested namespace. as_group: Whether arguments should be added to a new argument group. as_positional: Whether to add required parameters as positional arguments. default: Default value used to override parameter defaults. skip: Names of parameters or number of positionals that should be skipped. instantiate: Whether the class group should be instantiated by ``instantiate_classes``. fail_untyped: Whether to raise exception if a required parameter does not have a type. sub_configs: Whether subclass type hints should be loadable from inner config file. Returns: The list of arguments added. Raises: ValueError: When not given a class. ValueError: When there are required parameters without at least one valid type. z7Expected 'theclass' parameter to be a class type, got: zPExpected 'default' to be dict, Namespace, lazy instance or dataclass-like, got: N)linked_targetshelprequired)r0r.r2r3.csh|] }|tdqSN)len.0kprefixL/home/ubuntu/.local/lib/python3.10/site-packages/jsonargparse/_signatures.py |sz9SignatureArguments.add_class_arguments..csg|]}|vr|qSr>r>r9)argsr>r? }sz:SignatureArguments.add_class_arguments..cs"i|] \}}|vr||qSr>r>)r:r;v)r=r-r>r? "z:SignatureArguments.add_class_arguments..r>)r inspectisclassr ValueError isinstancer!dictrrr __class__rr#_add_signature_argumentssetlazy_get_init_argsas_dictis_convertible_to_dictconvert_to_dictupdateitems set_defaults)selfr(r)r*r+r,r-r.r/r0kwargsunaliased_class_typer2help__ added_argsdefaultsskip_not_addedr>)rAr=r-r?add_class_arguments-sj#      z&SignatureArguments.add_class_arguments themethodc CsXt|} tt| stdt| |rtt| |std|j||||||||dS)aAdds arguments from a class based on its type hints and docstrings. Note: Keyword arguments without at least one valid type are ignored. Args: theclass: Class which includes the method. themethod: Name of the method for which to add arguments. nested_key: Key for nested namespace. as_group: Whether arguments should be added to a new argument group. as_positional: Whether to add required parameters as positional arguments. skip: Names of parameters or number of positionals that should be skipped. fail_untyped: Whether to raise exception if a required parameter does not have a type. sub_configs: Whether subclass type hints should be loadable from inner config file. Returns: The list of arguments added. Raises: ValueError: When not given a class or the name of a method of the class. ValueError: When there are required parameters without at least one valid type. z2Expected "theclass" argument to be a class object.zCExpected "themethod" argument to be a callable member of the class.r0) r rFrGr rHhasattrcallablegetattrrL) rUr(r^r)r*r+r-r/r0unaliased_typer>r>r?add_method_argumentss z'SignatureArguments.add_method_argumentsfunctionc CsLt|stdd}t|drt|jr|j}d}|j||||||||dS)aAdds arguments from a function based on its type hints and docstrings. Note: Keyword arguments without at least one valid type are ignored. Args: function: Function from which to add arguments. nested_key: Key for nested namespace. as_group: Whether arguments should be added to a new argument group. as_positional: Whether to add required parameters as positional arguments. skip: Names of parameters or number of positionals that should be skipped. fail_untyped: Whether to raise exception if a required parameter does not have a type. sub_configs: Whether subclass type hints should be loadable from inner config file. Returns: The list of arguments added. Raises: ValueError: When not given a callable. ValueError: When there are required parameters without at least one valid type. z5Expected "function" argument to be a callable object.NrK__call__r_)rarHr`rrKrL) rUrer)r*r+r-r/r0 method_namer>r>r?add_function_argumentss z)SignatureArguments.add_function_argumentsr2r3c  Cst|||jd} dd|pgD} | rMt| dks"tdd| Dr)td| dd | d | d D}| | d d } |jd |d | d dd|rT|dnd}| D]$}|rc|j|vrcqY||j|jvr}td|jd|d||jdqY| d ur| }nt|||jd}|rt ||n|}|j ||||t| d k| d}g}| D]}|j ||||dd |pgD||| |d q|S)aAdds arguments from parameters of objects based on signatures and docstrings. Args: function_or_class: Object from which to add arguments. method_name: Class method from which to add arguments. nested_key: Key for nested namespace. as_group: Whether arguments should be added to a new argument group. as_positional: Whether to add required parameters as positional arguments. skip: Names of parameters or number of positionals that should be skipped. fail_untyped: Whether to raise exception if a required parameter does not have a type. sub_configs: Whether subclass type hints should be loadable from inner config file. instantiate: Whether the class group should be instantiated by ``instantiate_classes``. Returns: The list of arguments added. Raises: ValueError: When there are required parameters without at least one valid type. loggercSs"g|] }t|tr|dkr|qS)r)rIintr:sr>r>r?rB rEz?SignatureArguments._add_signature_arguments..rcss|]}|dkVqdS)rNr>r:pr>r>r? z>SignatureArguments._add_signature_arguments..z*Unexpected number of positionals to skip: cSsh|]}|jqSr>namernr>r>r?r@sz>SignatureArguments._add_signature_arguments..NrzSkipping parameters z because z% positionals requested to be skipped.--r5r6zUnable to add parameter 'z' from z because argument 'z' already exists.) config_loadr.cSsh|] }t|tr|qSr>)rIstrrlr>r>r?r@6)r-r/r0r2r+) rrjr8anyrHdebugrs_option_string_actionsrrb_create_group_if_requested_add_signature_parameter)rUfunction_or_classrgr)r*r+r-r/r0r.r2r3paramsskip_positionalsnamesr=param doc_group component containerrZr>r>r?rLs^!   z+SignatureArguments._add_signature_argumentsrZc sP|j} |j}|j}| tkr#|j} | tkr#t|rd} nt|tvr#t} |t j kr-d}d}n*|t j kr9| tk}d}n|t j krE| tk}d}n|durP| tk}d}nt d|t|j|j}d| d|d}|sw|tkrwt}|rsdn| } d}d}|r| dur| | vrd} d}d}|s| ddkrdS|r| |vr|j|d dSt| r|jj| } |tkr|stt| tf}d | vr|j| d <|s| | d <| durt|ts|st|}n|r|rd| d <d}t|}|r|d nd| }|r|r|s|nd|g}|jr4|}t|tst |d}d!dd|jD}||j"vr)|j"|}n |j#d|d|d}|t$t%t&t'hvsIt(|t$t%t&fsI|rR|| d<t)|nn|tkrzKt*j+|dd}|| d<||d}|r| dfdd|pygD}||d<nt)||o|pt*,|pt-|}t*j.|| |||j|d}Wnt/y}z|j|t$|WYd}~nd}~wwd| vsd| vr||| d }|r| 0|t*1|j2|i| }Wdn 1swY|dur ||_3|r t4|dkr ||j3d<|5|dS|r$|r&t/d!| d"|d#dSdS)$NTFz'The code should never reach here: kind=zSkipping parameter "z" from "z" because of: rrYz"Parameter requested to be skipped.r3r,r4r5r6rtparserz; css|]}t|VqdSr7)rv)r:or>r>r?rprqz>SignatureArguments._add_signature_parameter..z Conditional arguments [origins: ]rrtype) all_subtypes)r/r0 .init_args.cs&h|]}|r|tdqSr7) startswithr8rlr<r>r?r@s&z>SignatureArguments._add_signature_parameter..r-)rArV enable_pathrrjsub_add_kwargsaction)r/r0r.zXWith fail_untyped=True, all mandatory parameters must have a supported type. Parameter 'z' from 'z' does not specify a type.)6rskind annotation inspect_emptyr,rr$rrkindsPOSITIONAL_ONLYPOSITIONAL_OR_KEYWORD KEYWORD_ONLY RuntimeErrorrrparentrrjryis_factory_class__dataclass_fields__default_factoryrrdocobjectrroriginrIrrbjoingroupsadd_argument_grouprvrkfloatboolrr&ris_subclass_typehintis_return_subclass_typehintrprepare_add_argumentrHrRallow_default_instance_context add_argumentrr8append) rUrr)rrZr-r/r+r0r.r2r,rVrsrr is_requiredis_non_positionalsrc skip_messageis_required_link_targetrsubclasses_disableddestrAr group_namer subclass_skiprexrr>r<r?r|?s              *            z+SignatureArguments._add_signature_parameter2CONFIG | CLASS_PATH_OR_NAME | .INIT_ARG_NAME VALUEeOne or more arguments specifying "class_path" and "init_args" for any subclass of %(baseclass_name)s. baseclass.r4metavarc  st|tur |f}t|tsJ|rtdd|Ds"td|d} t|dkr3t|d|jd} |j||| d|dd } g} |durNfd d |D}t t ||d } t t | j }| ||d |idd| vrrd| d<|j| d| | |fd|d| dS)aAdds arguments to allow specifying any subclass of the given base class. This adds an argument that requires a dictionary with a ``class_path`` entry which must be a import dot notation expression. Optionally any init arguments for the class can be given in the ``init_args`` entry. Since subclasses can have different init arguments, the help does not show the details of the arguments of the base class. Instead a help argument is added that will print the details for a given class path. Args: baseclass: Base class or classes to use to check subclasses. nested_key: Key for nested namespace. as_group: Whether arguments should be added to a new argument group. skip: Names of parameters that should be skipped. required: Whether the argument group is required. metavar: Variable string to show in the argument's help. help: Description of argument to show in the help. **kwargs: Additional parameters like in :meth:`add_class_arguments`. Raises: ValueError: When given an invalid base class. css|] }tj|ddVqdS)T) also_listsN)rr)r:cr>r>r?rpsz.zIExpected 'baseclass' to be a subclass type or a tuple of subclass types: NrrriF)rur4r.csh|] }d|qS)rr>rlr)r>r?r@rwz.)rsrrbaseclass_name)rr3r,T)r0r.)rtuplerIallrHr8rrjr{rrr%rrrRr|)rUrr)r*r-r.r4rr3rVrgrouprZr str_baseclassr>rr?add_subclass_argumentssH $     z)SignatureArguments.add_subclass_argumentsc Cs|r|dur td|j||} |rk|dur.t|tr*t|dkr*t|d}nt|}|dur6t|n|} |jt || d} |rS|durS| j d|t |ddt |rk|durk|rk|dd | _|| _t| _| S) Nz+A nested_key is mandatory to make required.rrrrrt)basetype)r-rY)rH required_argsaddrIrr8rvget_object_namer strip_titlerr rFrGreplacer group_classgroup_instantiate_classinstantiate_class) rUobjr)r*rruconfig_load_typer4r.rrsr>r>r?r{s&   z-SignatureArguments._create_group_if_requested)NTFNNTTF)NTFNTF)TFNTFTNN)TNTFrr)TNFT)__name__ __module__ __qualname____doc__rrrvrrrJrrrMrklistr]rdrrhrLrrr|rrr{r>r>r>r?r'*sF   ]  4 6     ]        Nr1cCs2t|dr|jSt|ddddddS)Nrr5[rYrr6)r`rrvsplitr)rr>r>r?r0s "rcCsXz ||j\}}}Wntyi}|}|j}Ynwt}||jfi|||<dSr7)get_value_and_parentrKeyErrorr r)rcfgvaluerkeyinstantiator_fnr>r>r?r6s  rcCs|dur tdd|}|S)Nz\.$r6)resubstriprr>r>r?rAsrcCs |jtjkSr7)rK dataclasses_HAS_DEFAULT_FACTORY_CLASSrr>r>r?rGs rcCst|p t|p t|Sr7)r is_dataclassrrrr>r>r?rPKsrPc Cstrddl}|t|r||St|}i}t|D]2\}}t|}t|r/t|}n|t vrL| }t |D]\}}tt|rKt|||<q;|||<qt |rW|St ||dS)Nr) class_path init_args)rattrshasrasdictvarsrSrPrQr copy enumeraterr") rr value_typerrsattr attr_typenumitemr>r>r?rQOs(     rQ)?rrrFrargparserrtypingrrrr_actionsr _commonr r r r rrr _namespacer _optionalsrrrr_parameter_resolversrrr _typehintsrrrrrrrr _utilr!r"r#r$r%r&__all___ParameterKindr_emptyrr'rvrrrrrPrJrQr>r>r>r?s8 $ (