jddlmZddlZddlZddlmZmZmZm Z m Z m Z m Z m Z mZddlmZddgZGdde ZGd d e ZGd dZeed d d dddZdS)) annotationsN) CRITICALDEBUGERRORFilter Formatter getLoggerINFO StreamHandlerWARNING)stderrWatcherwatchc"eZdZdZfdZxZS)ColourFormatterz- Colour formatter for pretty log output. ct|}|jtkrd|zS|jtkrd|zS|jt krd|zS|jt krd|zS|jtkrd|zS|S)Nz %sz %sz %sz %sz %s)superformatlevelnorrr r r)selfrecords __class__s U/lsinfo/ai/hellotax_ai/base_platform/venv/lib/python3.11/site-packages/neo4j/debug.pyrzColourFormatter.format-s GGNN6 " " >X % %(1, , ^u $ $(1, , ^w & &&* * ^t # #&* * ^u $ $&* *H)__name__ __module__ __qualname____doc__r __classcell__rs@rrr)sB         rrceZdZdZdZdS) TaskIdFilterz)Injecting async task id into log records.c ttj|_n#t$r d|_YnwxYwdS)NT)idasyncio current_tasktask RuntimeError)rrs rfilterzTaskIdFilter.filter@sL W13344FKK   FKKK ts %(<<N)rrrrr*rrr#r#=s)33rr#cTeZdZdZeedddddfdZddZdZ dddZ ddZ xZ S) ra5Log watcher for easier logging setup. Example:: from neo4j.debug import Watcher with Watcher("neo4j"): # DEBUG logging to stderr enabled within this context ... # do something .. note:: The Watcher class is not thread-safe. Having Watchers in multiple threads can lead to duplicate log messages as the context manager will enable logging for all threads. .. note:: The exact logging format and messages are not part of the API contract and might change at any time without notice. They are meant for debugging purposes and human consumption only. :param logger_names: Names of loggers to watch. :param default_level: Default minimum log level to show. The level can be overridden by setting ``level`` when calling :meth:`.watch`. :param default_out: Default output stream for all loggers. The level can be overridden by setting ``out`` when calling :meth:`.watch`. :type default_out: stream or file-like object :param colour: Whether the log levels should be indicated with ANSI colour codes. :param thread_info: whether to include information about the current thread in the log message. Defaults to :const:`True`. :param task_info: whether to include information about the current async task in the log message. Defaults to :const:`True`. .. versionchanged:: 5.3 * Added ``thread_info`` and ``task_info`` parameters. * Logging format around thread and task information changed. FT default_level default_outcolour thread_info task_info logger_namest.Optional[str]r.intr/t.TextIOr0boolr1r2returnNonec6tt|||_d|jD|_||_||_i|_||_d}|rd|z}|rd|z}|sd|z}|rtnt}|||_ dS)Nc,g|]}t|Sr+)r ).0names r z$Watcher.__init__..|sGGGT4GGGrz%(asctime)s %(message)sz[Task %(task)-15s] z[Thread %(thread)d] z[%(levelname)-8s] ) rr__init__r3_loggersr.r/ _handlers _task_inforr formatter) rr.r/r0r1r2r3format_ formatter_clsrs rr?zWatcher.__init__qs gt%%'''(GGT5FGGG *&57#,  6+g5G  7,w6G 5*W4G+1@y &w//rc.||S)zEnable logging for all loggers.)r)rs r __enter__zWatcher.__enter__s  rc.|dSz Disable logging for all loggers.N)stop)rexc_typeexc_valexc_tbs r__exit__zWatcher.__exit__s rNlevelt.Optional[int]outt.Optional[t.TextIO]c||j}||j}|t|}||j|||jr!|t|j D]S}||j |j <| |||kr||TdS)a1Enable logging for all loggers. :param level: Minimum log level to show. If :data:`None`, the ``default_level`` is used. :param out: Output stream for all loggers. If :data:`None`, the ``default_out`` is used. :type out: stream or file-like object N)r.r/rJr setFormatterrCsetLevelrB addFilterr#r@rAr= addHandlergetEffectiveLevel)rrOrQhandlerloggers rrz Watcher.watchs =&E ;"C $$T^,,, ? .   lnn - - -n ' 'F*1DN6; '   g & & &''))E11&&&  ' 'rc|jD]E} ||j|j6#t $rYBwxYwdSrI)r@ removeHandlerrApopr=KeyError)rrZs rrJz Watcher.stopskm  F $$T^%7%7 %D%DEEEE      s2> A  A )r3r4r.r5r/r6r0r7r1r7r2r7r8r9)r8r)NN)rOrPrQrRr8r9)r8r9) rrrrrr r?rGrNrrJr r!s@rrrHs&&V# & 000000006  JN'''''6rFT)rOrQr0r1r2r3r4rOr5rQr6r0r7r1r2r8c Nt||||||d}||S)ahQuick wrapper for using :class:`.Watcher`. Create a Watcher with the given configuration, enable watching and return it. Example:: from neo4j.debug import watch watch("neo4j") # from now on, DEBUG logging to stderr is enabled in the driver .. note:: The exact logging format and messages are not part of the API contract and might change at any time without notice. They are meant for debugging purposes and human consumption only. :param logger_names: Names of loggers to watch. :param level: see ``default_level`` of :class:`.Watcher`. :param out: see ``default_out`` of :class:`.Watcher`. :type out: stream or file-like object :param colour: see ``colour`` of :class:`.Watcher`. :param thread_info: see ``thread_info`` of :class:`.Watcher`. :param task_info: see ``task_info`` of :class:`.Watcher`. :returns: Watcher instance :rtype: :class:`.Watcher` .. versionchanged:: 5.3 * Added ``thread_info`` and ``task_info`` parameters. * Logging format around thread and task information changed. r-)rr)rOrQr0r1r2r3watchers rrrs;R|5c# )+++G MMOOO Nr)r3r4rOr5rQr6r0r7r1r7r2r7r8r) __future__rr&typingtloggingrrrrrr r r r sysr __all__rr#rrr+rrrgs}"#"""""                         i(6nnnnnnnnf --------r