Coverage for /pythoncovmergedfiles/medio/medio/usr/local/lib/python3.8/site-packages/pyasn1/debug.py: 65%

1# 

2# This file is part of pyasn1 software. 

3# 

4# Copyright (c) 2005-2020, Ilya Etingof <etingof@gmail.com> 

5# License: https://pyasn1.readthedocs.io/en/latest/license.html 

6# 

7import logging 

8import sys 

9 

10from pyasn1 import __version__ 

11from pyasn1 import error 

12from pyasn1.compat.octets import octs2ints 

13 

14__all__ = ['Debug', 'setLogger', 'hexdump'] 

15 

16DEBUG_NONE = 0x0000 

17DEBUG_ENCODER = 0x0001 

18DEBUG_DECODER = 0x0002 

19DEBUG_ALL = 0xffff 

20 

21FLAG_MAP = { 

22 'none': DEBUG_NONE, 

23 'encoder': DEBUG_ENCODER, 

24 'decoder': DEBUG_DECODER, 

25 'all': DEBUG_ALL 

26} 

27 

28LOGGEE_MAP = {} 

29 

30 

31class Printer(object): 

32 # noinspection PyShadowingNames 

33 def __init__(self, logger=None, handler=None, formatter=None): 

34 if logger is None: 

35 logger = logging.getLogger('pyasn1') 

36 

37 logger.setLevel(logging.DEBUG) 

38 

39 if handler is None: 

40 handler = logging.StreamHandler() 

41 

42 if formatter is None: 

43 formatter = logging.Formatter('%(asctime)s %(name)s: %(message)s') 

44 

45 handler.setFormatter(formatter) 

46 handler.setLevel(logging.DEBUG) 

47 logger.addHandler(handler) 

48 

49 self.__logger = logger 

50 

51 def __call__(self, msg): 

52 self.__logger.debug(msg) 

53 

54 def __str__(self): 

55 return '<python logging>' 

56 

57 

58class Debug(object): 

59 defaultPrinter = Printer() 

60 

61 def __init__(self, *flags, **options): 

62 self._flags = DEBUG_NONE 

63 

64 if 'loggerName' in options: 

65 # route our logs to parent logger 

66 self._printer = Printer( 

67 logger=logging.getLogger(options['loggerName']), 

68 handler=logging.NullHandler() 

69 ) 

70 

71 elif 'printer' in options: 

72 self._printer = options.get('printer') 

73 

74 else: 

75 self._printer = self.defaultPrinter 

76 

77 self._printer('running pyasn1 %s, debug flags %s' % (__version__, ', '.join(flags))) 

78 

79 for flag in flags: 

80 inverse = flag and flag[0] in ('!', '~') 

81 if inverse: 

82 flag = flag[1:] 

83 try: 

84 if inverse: 

85 self._flags &= ~FLAG_MAP[flag] 

86 else: 

87 self._flags |= FLAG_MAP[flag] 

88 except KeyError: 

89 raise error.PyAsn1Error('bad debug flag %s' % flag) 

90 

91 self._printer("debug category '%s' %s" % (flag, inverse and 'disabled' or 'enabled')) 

92 

93 def __str__(self): 

94 return 'logger %s, flags %x' % (self._printer, self._flags) 

95 

96 def __call__(self, msg): 

97 self._printer(msg) 

98 

99 def __and__(self, flag): 

100 return self._flags & flag 

101 

102 def __rand__(self, flag): 

103 return flag & self._flags 

104 

105_LOG = DEBUG_NONE 

106 

107 

108def setLogger(userLogger): 

109 global _LOG 

110 

111 if userLogger: 

112 _LOG = userLogger 

113 else: 

114 _LOG = DEBUG_NONE 

115 

116 # Update registered logging clients 

117 for module, (name, flags) in LOGGEE_MAP.items(): 

118 setattr(module, name, _LOG & flags and _LOG or DEBUG_NONE) 

119 

120 

121def registerLoggee(module, name='LOG', flags=DEBUG_NONE): 

122 LOGGEE_MAP[sys.modules[module]] = name, flags 

123 setLogger(_LOG) 

124 return _LOG 

125 

126 

127def hexdump(octets): 

128 return ' '.join( 

129 ['%s%.2X' % (n % 16 == 0 and ('\n%.5d: ' % n) or '', x) 

130 for n, x in zip(range(len(octets)), octs2ints(octets))] 

131 ) 

132 

133 

134class Scope(object): 

135 def __init__(self): 

136 self._list = [] 

137 

138 def __str__(self): return '.'.join(self._list) 

139 

140 def push(self, token): 

141 self._list.append(token) 

142 

143 def pop(self): 

144 return self._list.pop() 

145 

146 

147scope = Scope()