]> crepu.dev Git - config.git/blame_incremental - djavu-asus/elpy/rpc-venv/lib/python3.11/site-packages/jedi/debug.py
ActualizaciĆ³n de Readme
[config.git] / djavu-asus / elpy / rpc-venv / lib / python3.11 / site-packages / jedi / debug.py
... / ...
CommitLineData
1import os
2import time
3from contextlib import contextmanager
4from typing import Callable, Optional
5
6_inited = False
7
8
9def _lazy_colorama_init():
10 """
11 Lazily init colorama if necessary, not to screw up stdout if debugging is
12 not enabled.
13
14 This version of the function does nothing.
15 """
16
17
18try:
19 if os.name == 'nt':
20 # Does not work on Windows, as pyreadline and colorama interfere
21 raise ImportError
22 else:
23 # Use colorama for nicer console output.
24 from colorama import Fore, init # type: ignore[import]
25 from colorama import initialise
26
27 def _lazy_colorama_init(): # noqa: F811
28 """
29 Lazily init colorama if necessary, not to screw up stdout is
30 debug not enabled.
31
32 This version of the function does init colorama.
33 """
34 global _inited
35 if not _inited:
36 # pytest resets the stream at the end - causes troubles. Since
37 # after every output the stream is reset automatically we don't
38 # need this.
39 initialise.atexit_done = True
40 try:
41 init(strip=False)
42 except Exception:
43 # Colorama fails with initializing under vim and is buggy in
44 # version 0.3.6.
45 pass
46 _inited = True
47
48except ImportError:
49 class Fore: # type: ignore[no-redef]
50 RED = ''
51 GREEN = ''
52 YELLOW = ''
53 MAGENTA = ''
54 RESET = ''
55 BLUE = ''
56
57NOTICE = object()
58WARNING = object()
59SPEED = object()
60
61enable_speed = False
62enable_warning = False
63enable_notice = False
64
65# callback, interface: level, str
66debug_function: Optional[Callable[[str, str], None]] = None
67_debug_indent = 0
68_start_time = time.time()
69
70
71def reset_time():
72 global _start_time, _debug_indent
73 _start_time = time.time()
74 _debug_indent = 0
75
76
77def increase_indent(func):
78 """Decorator for makin """
79 def wrapper(*args, **kwargs):
80 with increase_indent_cm():
81 return func(*args, **kwargs)
82 return wrapper
83
84
85@contextmanager
86def increase_indent_cm(title=None, color='MAGENTA'):
87 global _debug_indent
88 if title:
89 dbg('Start: ' + title, color=color)
90 _debug_indent += 1
91 try:
92 yield
93 finally:
94 _debug_indent -= 1
95 if title:
96 dbg('End: ' + title, color=color)
97
98
99def dbg(message, *args, color='GREEN'):
100 """ Looks at the stack, to see if a debug message should be printed. """
101 assert color
102
103 if debug_function and enable_notice:
104 i = ' ' * _debug_indent
105 _lazy_colorama_init()
106 debug_function(color, i + 'dbg: ' + message % tuple(repr(a) for a in args))
107
108
109def warning(message, *args, format=True):
110 if debug_function and enable_warning:
111 i = ' ' * _debug_indent
112 if format:
113 message = message % tuple(repr(a) for a in args)
114 debug_function('RED', i + 'warning: ' + message)
115
116
117def speed(name):
118 if debug_function and enable_speed:
119 now = time.time()
120 i = ' ' * _debug_indent
121 debug_function('YELLOW', i + 'speed: ' + '%s %s' % (name, now - _start_time))
122
123
124def print_to_stdout(color, str_out):
125 """
126 The default debug function that prints to standard out.
127
128 :param str color: A string that is an attribute of ``colorama.Fore``.
129 """
130 col = getattr(Fore, color)
131 _lazy_colorama_init()
132 print(col + str_out + Fore.RESET)