B ܇\=@sddlZddlZddlZddlZddlmZdZGdddeZGddde Z Gdd d eZ Gd d d eZ Gd d d ej ZGdddejZdS)N) UsageErrorZ file_or_dirc@sreZdZdZdZdddZddZddd Zd d Zdd d Z ddZ dddZ dddZ dddZ dddZdS)Parserz Parser 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. NcCs6td|d|_g|_||_||_i|_g|_i|_dS)Nzcustom options)parser) OptionGroup _anonymous_groups _processopt_usage_inidict _ininames extra_info)selfusageZ processoptr8lib/python3.7/site-packages/_pytest/config/argparsing.py__init__szParser.__init__cCs|jr|jr||dS)N)rdest)r optionrrr processoptionszParser.processoptioncCsjx|jD]}|j|kr|SqWt|||d}d}x"t|jD]\}}|j|kr` but will be shown in the respective group in the output of ``pytest. --help``. )rr)rnamer enumerateinsert)r r descriptionZaftergroupiZgrprrrgetgroup#s   zParser.getgroupcOs|jj||dS)a register a command line option. :opts: option names, can be short or long options. :attrs: same attributes which the ``add_option()`` function of the `argparse library `_ 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 optsattrsrrrr:szParser.addoptioncCs>ddlm}||_||jdd|D}|jj||dS)Nr)try_argcompletecSs&g|]}t|tjjrt|n|qSr) isinstancepypathlocalstr).0xrrr Osz Parser.parse..) namespace)_pytest._argcompleter! _getparser optparser parse_args)r argsr*r!rrrparseJs    z Parser.parsec Csddlm}t||j|jd}|j|jg}xT|D]L}|jr2|jpF|j }| |}x*|jD] }| }| } |j || qZWq2W||j tdd_|S)Nr)filescompleter)prog*)nargs)r+r1MyOptionParserr r2rroptionsrrZadd_argument_groupnamesr add_argument FILE_OR_DIRZ completer) r r1r-groupsrZdescZarggrouprnarrrr,Rs     zParser._getparsercCs<|j||d}x"|jD]\}}t|||qWt|tS)N)r*)r0__dict__itemssetattrgetattrr9)r r/rr*Z parsedoptionrvaluerrrparse_setoptioncszParser.parse_setoptioncCs|j||ddS)zZparses and returns a namespace object with known arguments at this point. )r*r)parse_known_and_unknown_args)r r/r*rrrparse_known_argsiszParser.parse_known_argscCs$|}dd|D}|j||dS)zparses and returns a namespace object with known arguments, and the remaining arguments unknown at this point. cSs&g|]}t|tjjrt|n|qSr)r"r#r$r%r&)r'r(rrrr)tsz7Parser.parse_known_and_unknown_args..)r*)r,rD)r r/r*r-rrrrCosz#Parser.parse_known_and_unknown_argscCs,|dks t|||f|j|<|j|dS)a register an ini-file option. :name: name of the ini-variable :type: type of the variable, can be ``pathlist``, ``args``, ``linelist`` or ``bool``. :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) <_pytest.config.Config.getini>`. )NZpathlistr/ZlinelistboolN)AssertionErrorr r append)r rhelptypedefaultrrraddiniws z Parser.addini)NN)rN)N)N)N)N)NN)__name__ __module__ __qualname____doc__r2rrrrr0r,rBrDrCrKrrrrr s      rc@s eZdZdZddZddZdS) ArgumentErrorz_ Raised if an Argument instance is created with invalid or inconsistent arguments. cCs||_t||_dS)N)msgr& option_id)r rQrrrrrszArgumentError.__init__cCs |jrd|j|jfS|jSdS)Nz option %s: %s)rRrQ)r rrr__str__szArgumentError.__str__N)rLrMrNrOrrSrrrrrPsrPc@sFeZdZdZeeeedZddZ ddZ ddZ d d Z d d Z d S)Argumentzclass 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 )intstringfloatcomplexcOsv||_g|_g|_|d|_d|dp,dkr@tjdtddy |d}Wntk r`Yn|Xt |t j r|d krtjd ||ftd dt |d d |d<n&tjd||ftd dt j||d<|d|_ n||_ y|d|_Wntk rYnX|||jsr|jr8|jd dddd|_n:y|jd dd|_Wn tk rptd|YnXdS)z3store parms in private vars for use in add_argumentrz%defaultrHrzHpytest now uses argparse. "%default" should be changed to "%(default)s" ) stacklevelrIZchoicez`type` argument to addoption() is the string %r. For choices this is optional and can be omitted, but when supplied should be a type (for example `str` or `int`). (options: %s)choicesrz`type` argument to addoption() is the string %r, but when supplied should be a type (for example `str` or `int`). (options: %s)rJN-_rzneed a long or short option)_attrs _short_opts _long_optsgetrwarningswarnDeprecationWarningKeyErrorr"sixZ string_typesrIrT_typ_maprJ_set_opt_stringsreplace IndexErrorrP)r r7r typrrrrsR     zArgument.__init__cCs |j|jS)N)rarb)r rrrr7szArgument.namesc Csd}|jr||jx6|D].}yt|||j|<Wq tk rLYq Xq W|jdr~|jd}|dd}||jd<|jS)Nzdefault dest helprHz%defaultz %(default)s)splitrrGr@r`AttributeErrorrcrk)r r attrr<rrrr s       zArgument.attrscCsx|D]}t|dkr&td||qt|dkrf|ddkrJ|ddksXtd|||j|q|dddkr|ddkstd|||j|qWd S) z]directly from optparse might not be necessary as this is passed to argparse later onr]z>invalid option string %r: must be at least two characters longrr^rzMinvalid short option string %r: must be of the form -x, (x any non-dash char)z--zGinvalid long option string %r: must start with --, followed by non-dashN)lenrPrarGrb)r roptrrrrjs&   zArgument._set_opt_stringscCsg}|jr|dt|jg7}|jr8|dt|jg7}|dt|jg7}t|drj|dt|jg7}t|dr|dt|jg7}dd |S) Nz _short_opts: z _long_opts: zdest: rIztype: rJz default: z Argument({})z, ) rareprrbrhasattrrIrJformatjoin)r r/rrr__repr__s  zArgument.__repr__N)rLrMrNrOrUr&rWrXrirr7r rjrwrrrrrTs<rTc@s0eZdZd ddZddZddZd d d ZdS)rrNcCs||_||_g|_||_dS)N)rrr6r)r rrrrrrrszOptionGroup.__init__cOsFt|dd|jD}|r*td|t||}|j|dddS)aE add 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 css |]}|D] }|VqqdS)N)r7)r'rrrrrr ,sz(OptionGroup.addoption..zoption names %s already addedF) shortupperN)set intersectionr6 ValueErrorrT_addoption_instance)r optnamesr Zconflictrrrrr#s   zOptionGroup.addoptioncOst||}|j|dddS)NT)ry)rTr})r r~r rrrr _addoption3s zOptionGroup._addoptionFcCsV|s4x.|jD]$}|ddkr |dr tdq W|jrF|j||j|dS)Nrr^rzlowercase shortoptions reserved)raislowerr|rrr6rG)r rryrrrrrr}7s   zOptionGroup._addoption_instance)rN)F)rLrMrNrrrr}rrrrrs rc@s(eZdZdddZddZd ddZdS) r5NcCs0|si}||_tjj|||jdtd||_dS)NF)r2rZadd_helpZformatter_class)_parserargparseArgumentParserrr DropShorterLongHelpFormatterr )r rr r2rrrrBszMyOptionParser.__init__cCs>d|j|f}t|jdr*d||jjf}t||dS)z1Transform argparse error message into UsageError.z %s: error: %s_config_source_hintz%s (%s)N)r2rtrrrZ format_usage)r messagerQrrrerrorQs zMyOptionParser.errorcCs|||\}}|rxj|D]b}|r|ddkrdd|g}x,t|jD]\}}|d||fqNW|d|qWt|t ||S)z'allow splitting of positional argumentsrr^zunrecognized arguments: %s z %s: %s ) rDrvsortedr r>rGrr@r9extend)r r/r*argvarglineskvrrrr.Zs zMyOptionParser.parse_args)NN)NN)rLrMrNrrr.rrrrr5As  r5c@seZdZdZddZdS)rashorten help for long options that differ only in extra hyphens - collapse **long** options that are the same except for extra hyphens - special action attribute map_long_option allows surpressing additional long options - shortcut if there are only two options and one of them is a short one - cache result on action object as this is called at least 2 times c Cstj||}|r"|ddkr"|St|dd}|r6|S|d}t|dkrvt|ddkslt|ddkrv||_|Sg}t|di}|dkri}i}x|D]}t|dks|ddkrq|d std |||dd} | d|kr| dd } | |kst|| t| kr| || <qWxh|D]`}t|dksJ|ddkrT| ||dd| | dd kr*| | dd dq*Wd ||_|jS) Nrr^_formatted_action_invocationz, r]rZmap_long_optionrz--z)long optional argument without "--": [%s]r=) r HelpFormatter_format_action_invocationr@rnrqr startswithrPrkrGrcrv) r actionZorgstrZresr6Z return_listZ option_mapZ short_longrZxxoptionZ shortenedrrrrrsD  ,           z6DropShorterLongHelpFormatter._format_action_invocationN)rLrMrNrOrrrrrrhsr)rrdr#rhZ_pytest.config.exceptionsrr9objectr ExceptionrPrTrrr5rrrrrrs {%'