Kj_ <UddlmZddlZddlmZddlmZddlZddlZddlZddl Z ddl Z ddl Z ddl m Z ddl mZddl mZddl mZdd l mZdd l mZdd l mZdd l mZdd l mZddlmZddlZddlmZddlmZerddlmZdZ edZ!de"d<ere!ee#e$ze%ze&zzdzZ'de"d<eejPdGddZ)e!e)ze*e!e)zdfzZ+de"d<ee!Dcic]}||c}e&de#de$d e%d!izZ,d"e"d#<d7d$Z-d8d%Z.d9d&Z/d:d'Z0eGd(d)Z1d;d*Z2Gd+d,Z3Gd-d.Z4Gd/d0ejjZ6Gd1d2ejnZ8ejrd3Z:dz#_ini_type_to_tag..MsDscDzp, one of the types str, bool, int, float, a union of these types such as `int | str`, or a `Literal` of strings)N)r-KeyError TypeError ValueErrorjoinrr)nametype_s r+_ini_type_to_tagr@Fsm%  i *4("UI=OyyDh{.CDDEF6 6   s  A Act|tur9t|}td|Dst d|d|t |St ||S)z7Normalize one member of an `addini(type=...)` argument.c3<K|]}t|tywr3) isinstancestrr6choices r+r8z&_ini_type_to_member..WsAv:fc*Asr/z': Literal choices must be strings, got )r r rallr<r!r@)r>r?r#s r+_ini_type_to_memberrHSsc%G#5/AAA.th7((/{4 7## D% ((r*ct|tr"djd|jDSt|trdjd|DS|S)z?Render an ini option type for --help output and error messages.z | c32K|]}t|ywr3r4rEs r+r8z!_ini_type_repr..csB6$v,Br9c32K|]}t|ywr3)_ini_type_repr)r6members r+r8z!_ini_type_repr..esDV.0Dr9)rCr!r=r#tupletypes r+rLrL`sI$ $zzBT\\BBB$zzDtDDD Kr*c|Dcgc]}|jds|}}|r|dn|d}|jdjddScc}w)N--r-_) startswithlstripreplace)optsopt long_optss r+_get_argparse_destr[isQ $=t(<=I=#)A,aC ::c? " "3 ,,>s AAceZdZdZ dddd ddZeddZejddZddZ d dd Z dd Z d dd Z d dd Z d dd Z defdd ddZy)ParserzParser for command line arguments and config-file values. :ivar extra_info: Dict of generic param -> value to display in case there's an error processing the command line arguments. NF)prog _ispytestcxt|ddlm}||_i|_t ||j||_|j jd}t|d|d|_ |jg|_ i|_ |j jtd }||_i|_i|_y) Nr)filescompleterr^zCustom options _anonymousTr_*)nargs)r_pytest._argcompletera _processopt extra_infoPytestArgumentParser optparseradd_argument_group OptionGrouprc_groups _opt2dest add_argument FILE_OR_DIR completer_inidict _ini_aliases)selfusage processoptr^r_raanonymous_arggroupfile_or_dir_args r+__init__zParser.__init__ws y!7%*,-eT__4P!^^>>?OP%  dd ( )+..55k5M$2!=? ,.r*c.|jjSr3rkr^rus r+r^z Parser.progs~~"""r*c&||j_yr3r|)ruvalues r+r^z Parser.progs#r*cZ|jr|jr|j|yyyr3)rhdest)ruoptions r+ processoptionzParser.processoptions*   {{  ( r*c|jD]}|j|k(s|cS|jj|xs|}t |||d}d}t |jD]\}}|j|k(sn|jj |dz||jjj |dz|jjj|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``. Trdrr) rnr>rkrlrm enumerateinsert_action_groupspop)rur> descriptionaftergrouparggroupigrps r+getgroupzParser.getgroups\\ EzzT!  >>44[5HDIHdDDA  - FAsxx5   AE5) %%,,QUDNN4Q4Q4U4U4WX r*c<|jj|i|y)aRegister a command line option. :param opts: Option names, can be short or long options. :param attrs: Same attributes as the argparse library's :meth:`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)rc addoptionrurXattrss r+rzParser.addoptions "!!4151r*cddlm}||j|Dcgc]}tj|}}|t j } d|_|jj|||`Scc}w#|`wxYw)zParse the arguments. Unlike ``parse_known_args`` and ``parse_known_and_unknown_args``, raises PrintHelp on `--help` and UsageError on unknown flags :meta private: r)try_argcompleteT namespace) rgrrkosfspathargparse Namespace_raise_print_helpparse_intermixed_args)rurrrxstrargss r+parsez Parser.parses{ 9')-.A299Q<..   **,I ,*.I '>>7797U+/+sA8#A==Bc.|j||dS)zbParse the known arguments at this point. :returns: An argparse namespace object. rr)parse_known_and_unknown_args)rurrs r+parse_known_argszParser.parse_known_argss000KANNr*c|Dcgc]}tj|}}tjdksdtjcxkrdkrgnnd|jj ||\}}|Jt |t}g}|D]&}|jdr|n|j|(||fS|jj||Scc}w)a Parse the known arguments at this point, and also return the remaining unknown flag arguments. :returns: A tuple containing an argparse namespace object for the known arguments, and a list of unknown flag arguments. ) )r )rrrrS) rrsys version_inforkrgetattrrqrUappendparse_known_intermixed_args) rurrrrunknownr unknown_flagsargs r+rz#Parser.parse_known_and_unknown_argss*..A299Q<..   j (Gs7G7G,T*,T!%!@!@)!T Iw( ((!)[9K')M T"%.."5;NNsS Tm+ +>>==gyQ Q/sCr))aliasesc(|d}nKt|ttjfvrt fdt |D}n t |}|turGt|ttfr&t|trdnd}tdd|dt|}|||f|j<|D][}||jvrtd |d |jj|x} t|d | |j|<]y) a Register a configuration file option. :param name: Name of the configuration. :param type: Type of the configuration. 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 * ``int``: an integer * ``float``: a floating-point number .. versionadded:: 8.4 The ``float`` and ``int`` types. For the scalar types, the plain Python type may be passed instead of the string tag: ``str``, ``bool``, ``int`` and ``float`` (for example ``type=int``). A union of these types accepts a value of any of its members, for example ``int | str``. In TOML configuration files the value may then be any of the member types; string-based formats (INI files, ``-o`` overrides) coerce it to the first member that accepts it. A ``Literal`` type of strings restricts the value to the given choices, for example ``Literal["auto", "long", "short"]``, and may also be a union member, for example ``int | Literal["auto"]``. Since the choices have no unambiguous implicit default, an explicit ``default`` must be passed. .. versionadded:: 9.2 Passing a type expression such as ``int``, ``int | str``, or a ``Literal`` of strings. For ``paths`` and ``pathlist`` types, they are considered relative to the config-file. In case the execution is happening without a config-file defined, they will be considered relative to the current working directory (for example with ``--override-ini``). .. versionadded:: 7.0 The ``paths`` variable type. .. versionadded:: 8.1 Use the current working directory to resolve ``paths`` and ``pathlist`` in the absence of a config-file. Defaults to ``string`` if ``None`` or not passed. :param default: Default value if no config-file option exists but is queried. :param aliases: Additional names by which this option can be referenced. Aliases resolve to the canonical name. .. versionadded:: 9.0 The ``aliases`` parameter. The value of configuration keys can be retrieved via a call to :py:func:`config.getini(name) `. Nrc36K|]}t|ywr3)rH)r6rMr>s r+r8z Parser.addini..Ws6<#D&1sunionr z ini option z has a zL type, which has no implicit default; pass an explicit `default` to `addini`zalias z- conflicts with existing configuration optionz is already an alias of )r rtypes UnionTyperNrrHrrCr!r<get_ini_default_for_typersrtget) rur>helprPdefaultrini_typekindaliasalreadys ` r+addiniz Parser.addini s7P <H  %!9 9@HH+46H f (UK$89",Xu"=w9 !$7OO/x8G#Xw7 d ,E % UI%RS ,,0077D E9,DWK!PQQ'+D  e $ ,r*)NN) rv str | Nonerwz!Callable[[Argument], None] | Noner^rr_rreturnNonerrD)rrDrr)rArgumentrr)N)r>rDrrDrrrrmrXrDrrrrr3)r Sequence[str | os.PathLike[str]]rargparse.Namespace | Nonerargparse.Namespace)rrrrrz$tuple[argparse.Namespace, list[str]]) r>rDrrDrPrrrr Sequence[str]rr)r$r%r&r'rzpropertyr^setterrrrrrrrrr)r*r+r]r]os!8</   //6/  /  / /:## [[$$) EI&)7A >2&04,.,-,  ,604 O. O- O  O04R.R-R . R:! b,"$b,b,b, b,  b,b, b,r*r]c4|dvrgS|dk(ry|dk(ry|dk(ryy) zs Used by addini to get the default value for a given config option type, when default is not supplied. )rrrrrFrrrgrr)rOs r+rrqs3  88   r*cfeZdZdZd dZd dZd dZed dZeddZ eddZ d dZ y )rz$An option defined in an OptionGroup.c||_yr3)_action)ruactions r+rzzArgument.__init__s  r*c.|jjSr3)r__dict__r}s r+rzArgument.attrss||$$$r*c.|jjSr3)roption_stringsr}s r+nameszArgument.namess||***r*c.|jjSr3)rrr}s r+rz Argument.dest||   r*c.|jjSr3)rrr}s r+rzArgument.defaults||###r*c.|jjSr3)rrPr}s r+rPz Argument.typerr*cft|dd}|yg}|dt|jzgz }|dt|jzgz }|jr|dt|jzgz }|dt|j zgz }dj dj|S) NrzArgument()zopts: zdest: ztype: z default: z Argument({})r1)rr5rrrPrformatr=)rurrs r+__repr__zArgument.__repr__sy$/ >. D..// DO+,, ;; XTYY/0 0D tDLL1122$$TYYt_55r*N)rargparse.Actionrr)rdict[str, Any])rrr)rr)rz Any | None) r$r%r&r'rzrrrrrrPrr)r*r+rrsU.%+!!$$!! 6r*rcVeZdZdZ d ddZd dZd dZ d dZy) rmz,A group of options shown in its own section.cRt|||_||_g|_||_yr3)r _arggroupr>optionsparser)rurr>rr_s r+rzzOptionGroup.__init__s( y!! ')  r*c t|jd|jD}|rtd|d|jrpd|vrlt |}|jj D]H}|jD]7}|j|k(std|d|jd|d|d J|j||d y ) aAdd 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. Note that lower-case short options (e.g. `-x`) are reserved. :param attrs: Same attributes as the argparse library's :meth:`add_argument() ` function accepts. c3JK|]}|jD]}|ywr3)r)r6rYr>s r+r8z(OptionGroup.addoption..s-* SYY[* -1D* * s!#z option names z already addedrz option dest z already used by z' (this is the option that maps to dest z ); pass dest=z$ explicitly to share the destinationFallow_reservedN) set intersectionrr<rr[rnrr_addoption_inner)rurXrconflictrrrs r+rzOptionGroup.addoptionst9))*  LL*   }XJnEF F ;;6.%d+D,, #mmF{{d*(*4(2C%||~01$$(8=A77  dE%@r*c,|j||dy)z}Like addoption(), but also allows registering short lower case options (e.g. -x), which are reserved for pytest core.TrN)rrs r+ _addoptionzOptionGroup._addoptions dE$?r*c|s=|D]8}t|dk\s|ddk(s|djs/td|jj|i|}t |}|j j||jrT|jD]%}|j|jj|<'|jj|yy)NrrSrz$lowercase short options are reserved) lenislowerr<rrprrrrrrror)rurXrrrYrrr>s r+rzOptionGroup._addoption_inners Ms8q=SVs]s1v~~7G$%KLL M-,,drDrz Parser | Noner_rrrr)rXr"rrrrrr)r$r%r&r'rzrrrr)r*r+rmrmsq6 )        AD@ .#.,:.LP. .r*rmc<eZdZdd dfdZddZxZS)rjNrbcFt|||dtdd||_y)NF@)rvr^add_helpformatter_class allow_abbrevfromfile_prefix_chars)superrzDropShorterLongHelpFormatterri)rurvrir^ __class__s r+rzzPytestArgumentParser.__init__s2 8"%  %r*c |jd|}|jr?|ddjdt|jj Dzz }t |j |z)z1Transform argparse error message into UsageError.z : error:  c32K|]\}}d|d|yw)z r0Nr))r6kvs r+r8z-PytestArgumentParser.error.. s%$"&!Q"QCr! $r9)r^rir=sorteditemsr format_usage)rumessagemsgs r+errorzPytestArgumentParser.errorsu 9WI. ?? 4$))$*01F1F1H*I$ C**,s233r*)rvrrizdict[str, str]r^rrr)rrDrr )r$r%r&rzr __classcell__rs@r+rjrjs8  %%#%  %  %& 4r*rjcDeZdZdZdfd Zdfd ZddZdfd ZxZS) rzShorten 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. cnd|vr!tjj|d<t||i|y)Nwidth)_pytest_ioget_terminal_widthrrz)rurkwargsrs r+rzz%DropShorterLongHelpFormatter.__init__s3 & %kk<<>F7O $)&)r*ct ||}|r |ddk7r|S|jd}t|dk(r$t|ddk(st|ddk(r|Sg}i}|D]p}t|dk(s|ddk(r|j ds Jd|d |dd}|j dd }||vst||t|ksl|||<r|D]s}t|dk(s|ddk(r|j ||dd|j|j dd k(sR|j |j dd dudj|S) NrrSr1rr rRz&long optional argument without "--": []r=) r_format_action_invocationsplitrrUrWrrr=) rurorgstrr return_list short_longrxxoption shortenedrs r+rz6DropShorterLongHelpFormatter._format_action_invocations26: fQi3&M,,t$ w<1 #gaj/Q"6#gaj/Q:NM %'  1F6{a6!9#3$$T* 8B *abzH ((b1I *c*Y2G.H3L/)1 9% 1 @F6{a6!9#3""6*abzZ^^FNN3,CDD""6>>#sA#>?  @ yy%%r*ct||Sr3)_split_help_text)rutextr s r+ _split_linesz)DropShorterLongHelpFormatter._split_lines;se,,r*cltjddt| |tjS)Nz[ \t]+$r)flags)resubr_format_action MULTILINE)rurrs r+r$z+DropShorterLongHelpFormatter._format_action>s-vv EG26:",,  r*)rrrrrr)rrrrDrrDr rrz list[str]) r$r%r&r'rzrrr$rr s@r+rrs! * &>-  r*rz^([-*+]|\(?\d+[.)])\s+c 2|jd\}}}g}|j|ztj|zj D]}|j }|s|j d'|dt|t|z }tj|} | r|dt| jzzn|} |jtj|||| d|r|ds |d=|r|ds |r|ds |d=|r|ds |S) aWWrap help text to ``width`` while preserving its line structure. Explicit line breaks are kept, so that a help text can use paragraphs and lists instead of being reflowed into one blob (see #6817). Each line keeps its own indentation, and the continuation lines of a list item are indented to hang under the item's text. rrNrF)initial_indentsubsequent_indentbreak_on_hyphensr) partitionstriptextwrapdedent splitlinesrVrr_BULLETmatchrextendwrap) rr firstnewlinerestlineslinestrippedindentbulletr)s r+rrJs >>$/E7DE(8??4+@@LLN ;;= LL  1D CM12x(BHFS3v||~+>%>>f MM%"3!&    & a !H a b "I b Lr*cReZdZdZ d dfd Z ddZxZS)OverrideIniActionzCustom argparse action that makes a CLI flag equivalent to overriding an option, in addition to behaving like `store_true`. This can simplify things since code only needs to inspect the config option and not consider the CLI flag. cLt|||dg|i|||_||_y)Nr)rrz ini_option ini_value) rurrrfr@rArrrs r+rzzOverrideIniAction.__init__ys. qB4B6B$"r*ct||jdt|dd}|g}|j|jd|j t|d|y)NT override_inir)setattrrrrr@rA)rurrrrcurrent_overridess r+__call__zOverrideIniAction.__call__s_  499d+#I~tD  $ "   DOO#4Adnn5E!FG >+<=r*r3) rrrrDrfzint | str | Noner@rDrArDrr)rzargparse.ArgumentParserrrrr)r$r%r&r'rzrFrr s@r+r>r>qsm#' #% # # #  # #  # >' >& >  >r*r>)r>rDr?objectrr)r>rDr?rGrz_IniTypeTag | _IniLiteral)rPr,rrD)rXrrrD)rPrrrr&)> __future__rrcollections.abcrr dataclassesrr"rr.rtypingrrrr r r r r r exceptionsr _pytest._ior _pytest.compatr_pytest.deprecatedrtyping_extensionsrrqrr(rrrrDr dataclassr!rNr,r-r@rHrLr[r]rrrmArgumentParserrj HelpFormatterrcompiler1rActionr>)r7s0r+rVs"$$   "!-* !M Y)8D3J4F4L+MMPTTKTd#$!;.{[7PRU7U1VVV>Fk=R(Scc(S& 7 W) %  )- ~,~,~,B""6"6JE.E.P48224@4 8#9#94 p "**. /$N">">g)Ts/ F