W< @ s d Z d d l Z d d l m Z d d l m Z m Z m Z m Z d d Z d d Z
d
d Z d d
Z d d Z
Gd d d e Z Gd d d e Z Gd d d e Z Gd d d e Z d S)aq
click.parser
~~~~~~~~~~~~
This module started out as largely a copy paste from the stdlib's
optparse module with the features removed that we do not need from
optparse because we implement them in Click on a higher level (for
instance type handling, help formatting and a lot more).
The plan is to remove more and more from here over time.
The reason this is a different module and not optparse from the stdlib
is that there are differences in 2.x and 3.x about the error messages
generated and optparse in the stdlib uses gettext for no good reason
and might cause us issues.
N)deque )
UsageError NoSuchOptionBadOptionUsageBadArgumentUsagec sh t t | } g } d f d d x | r | } | d k rm | j q9 | d k r f d d t | D } d k r | j | j t | q9 | d k r9 d k r t d t | | j d q9 W d k rRt | <g t | d d | d d <t | t f S) a Given an iterable of arguments and an iterable of nargs specifications,
it returns a tuple with all the unpacked arguments at the first index
and all remaining arguments as the second.
The nargs specification is the number of arguments that should be consumed
or `-1` to indicate that this position should eat up all the remainders.
Missing items are filled with `None`.
Nc sA y$ d k r | j S| j SWn t k
r< d SYn Xd S)N)popleftpop
IndexError)c)spos
/parser.py_fetch' s
z _unpack_args.<locals>._fetchr c s g | ] } q Sr
r
).0_)r argsr
r
<listcomp>5 s z _unpack_args.<locals>.<listcomp>r zCannot have two nargs < 0) r appendrangereversetuple TypeErrorlenreversedlist)r Z
nargs_specrvnargsxr
)r r r r _unpack_args s0
"
(r c C s6 | d k r t d | t d | | f d S)Nr z %s option requires an argumentz %s option requires %d arguments)r )r optr
r
r _error_opt_argsK s r! c C sn | d d } | j r& d | f S| d d | k rZ | d d | d d f S| | d d f S)Nr )isalnum)r firstr
r
r split_optQ s
r& c C sB | d k s | j d k r | St | \ } } | | j | S)N)Ztoken_normalize_funcr& )r ctxprefixr
r
r
normalize_optZ s r) c C s g } x t j d | t j D] } | j j } | d d | d d k r | d d d k r | d d j d d j d } y t | | } Wn t k
r Yn X| j | q W| S)
zDGiven an argument string this attempts to split it into small parts.z>('([^'\\]*(?:\\.[^'\\]*)*)'|"([^"\\]*(?:\\.[^"\\]*)*)"|\S+)\s*Nr z"'asciibackslashreplacezunicode-escaper, )
refinditerSgroupstripencodedecodetype UnicodeErrorr )stringr matchargr
r
r split_arg_stringa s 6
r9 c @ sF e Z d Z d d d d d d Z e d d Z d d Z d S) OptionNr c
C s g | _ g | _ t | _ x | D] } t | \ } } | sS t d | | j j | d t | d k r t | d k r | j j | q% | j j | | j j | q% W| d k r d } | | _ | | _
| | _ | | _ | | _
d S)Nz'Invalid start character for option (%s)r r store) _short_opts
_long_optssetprefixesr&
ValueErroraddr r destactionr constobj)
selfoptsrB rC r rD rE r r( valuer
r
r __init__u s(
$ zOption.__init__c C s
| j d k S)Nr; r )zstorezappend)rC )rF r
r
r takes_value s zOption.takes_valuec C s | j d k r" | | j | j <n | j d k rG | j | j | j <n | j d k rx | j j | j g j | n | j d k r | j j | j g j | j nK | j d k r | j j | j d d | j | j <n t d | j | j j | j d S) Nr; store_constr Z append_constcountr r zunknown action %r)
rC rG rB rD
setdefaultr getr@ orderrE )rF rH stater
r
r process s "%)zOption.process)__name__
__module__ __qualname__rI propertyrJ rQ r
r
r
r r: s s r: c @ s. e Z d Z d d d d Z d d Z d S)Argumentr Nc C s | | _ | | _ | | _ d S)N)rB r rE )rF rB r rE r
r
r rI s zArgument.__init__c C s | j d k rk t d d | D } | t | k rC d } n( | d k rk t d | j | j f | | j | j <| j j | j d S)Nr c s s! | ] } | d k r d Vq d S)Nr r
)r r r
r
r <genexpr> s z#Argument.process.<locals>.<genexpr>r zargument %s takes %d values) r sumr r rB rG rO r rE )rF rH rP Zholesr
r
r rQ s zArgument.process)rR rS rT rI rQ r
r
r
r rV s rV c @ s e Z d Z d d Z d S) ParsingStatec C s( i | _ g | _ | | _ g | _ d S)N)rG largsrargsrO )rF r[ r
r
r rI s zParsingState.__init__N)rR rS rT rI r
r
r
r rY s rY c @ s e Z d Z d Z d d d Z d d d d d d Z d d d d Z d
d Z d d
Z d d Z d d Z
d d Z d d Z d S) OptionParsera+ The option parser is an internal class that is ultimately used to
parse options and arguments. It's modelled after optparse and brings
a similar but vastly simplified API. It should generally not be used
directly as the high level Click classes wrap it for you.
It's not nearly as extensible as optparse or argparse as it does not
implement features that are implemented on a higher level (such as
types or defaults).
:param ctx: optionally the :class:`~click.Context` where this parser
should go with.
Nc C ss | | _ d | _ d | _ | d k r? | j | _ | j | _ i | _ i | _ t d d g | _ g | _ d S)NTF-z--)r' allow_interspersed_argsignore_unknown_options
_short_opt _long_optr>
_opt_prefixes_args)rF r' r
r
r rI s zOptionParser.__init__r c s | d k r | } f d d | D } t | | d | d | d | d | } j j | j x | j D] } | j | <qo Wx | j D] } | j | <q Wd S)ay Adds a new option named `dest` to the parser. The destination
is not inferred (unlike with optparse) and needs to be explicitly
provided. Action can be any of ``store``, ``store_const``,
``append``, ``appnd_const`` or ``count``.
The `obj` can be used to identify the option in the order list
that is returned from the parser.
Nc s" g | ] } t | j q Sr
)r) r' )r r )rF r
r r s z+OptionParser.add_option.<locals>.<listcomp>rC r rD rE )r: rb updater? r< r` r= ra ) rF rG rB rC r rD rE optionr r
)rF r
add_option s
zOptionParser.add_optionc C s; | d k r | } | j j t d | d | d | d S)zAdds a positional argument named `dest` to the parser.
The `obj` can be used to identify the option in the order list
that is returned from the parser.
NrB r rE )rc r rV )rF rB r rE r
r
r add_argument s zOptionParser.add_argumentc C st t | } y | j | | j | Wn1 t k
r] | j d k sV | j j rY Yn X| j | j | j f S)aZ Parses positional arguments and returns ``(values, args, order)``
for the parsed options and arguments as well as the leftover
arguments if there are any. The order is a list of objects as they
appear on the command line. If arguments appear multiple times they
will be memorized multiple times as well.
N) rY _process_args_for_options_process_args_for_argsr r' Zresilient_parsingrG rZ rO )rF r rP r
r
r
parse_args s
zOptionParser.parse_argsc C sy t | j | j d d | j D \ } } x1 t | j D] \ } } | j | | | q? W| | _ g | _ d S)Nc S s g | ] } | j q Sr
)r )r r r
r
r r s z7OptionParser._process_args_for_args.<locals>.<listcomp>)r rZ r[ rc enumeraterQ )rF rP Zpargsr idxr8 r
r
r ri s z#OptionParser._process_args_for_argsc C s x | j r | j j d } t | } | d k r: d S| d d | j k rr | d k rr | j | | q | j r | j j | q | j j d | d Sq Wd S)Nr z--r ) r[ r r rb
_process_optsr^ rZ r insert)rF rP r8 Zarglenr
r
r rh s % z&OptionParser._process_args_for_optionsc s& | j k r= f d d | j D } t d | | j } | j r | d k rr | j j d | | j } t | j | k r t | q| d k r | j j d } qt | j d | } | j d | =n% | d k r t
d n d } | j | | d S)Nc s% g | ] } | j r | q Sr
)
startswith)r Zword)r r
r r B s z0OptionParser._match_long_opt.<locals>.<listcomp>
possibilitiesr r z %s option does not take a value) ra r rJ r[ rn r r r! r r r rQ )rF r explicit_valuerP rp re r rH r
)r r _match_long_opt@ s$
z OptionParser._match_long_optc C s d } d } | d } g } xK| d d D]9} t | | | j } | j j | } | d 7} | s | j r | j | q- t | | j rI| t | k r | j j
d | | d d } | j }
t | j |
k r t |
| qO|
d k r | j j
d } qOt | j d |
} | j d |
=n d } | j | | | r- Pq- W| j r| r| j j | d j | d S)NFr r Tr" )r) r' r` rN r_ r r rJ r r[ rn r r! r r rQ rZ join) rF r8 rP stopir( Zunknown_optionsZchr re r rH r
r
r _match_short_opt` s<
z OptionParser._match_short_optc C s d } d | k r- | j d d \ } } n | } t | | j } y | j | | | WnW t k
r | d d | j k r | j | | S| j s | j j | Yn Xd S)N=r r# )
splitr) r' rr r rb rv r_ rZ r )rF r8 rP rq Zlong_optZ
norm_long_optr
r
r rm s
zOptionParser._process_opts)
rR rS rT __doc__rI rf rg rj ri rh rr rv rm r
r
r
r r\ s
$ /r\ )ry r- collectionsr
exceptionsr r r r r r! r& r) r9 objectr: rV rY r\ r
r
r
r <module> s " 3 0