k jzrddlmZmZmZmZmZmZmZmZm Z m Z m Z ddl Z ddl mZddlmZddlmZddlmZmZe rdd lmZmZGd d eZd ed e edfddfdZd edeedeegeffdZd ededeegeffdZGddeZGddZde jdefdZ dS)) AnyCallableDictIterableList NamedTupleOptionalSequence TYPE_CHECKINGTupleUnionN) Constraint)join_param_labels) first_bool) DecoratorF) HelpFormatter OptionGroupcHeZdZUdZeed<eeeed<d dZ d S) BoundConstraintSpecz^A NamedTuple storing a ``Constraint`` and the **names of the parameters** it has to check. constraint param_namescmdConstraintMixinreturnBoundConstraintc\t|j||jSN)rrget_params_by_namer)selfrs U/home/agentuser/manim-venv/lib/python3.11/site-packages/cloup/constraints/_support.pyresolve_paramsz"BoundConstraintSpec.resolve_paramss- O  " "4#3 4 4   N)rrrr) __name__ __module__ __qualname____doc__r__annotations__r r strr$r%r#rrsUx}%%%%      r%rfconstrrrcht|dsg|_|j|dS)N__cloup_constraints__)hasattrr0append)r-r.s r#_constraint_memor3s< 1- . .%"$""6*****r%paramscjt|t|dtdtffd }|S)zRegister a constraint on a list of parameters specified by (destination) name (e.g. the default name of ``--input-file`` is ``input_file``).r-rc(t||Sr )r3)r-specs r# decoratorzconstraint..decorator+sD!!!r%)rtupler)r.r4r8r7s @r#rr&sN vuV}} 5 5DQ1 r% param_addersc4dtdtffd }|S)a Return a decorator that adds the given parameters and applies a constraint to them. Equivalent to:: @param_adders[0] ... @param_adders[-1] @constraint(constr, ) This decorator saves you to manually (re)type the parameter names. It can also be used inside ``@option_group``. Instead of using this decorator, you can also call the constraint itself:: @constr(*param_adders) but remember that: - Python 3.9 is the first that allows arbitrary expressions on the right of ``@``; - using a long conditional/composite constraint as decorator may be less readable. In these cases, you may consider using ``@constrained_params``. .. versionadded:: 0.9.0 :param constr: an instance of :class:`Constraint` :param param_adders: function decorators, each attaching a single parameter to the decorated function. r-rc g}tD]/}|||jd}||0tt |ddd}t |||S)N)reversed__click_params__r2rr9r3)r-reversed_params add_paramparam bound_constrr.r:s r#r8z%constrained_params..decoratorVs!,// * *I IaLLL&r*E  " "5 ) ) ) )&vu_TTrT5J/K/KLL L)))r%)r)r.r:r8s`` r#constrained_paramsrD2s>HQ1 r%ceZdZUdZeed<eejed<d dZ dej ddfdZ dej de e eeffd ZdS) rzInternal utility ``NamedTuple`` that represents a ``Constraint`` bound to a collection of ``click.Parameter`` instances. Note: this is not a subclass of Constraint.rr4rNcD|j|jdSr )rcheck_consistencyr4)r"s r#rGz!BoundConstraint.check_consistencyks  ))$+66666r%ctxcF|j|j|dSr )r check_valuesr4r"rHs r#rJzBoundConstraint.check_valuesns" $$T[#66666r%ct|j|}|sdSdt|jz}||fS)Nz{%s})rhelprr4)r"rH constr_help param_lists r#get_help_recordzBoundConstraint.get_help_recordqsDo**3//  4/ <<< ;&&r%)rN)r&r'r(r)rr*r click ParameterrGContextrJr r r+rPr,r%r#rrcs33 U_ %%%%77777 7$7777'5='XeCHo5N''''''r%c "eZdZdZddddedeeeefde e deffd Z d e j deed eeffd Zd ed e jfdZdeed ee jfdZd e j ddd dfdZd e j d e fdZxZS)rz!Provides support for constraints.r,N) constraintsshow_constraintsargsrUrVkwargscXtj|i||_djD_t dt }t d|D_ t fd|D_ jjz_ dS)a :param constraints: sequence of constraints bound to specific groups of parameters. Note that constraints applied to option groups are collected from the option groups themselves, so they don't need to be included in this argument. :param show_constraints: whether to include a "Constraint" section in the command help. This is also available as a context setting having a lower priority than this attribute. :param args: positional arguments forwarded to the next class in the MRO :param kwargs: keyword arguments forwarded to the next class in the MRO ci|] }|j| Sr,)name).0rBs r# z,ConstraintMixin.__init__..s)< < < "'EJ< < < r% option_groupsc3XK|]%}|j t|j|jV&dSr )rroptions)r\grps r# z+ConstraintMixin.__init__..sD* * ~) CNCK 8 8))))* * r%c3pK|]0}t|tr|n|V1dSr ) isinstancerr$)r\r.r"s r#rbz+ConstraintMixin.__init__..seD D %V_==1**400D D D D D D r%N) super__init__rVr4_params_by_namegetattrr9optgroup_constraintsparam_constraintsall_constraints)r"rUrVrWrXr^ __class__s` r#rfzConstraintMixin.__init__|s* $)&))) 0< < +/;< < <  29PUPWPW1X1X $)* * $* * * % % ! @?DD D D D & D D D ? ?  S#84;QQQQr%rHrc|js2tj|r|jD]}|t |t|jtj otfd|j D}|js|rS|jD]}| |S)Nc3 K|]}|vV dSr r,)r\ help_flagrWs r#rbz-ConstraintMixin.parse_args..sAS S "+I S S S S S S r%)resilient_parsingrmust_check_consistencyrkrGre parse_argsrdcommandrQGroupanyhelp_option_namesrJ)r"rHrWr.should_show_subcommand_helprls ` r#rrzConstraintMixin.parse_argss$ +)J3)O)O +. + +((****ww!!#t,,'1ek&J&J' sS S S S /2/DS S S P P #   $? K* % %F    $ $ $ $ r%r[c` |j|S#t$rtd|dwxYw)Nz there's no CLI parameter named '')rgKeyError)r"r[s r#get_param_by_namez!ConstraintMixin.get_param_by_namesO G'- - G G GEdEEEFF F Gs -namesc:tfd|DS)Nc3BK|]}|VdSr )r{)r\r[r"s r#rbz5ConstraintMixin.get_params_by_name..s1DDdT++D11DDDDDDr%)r9)r"r|s` r#r!z"ConstraintMixin.get_params_by_names&DDDDeDDDDDDr% formatterrcfd|jD}d|D}|rE|d5||ddddS#1swxYwYdSdS)Nc3BK|]}|VdSr )rP)r\r.rHs r#rbz5ConstraintMixin.format_constraints..s1XXvv--c22XXXXXXr%cg|]}||Sr r,)r\recs r# z6ConstraintMixin.format_constraints..sAAA33r% Constraints)rjsectionwrite_dl)r"rHr records_genrecordss ` r#format_constraintsz"ConstraintMixin.format_constraintssXXXXAWXXX AA+AAA  ,""=11 , ,""7+++ , , , , , , , , , , , , , , , , , , , ,sAA"AcLt|jt|dddS)NrVF)rrVrhrKs r#must_show_constraintsz%ConstraintMixin.must_show_constraintss-  ! C+T 2 2    r%)r&r'r(r)rr r rrr boolrfrQrSrr+rrrRr{rr!rr __classcell__)rls@r#rrys++NP+/3R3R3R3Re$7$HIJ3R#4.3R 3R3R3R3R3R3Rjem49c*GcGeoGGGG E E(5?:SEEEE,em,,TX,,,,  4        r%rrscNt|tr|Std)NzCa Command must inherits from ConstraintMixin to support constraints)rdr TypeError)rss r#ensure_constraints_supportrs1'?++ M O OOr%)!typingrrrrrrr r r r r rQ_corercommonr_utilrrrclouprrrr3r+rrDrrCommandrr,r%r#rse  %%%%%%!!!!!!!!100000000      *    + +-/@@A+ ++++ z 8C= Xqc1f=M    . ..qc1f....b'''''j''',c c c c c c c c LO O/OOOOOOr%