o hR@sddlZddlZddlZddlZddlmZddlmZddlmZddlmZddlm Z ddlm Z ddlm Z dd lm Z dd lm Z dd lmZdd lmZdd lmZddlmZddlZddlmZddlmZddlmZddlmZddlmZddlmZerddlmZdZeGdddZ Gddde!Z"GdddZ#GdddZ$Gdd d ej%Z&Gd!d"d"ej'Z(dS)#N)gettext)Any)Callable)cast)Dict)List)Mapping)NoReturn)Optional)Sequence)Tuple) TYPE_CHECKING)Union)final) UsageError)ARGUMENT_PERCENT_DEFAULT)ARGUMENT_TYPE_STR)ARGUMENT_TYPE_STR_CHOICE)check_ispytest)Literal file_or_dirc @seZdZUdZdZeeed<  d/dddeedeedgdfd e d dfd d Z d0ddZ d1dededeed dfddZ dede d dfddZ d2deeedfdeejd ejfddZd3d!d"Z d2deeedfd ejdeejd eefd#d$Z d2deeedfdeejd ejfd%d&Z d2deeedfdeejd eejeeffd'd(Z  d/ded)ed*ed+d,e d df d-d.ZdS)4ParserzParser for command line arguments and ini-file values. :ivar extra_info: Dict of generic param -> value to display in case there's an error processing the command line arguments. NprogF _ispytestusage processoptArgumentrreturncCs@t|td|dd|_g|_||_||_i|_g|_i|_dS)NzCustom optionsTparserr) r OptionGroup _anonymous_groups _processopt_usage_inidict _ininames extra_info)selfrrrr*O/var/www/html/ai/venv/lib/python3.10/site-packages/_pytest/config/argparsing.py__init__+s zParser.__init__optioncCs"|jr |jr||dSdSdSN)r$dest)r)r-r*r*r+ processoption;s zParser.processoptionname descriptionafterr!cCsj|jD] }|j|kr|Sqt|||dd}d}t|jD] \}}|j|kr)nq|j|d||S)aGet (or create) a named option Group. :param name: Name of the option group. :param description: Long description for --help output. :param after: Name of another group, used for ordering --help output. :returns: The option group. The returned group object has an ``addoption`` method with the same signature as :func:`parser.addoption ` but will be shown in the respective group in the output of ``pytest --help``. Trr)r#r2r! enumerateinsert)r)r2r3r4groupigrpr*r*r+getgroup@s   zParser.getgroupoptsattrscOs|jj|i|dS)a!Register a command line option. :param opts: Option names, can be short or long options. :param attrs: Same attributes as the argparse library's :py:func:`add_argument() ` function accepts. After command line parsing, options are available on the pytest config object via ``config.option.NAME`` where ``NAME`` is usually set by passing a ``dest`` attribute, for example ``addoption("--long", dest="NAME", ...)``. N)r" addoption)r)r<r=r*r*r+r>ZszParser.addoptionargszos.PathLike[str] namespacecCs>ddlm}||_||jdd|D}|jj||dS)Nr)try_argcompletecSg|]}t|qSr*osfspath.0xr*r*r+ sz Parser.parse..r@)_pytest._argcompleterA _getparser optparser parse_args)r)r?r@rAstrargsr*r*r+parsejs   z Parser.parseMyOptionParserc Csddlm}t||j|jd}|j|jg}|D]&}|jr>|jp"|j }| |}|jD]}| }| } |j |i| q+q|j tdd} || _|S)Nr)filescompleter)r*)nargs)rLrSrRr(rr#r"optionsr3r2add_argument_groupnamesr= add_argument FILE_OR_DIR completer) r)rSrNgroupsr8descarggroupr-nafile_or_dir_argr*r*r+rMvs    zParser._getparsercCsB|j||d}|jD] \}}t|||q tttt|tS)NrK) rQ__dict__itemssetattrrrstrgetattrrZ)r)r?r-r@ parsedoptionr2valuer*r*r+parse_setoptionszParser.parse_setoptioncCs|j||ddS)zbParse the known arguments at this point. :returns: An argparse namespace object. rKr)parse_known_and_unknown_args)r)r?r@r*r*r+parse_known_argss zParser.parse_known_argscCs$|}dd|D}|j||dS)aParse the known arguments at this point, and also return the remaining unknown arguments. :returns: A tuple containing an argparse namespace object for the known arguments, and a list of the unknown arguments. cSrBr*rCrFr*r*r+rIrJz7Parser.parse_known_and_unknown_args..rK)rMrk)r)r?r@rNrPr*r*r+rjs z#Parser.parse_known_and_unknown_argshelptypezBLiteral['string', 'paths', 'pathlist', 'args', 'linelist', 'bool']defaultcCs,|dvsJ|||f|j|<|j|dS)aRegister an ini-file option. :param name: Name of the ini-variable. :param type: Type of the variable. Can be: * ``string``: a string * ``bool``: a boolean * ``args``: a list of strings, separated as in a shell * ``linelist``: a list of strings, separated by line breaks * ``paths``: a list of :class:`pathlib.Path`, separated as in a shell * ``pathlist``: a list of ``py.path``, separated as in a shell .. versionadded:: 7.0 The ``paths`` variable type. Defaults to ``string`` if ``None`` or not passed. :param default: Default value if no ini-file option exists but is queried. The value of ini-variables can be retrieved via a call to :py:func:`config.getini(name) `. )Nstringpathspathlistr?linelistboolN)r&r'append)r)r2rlrmrnr*r*r+addinis !z Parser.addiniNN)r-rrNr1Nr.)rrR)__name__ __module__ __qualname____doc__rr re__annotations__rrsr,r0r;rr>r rargparse NamespacerQrMrrirkr rjrur*r*r*r+r!s       rc@s<eZdZdZdededefddfddZdefd d ZdS) ArgumentErrorzURaised if an Argument instance is created with invalid or inconsistent arguments.msgr-rrNcCs||_t||_dSr.)rre option_id)r)rr-r*r*r+r,szArgumentError.__init__cCs |jr d|jd|jS|jS)Nzoption : )rrr)r*r*r+__str__szArgumentError.__str__)rxryrzr{rerr,rr*r*r*r+rsrc@seZdZdZeeeedZdede ddfddZ de efd d Z de ee ffd d Zd eeddfddZdefddZdS)rzClass that mimics the necessary behaviour of optparse.Option. It's currently a least effort implementation and ignoring choices and integer prefixes. https://docs.python.org/3/library/optparse.html#optparse-standard-option-types )introfloatcomplexrXr=rNc Osz||_g|_g|_d|dpdvrtjtddz|d}Wn ty(Yn=wt|t rb|dkrItjt j ||dd dt |d d |d<ntjt j ||dd dtj||d<|d|_ n||_ z|d |_Wn tyuYnw|||d }|r||_dS|jr|jd dddd|_dSz |jd dd|_WdSty}z d|_td||d}~ww)z5Store params in private vars for use in add_argument.%defaultrlr1) stacklevelrmchoice)typrXchoicesrrnr/N-_r5z???zneed a long or short option)_attrs _short_opts _long_optsgetwarningswarnrKeyError isinstancererformatrmrr_typ_maprn_set_opt_stringsr/replace IndexErrorr)r)rXr=rr/er*r*r+r,sR           zArgument.__init__cCs |j|jSr.)rrrr*r*r+rX s zArgument.namesc Csvd}||j|D]}z t|||j|<Wq ty!Yq w|jdr8|jd}|dd}||jd<|jS)Nzdefault dest helprlrz %(default)s)splitrtr/rfrAttributeErrorrr)r)r=attrr`r*r*r+r=#s      zArgument.attrsr<cCs|D]J}t|dkrtd||t|dkr1|ddkr#|ddks*td|||j|q|dddkr?|ddksFtd|||j|qd S) zhDirectly from optparse. Might not be necessary as this is passed to argparse later on. rz>invalid option string %r: must be at least two characters longrrr5zMinvalid short option string %r: must be of the form -x, (x any non-dash char)--zGinvalid long option string %r: must start with --, followed by non-dashN)lenrrrtr)r)r<optr*r*r+r3s4  zArgument._set_opt_stringscCsg}|jr|dt|jg7}|jr|dt|jg7}|dt|jg7}t|dr5|dt|jg7}t|drD|dt|jg7}dd |S) Nz _short_opts: z _long_opts: zdest: rmztype: rnz default: z Argument({}), ) rreprrr/hasattrrmrnrjoin)r)r?r*r*r+__repr__Ps  zArgument.__repr__)rxryrzr{rrerrrrr,rrXrr=r rrr*r*r*r+rs1rc @seZdZdZ  ddddededeed ed df d d Zd ede d dfddZ d ede d dfddZ dddded dfddZ dS)r!z,A group of options shown in its own section.r1NFrr2r3r rrcCs$t|||_||_g|_||_dSr.)rr2r3rVr )r)r2r3r rr*r*r+r,as  zOptionGroup.__init__r<r=cOsJt|dd|jD}|rtd|t|i|}|j|dddS)aMAdd an option to this group. If a shortened version of a long option is specified, it will be suppressed in the help. ``addoption('--twowords', '--two-words')`` results in help showing ``--two-words`` only, but ``--twowords`` gets accepted **and** the automatic destination is in ``args.twowords``. :param opts: Option names, can be short or long options. :param attrs: Same attributes as the argparse library's :py:func:`add_argument() ` function accepts. css"|] }|D]}|VqqdSr.)rX)rGrr2r*r*r+ }s  z(OptionGroup.addoption..zoption names %s already addedF shortupperN)set intersectionrV ValueErrorr_addoption_instance)r)r<r=conflictr-r*r*r+r>os zOptionGroup.addoptioncOs t|i|}|j|dddS)NTr)rr)r)r<r=r-r*r*r+ _addoptionszOptionGroup._addoptionr-rrcCsR|s|jD]}|ddkr|drtdq|jr!|j||j|dS)Nrrr5zlowercase shortoptions reserved)rislowerrr r0rVrt)r)r-rrr*r*r+rs  zOptionGroup._addoption_instancerw)F) rxryrzr{rer rrsr,rr>rrr*r*r*r+r!^s( r!c seZdZ  ddedeeeefdeeddffdd Zdede fd d Z  dd ee ed ee j de j fd dZejdddkr_dedeeee jeeeffddZZSZS)rRNr r(rrcs6||_tj||jdtdd|r||_dSi|_dS)NF)rradd_helpformatter_class allow_abbrev)_parsersuperr,r%DropShorterLongHelpFormatterr()r)r r(r __class__r*r+r,s zMyOptionParser.__init__messagecCs@|jd|}t|jdr|d|jjd}t||)z1Transform argparse error message into UsageError.z : error: _config_source_hintz ())rrrrr format_usage)r)rrr*r*r+errors zMyOptionParser.errorr?r@c Cs|||\}}|rF|D]1}|r=|ddkr=dd|g}t|jD]\}}|d|d|q%|d|q t|t ||S)z(Allow splitting of positional arguments.rrzunrecognized arguments: %s z r ) rkrsortedr(rcrtrrfrZextend) r)r?r@parsed unrecognizedarglineskvr*r*r+rOszMyOptionParser.parse_argsr)r arg_stringc Cs|sdS|d|jvr dS||jvr|j|}||dfSt|dkr$dSd|vr?|dd\}}||jvr?|j|}|||fS|jsG|dsv||}t|dkrktd}ddd|D}| |||d n t|dkrv|\}|S|j |r|j sdSd |vrdSd|dfS) Nrr5=rz4ambiguous option: %(option)s could match %(matches)srcss|]\}}}|VqdSr.r*)rGrr-r*r*r+rsz1MyOptionParser._parse_optional..)r-matchesr) prefix_chars_option_string_actionsrrr startswith_get_option_tuplesrrr_negative_number_matchermatch_has_negative_number_optionals) r)raction option_string explicit_arg option_tuplesrrV option_tupler*r*r+_parse_optionals>            zMyOptionParser._parse_optionalrv)rxryrzrr rrerr,r rr r}r~rOsys version_infor Actionr __classcell__r*r*rr+rRs:   rRcsNeZdZdZdededdffdd Zdejdeffd d Z d d Z Z S) ra+Shorten help for long options that differ only in extra hyphens. - Collapse **long** options that are the same except for extra hyphens. - Shortcut if there are only two options and one of them is a short one. - Cache result on the action object as this is called at least 2 times. r?kwargsrNcs,d|vr tj|d<tj|i|dS)Nwidth)_pytest_ioget_terminal_widthrr,)r)r?rrr*r+r,sz%DropShorterLongHelpFormatter.__init__rc sft|}|r|ddkr|St|dd}|r|S|d}t|dkr:t|ddks5t|ddkr:||_|Sg}i}|D]9}t|dksN|ddkrOq@|ds[td |||dd}|dd } | |vsut|| t|kry||| <q@|D],}t|dks|ddkr| ||dd| |dd kr| |dd dq|d |} | |_| S) Nrr_formatted_action_invocationrrr5rrz)long optional argument without "--": [%s]r1r) r_format_action_invocationrfrrrrrrrtrr) r)rorgstrresrV return_list short_longr-xxoption shortenedformatted_action_invocationrr*r+rsF   ,     z6DropShorterLongHelpFormatter._format_action_invocationcCs4ddl}g}|D] }||||q |S)z}Wrap lines after splitting on original newlines. This allows to have explicit line breaks in the help text. rN)textwrap splitlinesrwrapstrip)r)textrrrliner*r*r+ _split_liness  z)DropShorterLongHelpFormatter._split_lines) rxryrzr{rr,r}rrerrrr*r*rr+rs &r))r}rDrrrtypingrrrrrrr r r r r r _pytest._ior_pytest.compatr_pytest.config.exceptionsr_pytest.deprecatedrrrrtyping_extensionsrrZr Exceptionrrr!ArgumentParserrR HelpFormatterrr*r*r*r+sF                    4z5V