Admin Control Panel v2.0

✎ Editing: debug.py ← Back

#
# 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

__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)), 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()

Cancel