日志打印增加时间；支持设置最大历史记录数

2023-12-25 15:53:42 +08:00
parent 9d00127b54
commit 95720ac92d
7 changed files with 206 additions and 5 deletions
@@ -0,0 +1,43 @@
+import sys
+
+from datetime import datetime as dt
+
+old_stdout = sys.stdout
+old_stderr = sys.stderr
+
+
+class StdTimeFilter:
+    def __init__(self, is_stdout):
+        self.is_stdout = is_stdout
+        self.newline = True
+        self.encoding = "utf-8"
+
+    def write(self, message):
+        if message == '\n':
+            if self.is_stdout:
+                old_stdout.write(message)
+            else:
+                old_stderr.write(message)
+            self.newline = True
+        elif self.newline:
+            if self.is_stdout:
+                old_stdout.write('%s %s' % (str(dt.now()), message))
+            else:
+                old_stderr.write('%s %s' % (str(dt.now()), message))
+            self.newline = False
+        else:
+            if self.is_stdout:
+                old_stdout.write(message)
+            else:
+                old_stderr.write(message)
+
+    def flush(self):
+        if self.is_stdout:
+            old_stdout.flush()
+        else:
+            old_stderr.flush()
+
+
+def log_wrap():
+    sys.stdout = StdTimeFilter(True)
+    sys.stderr = StdTimeFilter(False)