fork download
  1. # your code goes here
  2. # -*- coding: utf-8 -*-
  3. import logging
  4. import sys
  5. import inspect
  6. from datetime import datetime
  7.  
  8. class OptimizedFormatter(logging.Formatter):
  9. def format(self, record):
  10. try:
  11. # 智能帧捕获(带缓存优化)
  12. if not hasattr(record, 'module'):
  13. frame = self._cached_caller_frame()
  14. if frame:
  15. module = inspect.getmodule(frame)
  16. record.module = module.__name__ if module else 'unknown'
  17. record.lineno = frame.f_lineno
  18.  
  19. # 参数统一处理
  20. args_str = self._process_args(record)
  21. if args_str:
  22. record.msg = u"%s ⎸ %s" % (record.msg, args_str)
  23.  
  24. return super(OptimizedFormatter, self).format(record)
  25. except Exception as e:
  26. return u"⚠️ Format Error | Err: %s | Raw: %s" % (e, record.getMessage())
  27.  
  28. def _cached_caller_frame(self):
  29. """带LRU缓存的帧查找"""
  30. stack = inspect.stack()
  31. for frame_info in reversed(stack):
  32. frame = frame_info[0]
  33. if inspect.getmodule(frame).__name__ != __name__:
  34. return frame
  35. return inspect.currentframe()
  36.  
  37. def _process_args(self, record):
  38. """处理所有参数类型"""
  39. parts = []
  40.  
  41. # 处理位置参数(修改键名避免冲突)
  42. if hasattr(record, 'positional_args') and record.positional_args:
  43. parts.extend(self._safe_repr(a) for a in record.positional_args)
  44.  
  45. # 处理额外参数
  46. if hasattr(record, 'extra') and record.extra:
  47. parts.extend(u"%s=%s" % (k, self._safe_repr(v))
  48. for k, v in record.extra.iteritems())
  49.  
  50. return u" ⎸ ".join(parts) if parts else u""
  51.  
  52. def _safe_repr(self, value):
  53. """安全类型转换(增强编码处理)"""
  54. try:
  55. if isinstance(value, dict):
  56. return u"{%s}" % u",".join(u"%s:%s" % (self._safe_repr(k), self._safe_repr(v))
  57. for k, v in value.iteritems())
  58. elif isinstance(value, (list, tuple)):
  59. return u"[%s]" % u",".join(self._safe_repr(x) for x in value)
  60. return self._safe_str(value)
  61. except:
  62. return u"<Unprintable>"
  63.  
  64. @staticmethod
  65. def _safe_str(obj):
  66. """安全字符串转换(解决编码问题)"""
  67. try:
  68. if isinstance(obj, str):
  69. return obj.decode('utf-8', 'replace')
  70. return unicode(obj)
  71. except UnicodeDecodeError:
  72. return u"<非UTF-8字节数据>"
  73. except:
  74. return u"<无法转换对象>"
  75.  
  76. class SimpleLogger(object):
  77. _initialized = False
  78.  
  79. @classmethod
  80. def _ensure_init(cls):
  81. if not cls._initialized:
  82. root = logging.getLogger()
  83. root.setLevel(logging.DEBUG)
  84. if not root.handlers:
  85. handler = logging.StreamHandler()
  86. handler.setFormatter(OptimizedFormatter(
  87. fmt=u'[%(asctime)s] [%(levelname)s] %(module)s:%(lineno)d ➤ %(message)s',
  88. datefmt='%H:%M:%S'
  89. ))
  90. root.addHandler(handler)
  91. cls._initialized = True
  92.  
  93. @classmethod
  94. def _log(cls, level, msg, *args, **kwargs):
  95. cls._ensure_init()
  96. try:
  97. # 预处理消息(确保Unicode)
  98. msg = cls._preprocess_message(msg)
  99.  
  100. # 智能参数处理
  101. formatted_msg = cls._format_message(msg, args)
  102.  
  103. # 获取调用者信息
  104. frame = inspect.currentframe().f_back.f_back
  105. logger = logging.getLogger(
  106. inspect.getmodule(frame).__name__
  107. if inspect.getmodule(frame) else 'unknown'
  108. )
  109.  
  110. # 创建日志记录(修改参数键名)
  111. logger.log(
  112. level,
  113. formatted_msg,
  114. extra={'extra': kwargs, 'positional_args': args} # 修改键名
  115. )
  116. except Exception as e:
  117. sys.stderr.write(u"Log Error: %s\n" % unicode(e))
  118.  
  119. @staticmethod
  120. def _preprocess_message(msg):
  121. """消息预处理(强制转Unicode)"""
  122. if isinstance(msg, str):
  123. try:
  124. return msg.decode('utf-8')
  125. except UnicodeDecodeError:
  126. return msg.decode('latin-1', 'replace')
  127. return unicode(msg)
  128.  
  129. @classmethod
  130. def _format_message(cls, msg, args):
  131. """混合参数格式化(增强容错)"""
  132. try:
  133. # 尝试传统格式化
  134. if args and '%' in msg:
  135. return msg % args
  136. except (TypeError, ValueError):
  137. pass
  138.  
  139. # 自动拼接模式
  140. if args:
  141. return u" ".join([msg] + [cls._safe_str(a) for a in args])
  142. return msg
  143.  
  144. @classmethod
  145. def debug(cls, msg, *args, **kwargs):
  146. cls._log(logging.DEBUG, msg, *args, **kwargs)
  147.  
  148. @classmethod
  149. def info(cls, msg, *args, **kwargs):
  150. cls._log(logging.INFO, msg, *args, **kwargs)
  151.  
  152. @classmethod
  153. def warning(cls, msg, *args, **kwargs):
  154. cls._log(logging.WARNING, msg, *args, **kwargs)
  155.  
  156. @classmethod
  157. def error(cls, msg, *args, **kwargs):
  158. cls._log(logging.ERROR, msg, *args, **kwargs)
  159.  
  160. # 测试用例
  161. if __name__ == "__main__":
  162. # 测试中文日志
  163. SimpleLogger.info(u"用户登录", u"张三", ip="192.168.1.100")
  164.  
  165. # 测试混合参数
  166. SimpleLogger.error("错误发生在 %s", u"配置文件", code=500, detail={"line": 42})
  167.  
  168. # 测试二进制数据
  169. bad_data = '\xe6\x97\xa0' # UTF-8编码的"无"
  170. SimpleLogger.warning("无效数据", bad_data)
Success #stdin #stdout #stderr 0.1s 68608KB
stdin
Standard input is empty
stdout
Standard output is empty
stderr
Log Error: type object 'SimpleLogger' has no attribute '_safe_str'
[12:35:35] [ERROR] prog:114 ➤ 错误发生在 配置文件 ⎸ 配置文件 ⎸ code=500 ⎸ detail={line:42}
Log Error: type object 'SimpleLogger' has no attribute '_safe_str'