forked from naksyn/PythonMemoryModule
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathdbgprint.py
More file actions
53 lines (42 loc) · 1.44 KB
/
dbgprint.py
File metadata and controls
53 lines (42 loc) · 1.44 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
import os
import sys
import logging
import inspect
options = {'active': False, 'cats': None}
# options = {'active': True, 'cats': ["HANDLE"]}
def get_stack_func_name(lvl):
info = inspect.stack()[lvl]
return info[0], info[3]
def do_dbgprint(msg, type=None):
if ("ALL" in options['cats']) or type.upper() in options['cats']:
frame, func = get_stack_func_name(2)
logger = logging.getLogger(frame.f_globals['__name__'] + ":" + func)
logger.debug(msg)
def do_nothing(*args, **kwargs):
return None
def parse_option(s):
if s[0] == "=":
s = s[1:]
if s:
cats = [x.upper().strip() for x in s.split('-')]
options['cats'] = cats
formt = 'DBG|%(name)s|%(message)s'
logging.basicConfig(format=formt, level=logging.DEBUG)
try:
if 'DBGPRINT' in os.environ:
parse_option(os.environ['DBGPRINT'])
dbgprint = do_dbgprint
elif any([opt.startswith("--DBGPRINT") for opt in sys.argv]):
dbgprint = do_dbgprint
option_str = [opt for opt in sys.argv if opt.startswith("--DBGPRINT")][0]
parse_option(option_str[len('--DBGPRINT'):])
elif options["active"]:
formt = 'DBG|%(name)s|%(message)s'
logging.basicConfig(format=formt, level=logging.DEBUG)
dbgprint = do_dbgprint
else:
dbgprint = do_nothing
except Exception as e:
dbgprint = do_nothing
print("dbgprint Error: {0}({1})".format(type(e), e))
x = type(e), e