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

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 

12 

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

14 

15DEBUG_NONE = 0x0000 

16DEBUG_ENCODER = 0x0001 

17DEBUG_DECODER = 0x0002 

18DEBUG_ALL = 0xffff 

19 

20FLAG_MAP = { 

21 'none': DEBUG_NONE, 

22 'encoder': DEBUG_ENCODER, 

23 'decoder': DEBUG_DECODER, 

24 'all': DEBUG_ALL 

25} 

26 

27LOGGEE_MAP = {} 

28 

29 

30class Printer(object): 

31 # noinspection PyShadowingNames 

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

33 if logger is None: 

34 logger = logging.getLogger('pyasn1') 

35 

36 logger.setLevel(logging.DEBUG) 

37 

38 if handler is None: 

39 handler = logging.StreamHandler() 

40 

41 if formatter is None: 

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

43 

44 handler.setFormatter(formatter) 

45 handler.setLevel(logging.DEBUG) 

46 logger.addHandler(handler) 

47 

48 self.__logger = logger 

49 

50 def __call__(self, msg): 

51 self.__logger.debug(msg) 

52 

53 def __str__(self): 

54 return '<python logging>' 

55 

56 

57class Debug(object): 

58 defaultPrinter = Printer() 

59 

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

61 self._flags = DEBUG_NONE 

62 

63 if 'loggerName' in options: 

64 # route our logs to parent logger 

65 self._printer = Printer( 

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

67 handler=logging.NullHandler() 

68 ) 

69 

70 elif 'printer' in options: 

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

72 

73 else: 

74 self._printer = self.defaultPrinter 

75 

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

77 

78 for flag in flags: 

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

80 if inverse: 

81 flag = flag[1:] 

82 try: 

83 if inverse: 

84 self._flags &= ~FLAG_MAP[flag] 

85 else: 

86 self._flags |= FLAG_MAP[flag] 

87 except KeyError: 

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

89 

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

91 

92 def __str__(self): 

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

94 

95 def __call__(self, msg): 

96 self._printer(msg) 

97 

98 def __and__(self, flag): 

99 return self._flags & flag 

100 

101 def __rand__(self, flag): 

102 return flag & self._flags 

103 

104_LOG = DEBUG_NONE 

105 

106 

107def setLogger(userLogger): 

108 global _LOG 

109 

110 if userLogger: 

111 _LOG = userLogger 

112 else: 

113 _LOG = DEBUG_NONE 

114 

115 # Update registered logging clients 

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

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

118 

119 

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

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

122 setLogger(_LOG) 

123 return _LOG 

124 

125 

126def hexdump(octets): 

127 return ' '.join( 

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

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

130 ) 

131 

132 

133class Scope(object): 

134 def __init__(self): 

135 self._list = [] 

136 

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

138 

139 def push(self, token): 

140 self._list.append(token) 

141 

142 def pop(self): 

143 return self._list.pop() 

144 

145 

146scope = Scope()