o HTh @s|dZddlmZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl m Z mZddlmZddlmZddlmZddlmZdd lmZmZdd lmZmZmZmZmZm Z m!Z!dd l"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,dd l-m.Z.dd l/m0Z0ddl1m2Z2m3Z3ddl4m5Z5ddl6m7Z7d8Z9d8Z:e;e dsdge _?ddZ@e@AdvrdZBne@AdvrdZBneCde@e jDoe jDEdZFe jGde jHde jIfd ZJe jKejLd!fZMe j$eNe jIfZOe jPe j'eNZQe j'e jReZSd8d$d%ZTGd&d'd'eUZVGd(d)d)ejWZXGd*d+d+eZYeYjZZ[eYj\Z]d9d:d2d3Z^d;d5d6Z_e`d7kr, use: --help-all z The command-line option below sets the respective configurable class-parameter: --Class.parameter=value This line is evaluated in Python, so simple expressions are allowed. For instance, to set `C.a=[0,1,2]`, you may type this: --C.a='range(3)' argvzp Subcommands are launched as `{app} cmd [args]`. For information on using subcommand 'cmd', do: `{app} cmd -h`. -TRAITLETS_APPLICATION_RAISE_CONFIG_FILE_ERROR>1trueT>r%0falseFzUnsupported value for environment variable: 'TRAITLETS_APPLICATION_RAISE_CONFIG_FILE_ERROR' is set to '%s' which is none of {'0', '1', 'false', 'true', ''}.z pythonw.exeT.)boundzlogging.LoggerAdapter[t.Any]methodreturncs$td fdd }tt|S) aMethod decorator for catching invalid config (Trait/ArgumentErrors) during init. On a TraitError (generally caused by bad config), this will print the trait's message, and exit the app. For use on init methods, to prevent invoking excepthook on invalid input. app Applicationargst.Anykwargsr.c slz |g|Ri|WSttfy5}z|jd||jd|j|dWYd}~dSd}~ww)Nz0Bad config encountered during initialization: %szConfig at the time: %s)rr logfataldebugconfigexit)r/r1r3er-_/home/air/segue/gemini/backup/venv/lib/python3.10/site-packages/traitlets/config/application.pyinnerssz!catch_config_error..innerN)r/r0r1r2r3r2r.r2) functoolswrapstcastr+)r-r>r<r;r=catch_config_errorjs  rCc@s eZdZdS)ApplicationErrorN)__name__ __module__ __qualname__r<r<r<r=rDsrDcs,eZdZdZejZdZd fdd ZZ S) LevelFormattera Formatter with additional `highlevel` record This field is empty if log level is less than highlevel_limit, otherwise it is formatted with self.highlevel_format. Useful for adding 'WARNING' to warning messages, without adding 'INFO' to info, etc. z %(levelname)s |recordlogging.LogRecordr.strcs.|j|jkr|j|j|_nd|_t|S)Nr%)levelnohighlevel_limithighlevel_format__dict__ highlevelsuperformat)selfrI __class__r<r=rRs  zLevelFormatter.format)rIrJr.rK) rErFrG__doc__loggingWARNrMrNrR __classcell__r<r<rTr=rHs  rHcseZdZUdZedZded<edZded<eeZded<ee Z ded<ee Z ded <e Z e ZeZded <gZd ed < ddddZedZded<eZded<eeZedejddjddZeZedddjddZedd djddZ dd"d#Z!e"d$d%d&d'dd+d,Z#e"d-d.d/dd0d1Z$dd2d3Z%e&d-dd5d6Z'e(d7djddZ)d8d9iZ*d!ed:<dd&ej+iid;fdddiid?fd@Z,d!edA<e(Z-dBedC<e.dDddEZ/eeZ0e.e1dFidGdZ2eZ3dHedI<edJdjddZ4edKdjddZ5e"d>ddLdMZ6e"d<ddNdOZ7ddRdSZ8e"dTe9dfdUdV Z:e;dddXdYZdd^d_Z?ddadbZ@ddcddZAddedfZBddgdhZCddidjZDddkdlZEddmdnZFddpdqZGdddsdtZHdddudvZIddxdyZJddzd{ZKdd|d}ZLdd~dZMdddZNdddZOe;ddddZPdddZQdddZReSddddZTeSdddZUe;ddddZVeS rddddZWeXdddZYe; ddddZZe;dddZ[ ddddZ\ddddZ]dddZ^ddddZ_dddZ`eSddddZaZbS)r0z8A singleton application with full configuration support. applicationzstr | Unicode[str, str | bytes]namezThis is an application. descriptionoption_descriptionkeyvalue_descriptionsubcommand_descriptionexamples ClassesTypeclassesNClassesType | Noner.+t.Generator[type[Configurable], None, None]ccsX|dur|j}t}|D]}t|D]}t|tr(||vr(|||Vqq dS)aIterate through configurable classes, including configurable parents :param classes: The list of classes to iterate; if not set, uses :attr:`classes`. Children should always be after parents, and each class should only be yielded once. N)rbsetreversedmro issubclassr add)rSrbseencparentr<r<r=_classes_inc_parentss  z Application._classes_inc_parentsz0.0versionzlist[str] | List[str]r$) r (2DEBUGINFOrXERRORCRITICALz#Set the log level by value or name.) default_valuehelpTr8z%Y-%m-%d %H:%M:%Sz:The date format used by logging formatters for %(asctime)s)ryz#[%(name)s]%(highlevel)s %(message)szThe Logging format templateStrDictcCslddddt|jddid|jjd|jj|j|jdi|jjddgd id d }t r4|d =|d =|S)aReturn the base logging configuration. The default is to log to stderr using a StreamHandler, if no default handler already exists. The log handler level starts at logging.WARN, but this can be adjusted by setting the ``log_level`` attribute. The ``logging_config`` trait is merged into this allowing for finer control of logging. r4consolezlogging.StreamHandlerzext://sys.stderr)class formatterlevelstream.)r}rRdatefmtrt)rhandlersF)rnr formattersloggersdisable_existing_loggersrr) rW getLevelName log_level_log_formatter_clsrFrE log_format log_datefmtrU IS_PYTHONWrSr8r<r<r=get_default_logging_configs6   z&Application.get_default_logging_configrrrlogging_configchangerNonecCs0|j}t|trtttt||_|dSN) r isinstancerKrArBintgetattrrW_configure_logging)rSrrr<r<r=_observe_logging_changes  z#Application._observe_logging_changer5r)typecC |dSr)rrSrr<r<r=_observe_logging_default z$Application._observe_logging_defaultcCs*|}t||jp it|d|_dS)NT)rrrr_logging_configuredrr<r<r=r!s zApplication._configure_logging AnyLoggercCsHt|jj}d|_|}|dur"|jr|S|js |S|j}|dus|S)z#Start logging for this application.FN)rW getLoggerrUrE propagaterrl)rSr5_logr<r<r= _log_default(szApplication._log_defaulta! Configure additional log handlers. The default stderr logs handler is configured by the log_level, log_datefmt and log_format settings. This configuration can be used to configure additional handlers (e.g. to output the log to a file) or for finer control over the default handlers. If provided this should be a logging configuration dictionary, for more information see: https://docs.python.org/3/library/logging.config.html#logging-config-dictschema This dictionary is merged with the base logging configuration which defines the following: * A logging formatter intended for interactive use called ``console``. * A logging handler that writes to stderr called ``console`` which uses the formatter ``console``. * A logger with the name of this application set to ``DEBUG`` level. This example adds a new handler that writes to a file: .. code-block:: python c.Application.logging_config = { "handlers": { "file": { "class": "logging.FileHandler", "level": "DEBUG", "filename": "", } }, "loggers": { "": { "level": "DEBUG", # NOTE: if you don't list the default "console" # handler here then it will be disabled "handlers": ["console", "file"], }, }, } z log-levelzApplication.log_levelaliasesz5Set log-level to debug, for the most verbose logging. show_configz&z.Application.emit_alias_help..r--zEquivalent to: [--%s]z7Failed collecting help-message for alias %r, due to: %s)rrbrgrArBTyper rEitemsrtuplesplit class_traitsclass_get_trait_help splitlinesrrrreplacer Exceptionr5error) rS classdictrrkaliaslongnamefhelprrtrait fhelp_linesexr<r<r=rs>       zApplication.emit_alias_helpcCr)z Print the flag part of the help.rN)rremit_flag_helprr<r<r=print_flag_help'rzApplication.print_flag_helpc cs|jsdS|jD]X\}\}}z=t|ts|f}t|td}ddd|D}|Vtt| Vddd|D}d|}tt|VWq t yc}z |j d ||d}~wwdS) z.Yield the lines for the flag part of the help.Nrrcsrrrrr<r<r=r5rz-Application.emit_flag_help.. css<|]\}}|D]\}}d|d|d|Vq qdS)rr=N)r)rclname props_dictpropvalr<r<r=r8s zEquivalent to: [%s]z6Failed collecting help-message for flag %r, due to: %s) rrrrrrrrrstriprr5r)rSrcfgrcfg_listcfg_txtrr<r<r=r+s,   zApplication.emit_flag_helpcCr)z#Print the options part of the help.rN)rremit_options_helprr<r<r= print_optionsCrzApplication.print_optionsccsl|js |js dSd}|Vdt|Vt|jD]}|VdVq|EdH|EdHdVdS)z1Yield the lines for the options part of the help.NOptionsrr%)rrrrr]rr)rSheaderpr<r<r=rGs  zApplication.emit_options_helpcCr)z&Print the subcommand part of the help.rN)rremit_subcommands_helprr<r<r=print_subcommandsVrzApplication.print_subcommandsccs|jsdSd}|Vdt|Vt|jj|jdD]}|VdVq|jD]\}\}}|V|r>tt| Vq*dVdS)z4Yield the lines for the subcommand part of the help.N Subcommandsr)r/r%) rrrr_rRr[rrrr)rSrrsubc_ryr<r<r=r Zs z!Application.emit_subcommands_helpboolccs|s dVdVdSdS)zzYield the very bottom lines of the help message. If classes=False (the default), print `--help-all` msg. z5To see all available configurables, use `--help-all`.r%Nr<rSrbr<r<r=emit_help_epilogueks  zApplication.emit_help_epilogueFcCstd|j|ddS)zPrint the help for each Configurable class in self.classes. If classes=False (the default), only flags and aliases are printed. rrbN)rr emit_helprr<r<r= print_helptszApplication.print_helpccs|EdH|EdH|EdH|rA|}|dur4dVdVt|jD]}|VdVq+|D] }|VdVq6|EdH||EdHdS)zYield the help-lines for each Configurable class in self.classes. If classes=False (the default), only flags and aliases are printed. Nz Class optionsz =============r%) emit_descriptionr r_classes_with_config_traitsrr^class_get_help emit_examplesr)rSrb help_classesrrr<r<r=r{s" zApplication.emit_helprKcCsddd|DS)zwGenerate rST format documentation for the config options this application Returns a multiline string. rcss|]}|VqdSr)class_config_rst_doc)rrkr<r<r=rz6Application.document_config_options..)rrmrr<r<r=document_config_optionssz#Application.document_config_optionscCr)z"Print the application description.rN)rrrrr<r<r=print_descriptionrzApplication.print_descriptionccs,t|jp |jp dD]}|VdVq dS)z-Yield lines with the application description.r%N)rr\rV)rSrr<r<r=rs zApplication.emit_descriptioncCr)z1Print usage and examples (see `emit_examples()`).rN)rrrrr<r<r=print_examplesrzApplication.print_examplesccs<|jrdVdVdVtt|jVdVdSdS)zYield lines with the usage and examples. This usage string goes at the end of the command line help string and should contain examples of the application's usage. Examplesz--------r%N)r`rrrrr<r<r=rs zApplication.emit_examplescCst|jdS)zPrint the version string.N)rrnrr<r<r= print_versionszApplication.print_versionr cCs|j|}|dus J|\}}t|trt|}t|tr0t|tr0|j |j |d|_ nt |r:|||_ nt d||j |dS)z"Initialize a subcommand with argv.N)rlz%Invalid mappings for subcommand '%s'!)rgetrrKr#rrhr0rUclear_instanceinstancercallableAssertionErrorr)rSr r$rrr r<r<r=initialize_subcommands      z!Application.initialize_subcommand)tuple[dict[str, t.Any], dict[str, t.Any]]cCs^tt}|jD]}|j}|ddD] }||j|qqi}|jD]9\}}t|t r4|\}}| dd\}} ||} t | dkrJ| d}t|t sR|f}|D] } d || g|| <qTq'i} |j D]B\} \}}i}|D]"\}}||} t | dkr| d}||vr|||qt|||<qtt| t s| f} | D]}||f| |<qqh| |fS)aFlatten flags and aliases for loaders, so cl-args override as expected. This prevents issues such as an alias pointing to InteractiveShell, but a config file setting the same trait in TerminalInteraciveShell getting inappropriate priority over the command-line arg. Also, loaders expect ``(key: longname)`` and not ``key: (longname, help)`` items. Only aliases with exactly one descendent in the class list will be promoted. r4rrr)rlistrbrErgappendrrrrrrrrupdate)rSmro_treerclsnamerlrrrr rchildrenalrrflagdictrynewflagsubdictkr<r<r= flatten_flagssF       zApplication.flatten_flagslist[str] | NonercCst|||||j|jdS)N)rbr5r)rr5r)rSr$rrrbr<r<r=_create_loaderszApplication._create_loadercheck_argcomplete list[str]c CsT|r'dtjvr'zddlm}|}|dusJ|WSttfy&YtjSwtjS)aGet `sys.argv` or equivalent from `argcomplete` `argcomplete`'s strategy is to call the python script with no arguments, so ``len(sys.argv) == 1``, and run until the `ArgumentParser` is constructed and determine what completions are available. On the other hand, `traitlet`'s subcommand-handling strategy is to check ``sys.argv[1]`` and see if it matches a subcommand, and if so then dynamically load the subcommand app and initialize it with ``sys.argv[1:]``. This helper method helps to take the current tokens for `argcomplete` and pass them through as `argv`. _ARGCOMPLETEr)get_argcomplete_cwordsN)osenviron#traitlets.config.argcomplete_configr8 ImportErrorModuleNotFoundErrorrr$)rr5r8cwordsr<r<r= _get_sys_argvs  zApplication._get_sys_argvc Cs@dtjvrdSz ddlm}|WdSttfyYdSw)aHelper for `argcomplete` to recognize `traitlets` subcommands `argcomplete` does not know that `traitlets` has already consumed subcommands, as it only "sees" the final `argparse.ArgumentParser` that is constructed. (Indeed `KVArgParseConfigLoader` does not get passed subcommands at all currently.) We explicitly manipulate the environment variables used internally by `argcomplete` to get it to skip over the subcommand tokens. r7Nr)increment_argcomplete_index)r9r:r;r@r<r=)rr@r<r<r="_handle_argcomplete_for_subcommand3s   z.Application._handle_argcomplete_for_subcommandcst|trJ|dur|jt|jddd}dd|D|_|r1|ddkr1|dddg}|jrZt|dkrZ|d|dd}}td |rZ||jvrZ| | ||Sz |d| d Wn t yp|Ynwt fd d d Dr|dv|ddvsdvr||d|\}}t|}|j||||d}z t||_Wntyw||j|j|_dS)z!Parse the command line arguments.N)r5r4cSsg|]}t|qSr<r!)rargr<r<r= Msz2Application.parse_command_line..rry-hz ^\w(\-?\w)*$rc3s|]}|vVqdSrr<)rxinterpreted_argvr<r=rdrz1Application.parse_command_line..)rD --help-allz--helprHz --versionz-Vr)rrKr?rrr$rrematchrAr%index ValueErroranyrr9rr2r'rr4r load_config cli_config SystemExit update_config extra_args)rSr$r subargvrrrbloaderr<rFr=rGs@         zApplication.parse_command_line basefilenamepath#str | t.Sequence[str | None] | NoneAnyLogger | Noneraise_config_file_errorst.Generator[t.Any, None, None]c csZt|ts |dur |g}t|D]}|j|d||d}|r*|d||p(t|j|d||d}g}g} ||fD]m} d} z| } Wn#t yOYn$t yi| j pY|} |r]|rg|j d| ddYn w|rs|d | j | rt | |D]\} } | | }|r|r|d | | j tj|d d qz| | j fV|| | | j q.rr2r.rcstfdd|jDS)Nc3s |] }|vo |VqdSrr<)rb cls_to_configr<r=rszZApplication._classes_with_config_traits..is_any_parent_included..)rM __bases__)rr~r<r=is_any_parent_includedszGApplication._classes_with_config_traits..is_any_parent_includedTc3s$|] \}}||p |fVqdSrr<)rrinc_yes)rr<r=rs  )rr2r.r)rbrrmrr)rSrb to_incl_origclrr<)rrr=rs( z'Application._classes_with_config_traitscCsnd|jg}|d|d|d|dur|jn|}t||}|D] }|||q'd|S)z/generate default config file from Configurablesz# Configuration file for %s.r%zc = get_config() #noqaNr)r[r(rbr'rclass_config_sectionr)rSrblinesconfig_classesrr<r<r=generate_config_file s     z Application.generate_config_filec CsXt|ddr*|jjD]}tt |Wdn1swYq d|_dSdS)NrF)rr5rrrcloser)rShandlerr<r<r=close_handlerss     zApplication.close_handlersr exit_statusint | str | NonecCs&|jd|j|t|dS)NzExiting application: %s)r5r7r[rrr9)rSrr<r<r=r9#szApplication.exitcCrr)rrr<r<r=__del__(s zApplication.__del__cKs&|jdi|}|||dS)zLaunch a global instance of this Application If a global instance already exists, this reinitializes and starts it Nr<)r"rr)rr$r3r/r<r<r=launch_instance+s  zApplication.launch_instancer)rbrcr.rd)r.r{)rrr.r)r.r)r.r)r3r2r.r)r$rr.r)r.r)rbrr.r)F)rbrr.r)r.rK)r rKr$rr.r)r.r&) r$r3rr{rr{rbrcr.r)r5rr.r6)NF) rUrKrVrWr5rXrYrr.rZ)r.r6)rirKrVrWr.r)rbrcr.rK)r)rrr.r)r$rr3r2r.r)crErFrGrVrr[__annotations__r\r]r^r_rr\rr^r`rbrmrnrr$rr&rYrrWrXtagrrHrrrrrrrrrrrrrrtrrrrrRr rOrrrrrrrrrCrrrrrrrrrr r rrrrrrrrrr%r2r4 classmethodr?rArrkpropertyrlrrr{rrrr9rrrYr<r<rTr=r0s$     3      07            !  &               ;  27   * r0r[rK configurableset_help unset_helpr{cCsV|pd|}|p d|}|d\}}||dii}||dii}|||fd|||fiS)aHelper for building basic --trait, --no-trait flags. Parameters ---------- name : str The name of the flag. configurable : str The 'Class.trait' string of the trait to be set/unset with the flag set_help : unicode help string for --name flag unset_help : unicode help string for --no-name flag Returns ------- cfg : dict A dict with two keys: 'name', and 'no-name', for setting and unsetting the trait, respectively. z set %s=Truez set %s=FalserTFzno-)r)r[rrrrrsetterunsetterr<r<r= boolean_flag>s    rr cCstr tjStS)z|Get the config object for the global Application instance, if there is one otherwise return an empty config object )r0 initializedr"r8r r<r<r<r= get_config]s r__main__)r-r+r.r+)r%r%) r[rKrrKrrKrrKr.r{)r.r )brV __future__rr?rrWr9rrIrtypingrA collectionsrr contextlibrrrlogging.configrtextwraprtraitlets.config.configurabler r traitlets.config.loaderr r r rrrrtraitlets.traitletsrrrrrrrrrrtraitlets.utils.bunchrtraitlets.utils.nested_updatertraitlets.utils.textrrutilsr"utils.importstringr#rr]r^hasattrr$r_r:r _envvarlowerr&rL executableendswithrTypeVarCallableAnyr+UnionLoggerrrKr{OptionalrrrarCrrD FormatterrHr0rdefault_aliasesr default_flagsrrrErr<r<r<r=s     $0       )