U -e'@sdZddlZddlZddlZzddlZddlmZWn,ek r`GdddZeZe ZYnXej ej e dddZ ddd d ZGd d d eZdS) z;Helper utilities for integrating argcomplete with traitletsN)CompletionFinderc@seZdZeejdddZdS) StubModule)attrreturncCs&|dstdtd|ddS)N__zNo module named 'argcomplete'z*argcomplete stub module has no attribute '') startswithModuleNotFoundErrorAttributeError)selfrr d/var/www/html/Darija-Ai-Train/env/lib/python3.8/site-packages/traitlets/config/argcomplete_config.py __getattr__s zStubModule.__getattr__N)__name__ __module__ __qualname__strtAnyrr r r r rsr)rcCszdtjkrdStjd}ttjd}zt||\}}}}}Wntk rVYdSXttjdd}||d}|S)aGet current words prior to completion point This is normally done in the `argcomplete.CompletionFinder` constructor, but is exposed here to allow `traitlets` to follow dynamic code-paths such as determining whether to evaluate a subcommand. _ARGCOMPLETENZ COMP_LINEZ COMP_POINT)osenvironint argcompleteZ split_liner )Z comp_lineZ comp_pointZcword_prequote cword_prefixZ cword_suffix comp_wordsZlast_wordbreak_posstartr r r get_argcomplete_cwordss*    rc Csjz tttjddtjd<WnDtk rdztdtjdWnttfk r^YnXYnXdS)aUAssumes ``$_ARGCOMPLETE`` is set and `argcomplete` is importable Increment the index pointed to by ``$_ARGCOMPLETE``, which is used to determine which word `argcomplete` should start evaluating the command-line. This may be useful to "inform" `argcomplete` that we have already evaluated the first word as a subcommand. rrz!Unable to increment $_ARGCOMPLETEN) rrrr ExceptionrdebugKeyErrorr r r r r increment_argcomplete_indexBs r"cseZdZUdZejed<gZej ej ed<gZ ej e ed<e ej ej ej e fdddZej dd d d Zej e e ej ej e d fd d Zeje ej e dfdd ZZS)ExtendedCompletionFinderaAn extension of CompletionFinder which dynamically completes class-trait based options This finder adds a few functionalities: 1. When completing options, it will add ``--Class.`` to the list of completions, for each class in `Application.classes` that could complete the current option. 2. If it detects that we are currently trying to complete an option related to ``--Class.``, it will add the corresponding config traits of Class to the `ArgumentParser` instance, so that the traits' completers can be used. 3. If there are any subcommands, they are added as completions for the first word Note that we are avoiding adding all config traits of all classes to the `ArgumentParser`, which would be easier but would add more runtime overhead and would also make completions appear more spammy. These changes do require using the internals of `argcomplete.CompletionFinder`. _parserconfig_classes subcommands)rrcshdd|jD}|}dkrFdddfdd|D}ntdkrdfdd|D}|S) zMatch the word to be completed against our Configurable classes Check if cword_prefix could potentially match against --{class}. for any class in Application.classes. cSsg|]}|d|jdfqS)--.)r).0clsr r r rszDExtendedCompletionFinder.match_class_completions..r(Nrcs g|]\}}|kr||fqSr r r)r*crr r r+vsrcs"g|]\}}|r||fqSr rr,r.r r r+xs )r%indexlen)r rZclass_completionsmatched_completionsr r.r match_class_completionsls  z0ExtendedCompletionFinder.match_class_completionsN)r*rcCszh|jddD]R\}}|jdp0t|dd}|jd}||jjd|jd|t|j |d_ qWnt k r|YnXdS) a-Add dummy arguments to our ArgumentParser for the traits of this class The argparse-based loader currently does not actually add any class traits to the constructed ArgumentParser, only the flags & aliaes. In order to work nicely with argcomplete's completers functionality, this method adds dummy arguments of the form --Class.trait to the ArgumentParser instance. This method should be called selectively to reduce runtime overhead and to avoid spamming options across all of Application.classes. T)configZ argcompleterN multiplicityr'r()typehelpnargs) Z class_traitsitemsmetadatagetgetattrr$ add_argumentrrr7 completerr )r r*Z traitnameZtraitr>r5r r r inject_class_to_parser}s"   z/ExtendedCompletionFinder.inject_class_to_parser)rrargsrc s|jj}tdkod|k}|rR|}t|dkr|dd}||npt|dkrd|dkr|s|dddD]D}t|dkr||d|kr|||}|r|dd}||qq|tj|f|} |jrt|dkrt d| fdd|jD| S) a Overriden to dynamically append --Class.trait arguments if appropriate Warning: This does not (currently) support completions of the form --Class1.Class2.<...>.trait, although this is valid for traitlets. Part of the reason is that we don't currently have a way to identify which classes may be used with Class1 as a parent. Warning: This is an internal method in CompletionFinder and so the API might be subject to drift. rrr(NzAdding subcommands forc3s|]}|r|VqdSNr/)r)Zsubcr.r r s z.) r$ prefix_charsr1r3r?super_get_completionsr&rr extend) r rrr@rDZ is_optionr2Z matched_clsZ prev_word completions __class__r.r rFs(        z)ExtendedCompletionFinder._get_completions)parserrrcsJt||}|dr|S||}t|dkrF|dd|D|S)z6Overriden to add --Class. completions when appropriater(rcss|]\}}|VqdSrBr )r)r*optr r r rCszCExtendedCompletionFinder._get_option_completions..)rE_get_option_completionsendswithr3r1rG)r rKrrHr2rIr r rMs   z0ExtendedCompletionFinder._get_option_completions)rrr__doc__argparseArgumentParser__annotations__r%rListrr&rTupler3r?rFrM __classcell__r r rIr r#Us  "8r#)rOrPrtypingrrr ImportErrorrobjectOptionalrSrrr"r#r r r r s '