U f{jHA@sdZddlZddlmZddlZddlZddlmZddlZddl Z ddl Z ddl m Z ddlZddlmZmZmZddlmZe dZd e_Gd d d e jZeZeeed ed efdZeedddZeGdddZed0ddZd1ddZ ddZ!GdddeZ"Gddde#Z$ej%dd Z&Gd!d"d"e#Z'e(d#Z)e*dd$fd%d&Z+d'd(Z,d2d)d*Z-ej%d3d,d-Z.d.d/Z/dS)4zSome utility functions.N) FunctionMaker)StringIO)AnyCallableTypeVar)fill_docmneFc@seZdZddZddZdS) _FrameFiltercCs d|_dS)Nr add_framesselfrQ/var/www/html/bwave/python/venv/lib/python3.8/site-packages/mne/utils/_logging.py__init__sz_FrameFilter.__init__cCsd|_|jrtd|jddddd}t|r|ddd|d<t|dkrvd|d|d<d |d|d<t|dddD]\}}d |||<qd ||_d S) NUnknownz :ru┌ru└u├ T) frame_infor _frame_infolenljust enumeratejoin)rrecordriiinforrrfilter!s   z_FrameFilter.filterN)__name__ __module__ __qualname__rr rrrrr sr _FuncT.)bound)functionreturncCspz t|Wntk r YnXd}tt|d}t|dddd|j}t||j|jd}|j||fddi|S)aVerbose decorator to allow functions to override log-level. Parameters ---------- function : callable Function to be decorated by setting the verbosity level. Returns ------- dec : callable The decorated function. See Also -------- set_log_level set_config Notes ----- This decorator is used to set the verbose level during a function or method call, such as :func:`mne.compute_covariance`. The `verbose` keyword argument can be 'DEBUG', 'INFO', 'WARNING', 'ERROR', 'CRITICAL', True (an alias for 'INFO'), or False (an alias for 'WARNING'). To set the global verbosity level for all functions, use :func:`mne.set_log_level`. This function also serves as a docstring filler. Examples -------- You can use the ``verbose`` argument to set the verbose level on the fly:: >>> import mne >>> cov = mne.compute_raw_covariance(raw, verbose='WARNING') # doctest: +SKIP >>> cov = mne.compute_raw_covariance(raw, verbose='INFO') # doctest: +SKIP Using up to 49 segments Number of samples used : 5880 [done] adef %(name)s(%(signature)s): try: do_level_change = verbose is not None except (NameError, UnboundLocalError): raise RuntimeError('Function/method %%s does not accept verbose ' 'parameter' %% (_function_,)) from None if do_level_change: with _use_log_level_(verbose): return _function_(%(shortsignature)s) else: return _function_(%(shortsignature)s))Z_use_log_level_Z _function_N) __wrapped__r# __globals__Z addsourceT) r TypeErrordict use_log_levelrr"r#r)make)r&bodyZevaldictfmattrsrrrverbose;s)  r1c@s.eZdZdZddddZddZdd ZdS) r,a Context manager for logging level. Parameters ---------- %(verbose)s %(add_frames)s See Also -------- mne.verbose Notes ----- See the :ref:`logging documentation ` for details. Examples -------- >>> from mne import use_log_level >>> from mne.utils import logger >>> with use_log_level(False): ... # Most MNE logger messages are "info" level, False makes them not ... # print: ... logger.info('This message will not be printed') >>> with use_log_level(True): ... # Using verbose=True in functions, methods, or this context manager ... # will ensure they are printed ... logger.info('This message will be printed!') This message will be printed! Nr cCs||_||_tj|_dSN)_level _add_frames_filterr _old_frames)rr1r rrrrszuse_log_level.__init__cCst|jd|jd|_dS)NT)return_old_levelr ) set_log_levelr3r4 _old_levelr rrr __enter__s zuse_log_level.__enter__cGs&|jdk r|jnd}t|j|ddS)Nr )r4r6r8r9)rargsr rrr__exit__szuse_log_level.__exit__r!r"r#__doc__rr:r<rrrrr,sr,c Csddlm}ddlm}m}||tttdfd|dkrD|dd}nt|tr`|dkr\d}nd }t|tr| }t t j t j t jt jt jd }|d||d ||}tj}||krt||dk rt|t_|rd nd }|d7}t |}tjD]} | |q|r|SdS)aSet the logging level. Parameters ---------- verbose : bool, str, int, or None The verbosity of messages to print. If a str, it can be either DEBUG, INFO, WARNING, ERROR, or CRITICAL. Note that these are for convenience and are equivalent to passing in logging.DEBUG, etc. For bool, True is the same as 'INFO', False is the same as 'WARNING'. If None, the environment variable MNE_LOGGING_LEVEL is read, and if it doesn't exist, defaults to INFO. return_old_level : bool If True, return the old verbosity level. %(add_frames)s Returns ------- old_level : int The old level. Only returned if ``return_old_level`` is True. r) get_config) _check_option_validate_typeNr1ZMNE_LOGGING_LEVELINFOTWARNING)DEBUGrBrCERRORCRITICALz(when a string)z%(frame_info)s %(message)s)configr?checkr@rAboolstrint isinstanceupperr+loggingrDrBrCrErFloggerlevelsetLevelr5r Formatterhandlers setFormatter) r1r7r r?r@rAZ logging_typesZ old_verbosefmthandlerrrrr8s:            r8rHcCs|tt|dk rRt|r6|dkr6tjdtddd}|r>dnd}tj||d}n t t }| t |t |dS) aSet the log to print to a file. Parameters ---------- fname : str, or None Filename of the log to print to. If None, stdout is used. To suppress log outputs, use set_log_level('WARN'). output_format : str Format of the output messages. See the following for examples: https://docs.python.org/dev/howto/logging.html e.g., "%(asctime)s - %(levelname)s - %(message)s". overwrite : bool | None Overwrite the log file (if it exists). Otherwise, statements will be appended to the log (default). None is the same as False, but additionally raises a warning to notify the user that log entries will be appended. NzbLog entries will be appended to the file. Use overwrite=False to avoid this message in the future.) stacklevelFwa)mode)_remove_close_handlersrQopisfilewarningswarnRuntimeWarningrP FileHandler StreamHandler WrapStdOutrVrT addHandler)fnameZ output_format overwriter]Zlhrrr set_log_files  rjcCsDt|jD]4}t|tjtjfr t|tjr4|||q dSr2)listrUrNrPrdreclose removeHandler)rQhrrrr^ s  r^cs"eZdZdZdfdd ZZS)ClosingStringIOz&StringIO that closes after getvalue().Tcst}|r||S)zGet the value.)supergetvaluerl)rrlout __class__rrrqs zClosingStringIO.getvalue)T)r!r"r#r>rq __classcell__rrrsrrosroc@s*eZdZdZd ddZddZddZdS) catch_loggingzyStore logging. This will remove all other logging handlers, and return the handler to stdout when complete. NcCs ||_dSr2)r1)rr1rrrr'szcatch_logging.__init__cCsv|jdk rt|j|_n t|_t|_t|j|_ |j t dd|j _ t tt|j |j|jS)NrHT)r1r,_ctx contextlib nullcontextro_datarPre_lhrVrT_mne_file_liker^rQrgr:r rrrr:*s    zcatch_logging.__enter__cGs$|jj|t|jtddSr2)rwr<rQrmr{rj)rr;rrrr<8s  zcatch_logging.__exit__)Nr=rrrrrv s rvc cs,tjdd}td|VW5QRXdS)NT)ralways)racatch_warnings simplefilter)r[rrr_record_warnings>s rc@seZdZdZddZdS)rfzDynamically wrap to sys.stdout. This makes packages that monkey-patch sys.stdout (e.g.doctest, sphinx-gallery) work properly. cCs(ttj|rttj|Std|dS)Nz$'file' object has not attribute '%s')hasattrsysstdoutgetattrAttributeError)rnamerrr __getattr__Ns  zWrapStdOut.__getattr__N)r!r"r#r>rrrrrrfGsrfz^$)r c sdd|D}dd|D}d}tjtjkrt}|r|jj|j}t st fdd|Dsvdkrdkrt t d krq|j}q4~t||||td it d dtjDstjrt|dS) aEmit a warning with trace outside the mne namespace. This function takes arguments like warnings.warn, and sends messages using both ``warnings.warn`` and ``logger.warn``. Warnings can be generated deep within nested function calls. In order to provide a more helpful warning, this function traverses the stack until it reaches a frame outside the ``mne`` namespace that caused the error. Parameters ---------- message : str Warning message. category : instance of Warning The warning class. Defaults to ``RuntimeWarning``. module : str The name of the module emitting the warning. ignore_namespaces : list of str Namespaces to ignore when traversing the stack. .. versionadded:: 0.24 cSsg|]}t|qSr) importlib import_module.0nsrrr qszwarn..cSsg|]}t|jqSr)r_dirname__file__rrrrrrsNc3s|]}|VqdSr2) startswith)rrdrhrr }szwarn..ZunittestcaseZtests__warningregistry__css&|]}t|tjpt|ddVqdS)r|FN)rNrPrdr)rrnrrrrs)rQrRrPWARNinspect currentframef_code co_filenamef_lineno_verbose_dec_researchanyr_basenamerf_backra warn_explicitglobalsgetrUr5r warning)messagecategorymoduleZignore_namespacesZ root_dirsframelinenorrrrbZsB   rbcCsHtjj}t|jjr |j}t|j}|dkr8dn |d }|S)z)Get the call line from within a function.Nunknownr) rrrrrrr getframeinfo code_contextstrip)rcontextrrr_get_call_lines   rcstt|dkrdn|fdd|jD}fdd|jD}ddtt||D}t|D]}|j|q^dS)aRemove particular records from ``warn_record``. This helper takes a list of :class:`warnings.WarningMessage` objects, and remove those matching category and/or text. Parameters ---------- category: WarningMessage type | None class of the message to filter out match : str | None text or regex that matches the error message to filter out Nz.*cs"g|]}dk r|jkndqS)NTrrr[rrrrsz'filter_out_warnings..cs"g|]}|jjddk qS)rN)matchrr;r)regexprrrscSs g|]\}\}}|r|r|qSrr)rindcmrrrrs )recompileZ_listrzipreversedpop)Z warn_recordrrZ is_categoryZis_matchrir)rrrfilter_out_warningss   rrGc cs~tj}t}|t_z dVW5|t_d}|dD]@}|sP|rP|d7}q6t|D]}tdqXt||q6XdS)zWrap stdout writes to logger.info, with an optional indent prefix. Parameters ---------- indent : str The indentation to add. cull_newlines : bool If True, cull any new/blank lines at the end. rrrN) rrrorqsplitrrangerQr)indentZ cull_newlinesZ orig_stdoutZmy_outZpending_newlinesline_rrrwrapped_stdouts     rc Cst}zzv|j}t}t|D]V}z|jd}Wntk rFYnX||dd|j |j}|dkr qxq |WW$St k rdgYWSXW5~XdS)Nr!zmne.:r) rrrrkr f_globalsKeyErrorappendlstripr Exception)nrinfosrrrrrrs"  r)NFN)NrHN)NN)rGF)0r>rx decoratorrrriorrrrPos.pathpathr_ratypingrrrdocsr getLoggerrQ propagateFilterr r5 addFilterr$r1r,r8rjr^roobjectrvcontextmanagerrrfrrrcrbrrrrrrrrsN      D- 3 +     :