o wi=@sRddlZddlZddlZddlZddlZddlZddlZddlmZmZddl m Z ej ddZ ej dkr=ddlm Z nGdddeZ d d Zd d Zd dZddZddZd-ddZd.ddZd/ddZddZGdddeZeZe ejjjddZddZiZd d!Z d"d#Z!d0d%d&Z"Gd'd(d(ejZd)d*Z#d1d+d,Z$dS)2N)datetimedate)gettext3)getfullargspecc@seZdZdZddZdS)rz?A quick and dirty replacement for getfullargspec for Python 2.XcCs,t|\|_|_|_|_t|di|_dS)N__annotations__)inspect getargspecargsvarargsvarkwdefaultsgetattr annotations)selffrF/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/plac_core.py__init__szgetfullargspec.__init__N)__name__ __module__ __qualname____doc__rrrrrrs rcCtt|dddS)zReturns year-month-dayz%Y-%m-%dr)rtimestrptimesrrrto_dater cCr)z)Returns year-month-day hour-minute-secondz%Y-%m-%d %H-%M-%Sr)rrrrrrr to_datetimer!r#cCst|r t|}|St|rt|}|jd=|St|r7|jtjur,tdd}nt|j}|jd=|St|drGt|j }|jd=|St t dt |)zGiven a callable return an object with attributes .args, .varargs, .varkw, .defaults. It tries to do the "right thing" with functions, methods, classes and generic callables.rcSsdSNrrrrr.szgetargspec..__call__z%Could not determine the signature of ) r isfunctionrismethodr isclassrobjecthasattrr' TypeError_str) callableobjargspecrrrr #s(     r c sfdd}|S)z Returns a decorator annotating a function with the given annotations. This is a trick to support function annotations in Python 2.X. cs^t|}|j}|jr||j|jr||jD]}||vr)ttd|q|_|S)Nz$Annotating non-existing argument: %s)rr r appendr NameErrorr.r)rfasr argnameannrrannotate@s   zannotations..annotater)r7r8rr6rr;s  rcCs0z||j|<W|Sty||i|_Y|Swr$)rAttributeError)argr7rrrr _annotatePs  r;c Cstt||dd|||fS)z7 Decorator for annotating positional arguments positionalN functoolspartialr;)r:helptypechoicesmetavarrrrposXsrDc Cs&|p|d}tt||d||||fS)z5 Decorator for annotating optional arguments roptionr=)r:r@rAabbrevrBrCrrropt`s rGc Cs"tt||d|p |ddddfS)z( Decorator for annotating flags flagrNr=)r:r@rFrrrflgisrIcCs<t|dot|dot|dot|dot|dot|dS)zt An object is an annotation object if it has the attributes help, kind, abbrev, type, choices, metavar. r@kindrFrArBrC)r,objrrr is_annotationqsrMc@s*eZdZ  dddZddZeeZdS) AnnotationNr<cCsP|dvsJ||dkr|dusJ|||_||_||_||_||_||_dS)N)r<rErHr<)r@rJrFrArBrC)rr@rJrFrArBrCrrrr|s zAnnotation.__init__cCs>t|r|St|rt|}||St|r||S||S)z9Helper to convert an object into an annotation, if needed)rMr r*r/iterable)clsrLrrrfrom_s zAnnotation.from_)Nr<NNNN)rrrrrQ classmethodrrrrrN{s   rNcCsLt|jr t|jndtjd}t|D] }|tvr#t ||||<q|S)zN Extracts the configuration of the underlying ArgumentParser from obj N) descriptionformatter_class) dictrtextwrapdedentrstripargparseRawDescriptionHelpFormatterdir PARSER_CFGr)rLcfgnamerrrpconfs r`cKszt|WStyYnwt|}||tdi|t|<}||_|dt|dd|_ t |drGt |sG| |j|d|S|||S)zg obj can be a callable or an object with a .commands attribute. Returns an ArgumentParser. case_sensitiveTcommands subcommandsNr)_parser_registryKeyErrorr`copyupdateArgumentParserrLgetrrar,r r*addsubcommandsrb populate_from)rL confparamsconfparserrrr parser_froms"      rocCsXg}i}|D]!}td|}|r"|d}|t|dd||<q||q||fS)z= Returns two lists: regular args and name=value args z([a-zA-Z_]\w*)=rSN)rematchgrouplenr2)r arglistkwargsr:rqr_rrr_extract_kwargss   rvTcs|s dd|D}fdd|D}t|dkr |dSfdd|D}t|}|dkr5|dS|dkrCttd|fdS) zL Extract the command name from an abbreviation or raise a NameError cSsg|]}|qSr)upper).0crrr sz_match_cmd..csg|]}|kr|qSrrrxr_rFrrrzsrSrcsg|] }|r|qSr) startswithr{r|rrrzsz!Ambiguous command %r: matching %sN)rwrsr3r.)rFrbraperfect_matchesmatchesnrr|r _match_cmds rcsneZdZdZdZedkrfddZddZdd Zd d Z dddZ ddZ ddZ ddZ ddZZS)rhzk An ArgumentParser with .func and .argspec attributes, and possibly .commands and .subparsers. T)r csBtt|j|i||jdjtdkrtd|jd_dSdS)NrSzoptional argumentsoptions)superrhr_action_groupstitler.)rr ru __class__rrrszArgumentParser.__init__cCs|S)z6Can be overridden to preprocess command-line argumentsr)rr:rrraliasszArgumentParser.aliasc sfdd|D}d}tdr,|\}}|dur&|dur&||fS|dur,|tdr=jjr=|\}n|g}tdsLttdrjjrjj}jj}|j vrtj |}t |\}} |j |<n|j vrj |}t |\}} |j |<|r|s t d|ni} tjjt| @} | r t dd| fd djjD}tjjpd g} |j|| |i| fS) z Call the underlying function with the args. Works also for command containers, by dispatching to the right subparser. csg|]}|qSr)rrxar%rrrzsz*ArgumentParser.consume..N subparsersr1zUnrecognized arguments: %szcolliding keyword arguments: %s csg|] }t|dqS)r.)rrYr)nsrrrzs)r,_extract_subparser_cmdmissingr1r parse_known_args parse_args SystemExitr __dict__poprverrorr.setr joinrfunc) rr rtcmdsubp extraoptsvr lstru collisionr r)rrrconsumesH          zArgumentParser.consumecCs\|jd}|jj}t|D]\}}||s+t|||j}||=|||p'|fSq dS)zP Extract the right subparser from the first recognized argument rNN) prefix_charsr_name_parser_map enumerater}rrari)rrt optprefixname_parser_mapir:rrrrr!s  z%ArgumentParser._extract_subparser_cmdNrc Cst||r|j|jvrttd|t|ds |j|d|_n|r(|j|dtt |dd}t |dd}|D]*}t |||d}|j rNt |j nd} |jj|f|| d t||q8dS) zS Extract a list of subcommands from obj and add them to the parser zThe prefix %r is already taken!r)r cmdprefixradd_helpTN)rr@)r,rr ValueErrorr.add_subparsersradd_argument_grouprsrrrWrXrY add_parserr`rk) rrbrLrr prefixlenrrrdocrrrrj.s.    zArgumentParser.addsubcommandscCs||_t||_|t|<dS)z Extracts the signature from a callable object and adds an .argspec attribute to the parser. Also adds a .func reference to the object. N)rr r1rd)rrLrrr_set_func_argspecBs  z ArgumentParser._set_func_argspecc Cs|||j}|jp d}t|j}t|}tf|||}t|ddd}|_t|j|D]\}} |j |d} t | } | j } | turJd} n6| } | jdurXdt| | _| jdurxt| trft| _nt| trot| _n | durxt| | _| s| dkrd} | jd vr|d r|d }n|d d}| jr|| j|d |f}n'||f}n!| tur|j|| j| j| j| d n|j|d | j| | j| j| d| jdkr| tur| pt| } |j|| j| | j| j| dq0| jdkr| tur| durttd|| f|j|d| jdq0|jr.t |j |jd} |j|jd| jg| j| j d|j rNt |j |j d} |j|j d| ji| j| j ddSdS)z Extract the arguments from the attributes of the passed function and return a populated ArgumentParser instance. rr-rNz[%s]rz'')rErHr.r)r@rArBrC?)nargsr@defaultrArBrCrE)r@rrArBrCrHFz#Flag %r wants default False, got %r store_true)actionr@*)rr@rrArC)!rr1rrsr NONErprefixziprrirNrQrCr@r/rA isinstancerr#rr rJendswithrYreplacerF add_argumentrBr-r.r r )rrrrn_args n_defaults alldefaultsrr_rr7rrCdfltsuffix shortlongrrrrkKs                       zArgumentParser.populate_fromcstjdfdd}||S)zMay raise a SystemExit __missing__csd|S)Nz No command %r)r)r_r%rrr&sz(ArgumentParser.missing..)rrL)rr_missrr%rrszArgumentParser.missingcCs t||jD]}t|qdS)zUseful for debuggingN)print_actions)rrrrr print_actionss  zArgumentParser.print_actions)Nr)rrrrraversionrrrrrjrrkrr __classcell__rrrrrhs ,  IrhcCs&t|dot| ot|ttf S)zAAny object with an __iter__ method which is not a string or class__iter__)r,r r*rr/bytesrKrrrrOs&rOcCsZ|dur tjdd}t|}|r|jddd|d||\}}t|r+|r+t|S|S)a If obj is a function or a bound method, parse the given arglist by using the parser inferred from the annotations of obj and call obj with the parsed arguments. If obj is an object with attribute .commands, dispatch to the associated subparser. NrSz --versionz-vr)rr)sysargvrorrrOlist)rLrteagerrrnrresultrrrcalls r)NNNN)NNNNNr)T)NTN)%rprrr rWr>rZrrrr. version_inforrr+r r#r rr;rDrGrIrMrNrrhrr r]r`rdrorvrrOrrrrrsD        >