123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147 |
- #
- # This file is part of pyasn1 software.
- #
- # Copyright (c) 2005-2020, Ilya Etingof <etingof@gmail.com>
- # License: https://pyasn1.readthedocs.io/en/latest/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>'
-
-
- 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=logging.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()
|