添加日志功能

Author: loopyme

src/DrcomExecutor/__main__.py

@@ -33,6 +33,9 @@ def parse_args() -> argparse.Namespace:
         parser.add_argument(
             "-c", "--config_path", help="查询配置文件路径", action="store_true",
         )
+        parser.add_argument(
+            "-l", "--log_path", help="查询日志文件路径", action="store_true",
+        )
         parser.add_argument(
             "-r", "--reset", help="重置配置项", action="store_true",
         )
@@ -59,7 +62,12 @@ def parse_args() -> argparse.Namespace:
     if args.config_path:
         print(f"配置文件位于{Config.path}\n")
         sys.exit()
+    if args.log_path:
+        print(f"日志文件位于{Config.log_path}\n")
+        sys.exit()
 
     config.dump()
 
     main()
+if __name__ == '__main__':
+    main()

src/DrcomExecutor/config/config.py

@@ -6,6 +6,7 @@
 class Config:
     path = Path(__file__).parent / "config.yaml"
     default_path = Path(__file__).parent / "config_default.yaml"
+    log_path = Path(__file__).parent / "DE.log"
 
     def __init__(self):
         if not self.path.is_file():
@@ -14,6 +15,7 @@ def __init__(self):
             else:
                 Config.reset()
         self.data = self.read_yaml(self.path)
+        self.log_path.open('a')
 
     @staticmethod
     def read_yaml(file_path):

src/DrcomExecutor/utils.py

@@ -72,16 +72,21 @@ def reset_config():
 
 def log(msg, error=False, warning=False):
     global ERROR_COUNT
+
+    def log_print(text):
+        config.log_path.write_text(text+"\n")
+        print(text)
+
     if error:
-        print("[{}]  {} - {}".format(datetime.now(), msg, ERROR_COUNT))
+        log_print("[{}]  {} - {}".format(datetime.now(), msg, ERROR_COUNT))
         if config["behavior"]["print_traceback"]:
-            traceback.print_exc()
+            log_print(traceback.format_exc())
         time.sleep(min(2 ** ERROR_COUNT, config["behavior"]["exp_backoff_limit"]))
         ERROR_COUNT += 1
     elif warning:
-        print("[{}]  {}".format(datetime.now(), msg))
+        log_print("[{}]  {}".format(datetime.now(), msg))
     else:
-        print("[{}]  {}".format(datetime.now(), msg))
+        log_print("[{}]  {}".format(datetime.now(), msg))
         reset_error_count()