158 lines
3.6 KiB
Python
158 lines
3.6 KiB
Python
|
#
|
||
|
# This file is part of pyasn1 software.
|
||
|
#
|
||
|
# Copyright (c) 2005-2019, Ilya Etingof <etingof@gmail.com>
|
||
|
# License: http://snmplabs.com/pyasn1/license.html
|
||
|
#
|
||
|
import logging
|
||
|
import sys
|
||
|
|
||
|
from pyasn1 import __version__
|
||
|
from pyasn1 import error
|
||
|
from pyasn1.compat.octets import octs2ints
|
||
|
|
||
|
__all__ = ['Debug', 'setLogger', 'hexdump']
|
||
|
|
||
|
DEBUG_NONE = 0x0000
|
||
|
DEBUG_ENCODER = 0x0001
|
||
|
DEBUG_DECODER = 0x0002
|
||
|
DEBUG_ALL = 0xffff
|
||
|
|
||
|
FLAG_MAP = {
|
||
|
'none': DEBUG_NONE,
|
||
|
'encoder': DEBUG_ENCODER,
|
||
|
'decoder': DEBUG_DECODER,
|
||
|
'all': DEBUG_ALL
|
||
|
}
|
||
|
|
||
|
LOGGEE_MAP = {}
|
||
|
|
||
|
|
||
|
class Printer(object):
|
||
|
# noinspection PyShadowingNames
|
||
|
def __init__(self, logger=None, handler=None, formatter=None):
|
||
|
if logger is None:
|
||
|
logger = logging.getLogger('pyasn1')
|
||
|
|
||
|
logger.setLevel(logging.DEBUG)
|
||
|
|
||
|
if handler is None:
|
||
|
handler = logging.StreamHandler()
|
||
|
|
||
|
if formatter is None:
|
||
|
formatter = logging.Formatter('%(asctime)s %(name)s: %(message)s')
|
||
|
|
||
|
handler.setFormatter(formatter)
|
||
|
handler.setLevel(logging.DEBUG)
|
||
|
logger.addHandler(handler)
|
||
|
|
||
|
self.__logger = logger
|
||
|
|
||
|
def __call__(self, msg):
|
||
|
self.__logger.debug(msg)
|
||
|
|
||
|
def __str__(self):
|
||
|
return '<python logging>'
|
||
|
|
||
|
|
||
|
if hasattr(logging, 'NullHandler'):
|
||
|
NullHandler = logging.NullHandler
|
||
|
|
||
|
else:
|
||
|
# Python 2.6 and older
|
||
|
class NullHandler(logging.Handler):
|
||
|
def emit(self, record):
|
||
|
pass
|
||
|
|
||
|
|
||
|
class Debug(object):
|
||
|
defaultPrinter = Printer()
|
||
|
|
||
|
def __init__(self, *flags, **options):
|
||
|
self._flags = DEBUG_NONE
|
||
|
|
||
|
if 'loggerName' in options:
|
||
|
# route our logs to parent logger
|
||
|
self._printer = Printer(
|
||
|
logger=logging.getLogger(options['loggerName']),
|
||
|
handler=NullHandler()
|
||
|
)
|
||
|
|
||
|
elif 'printer' in options:
|
||
|
self._printer = options.get('printer')
|
||
|
|
||
|
else:
|
||
|
self._printer = self.defaultPrinter
|
||
|
|
||
|
self._printer('running pyasn1 %s, debug flags %s' % (__version__, ', '.join(flags)))
|
||
|
|
||
|
for flag in flags:
|
||
|
inverse = flag and flag[0] in ('!', '~')
|
||
|
if inverse:
|
||
|
flag = flag[1:]
|
||
|
try:
|
||
|
if inverse:
|
||
|
self._flags &= ~FLAG_MAP[flag]
|
||
|
else:
|
||
|
self._flags |= FLAG_MAP[flag]
|
||
|
except KeyError:
|
||
|
raise error.PyAsn1Error('bad debug flag %s' % flag)
|
||
|
|
||
|
self._printer("debug category '%s' %s" % (flag, inverse and 'disabled' or 'enabled'))
|
||
|
|
||
|
def __str__(self):
|
||
|
return 'logger %s, flags %x' % (self._printer, self._flags)
|
||
|
|
||
|
def __call__(self, msg):
|
||
|
self._printer(msg)
|
||
|
|
||
|
def __and__(self, flag):
|
||
|
return self._flags & flag
|
||
|
|
||
|
def __rand__(self, flag):
|
||
|
return flag & self._flags
|
||
|
|
||
|
_LOG = DEBUG_NONE
|
||
|
|
||
|
|
||
|
def setLogger(userLogger):
|
||
|
global _LOG
|
||
|
|
||
|
if userLogger:
|
||
|
_LOG = userLogger
|
||
|
else:
|
||
|
_LOG = DEBUG_NONE
|
||
|
|
||
|
# Update registered logging clients
|
||
|
for module, (name, flags) in LOGGEE_MAP.items():
|
||
|
setattr(module, name, _LOG & flags and _LOG or DEBUG_NONE)
|
||
|
|
||
|
|
||
|
def registerLoggee(module, name='LOG', flags=DEBUG_NONE):
|
||
|
LOGGEE_MAP[sys.modules[module]] = name, flags
|
||
|
setLogger(_LOG)
|
||
|
return _LOG
|
||
|
|
||
|
|
||
|
def hexdump(octets):
|
||
|
return ' '.join(
|
||
|
['%s%.2X' % (n % 16 == 0 and ('\n%.5d: ' % n) or '', x)
|
||
|
for n, x in zip(range(len(octets)), octs2ints(octets))]
|
||
|
)
|
||
|
|
||
|
|
||
|
class Scope(object):
|
||
|
def __init__(self):
|
||
|
self._list = []
|
||
|
|
||
|
def __str__(self): return '.'.join(self._list)
|
||
|
|
||
|
def push(self, token):
|
||
|
self._list.append(token)
|
||
|
|
||
|
def pop(self):
|
||
|
return self._list.pop()
|
||
|
|
||
|
|
||
|
scope = Scope()
|