:sparkles: 使用logger替代print

1 files changed, 10 insertions, 10 deletions

diff --git a/src/hydrorollcore/cli.py b/src/hydrorollcore/cli.py
index 8b03b60b..6daff17a 100644
--- a/src/hydrorollcore/cli.py
+++ b/src/hydrorollcore/cli.py
@@ -1,7 +1,6 @@
 from pathlib import Path
 from .consts import templates
-
-# from tkinter import messagebox
+from .log import logger
 
 import argparse
 import os
@@ -9,18 +8,19 @@ import sys
 
 
 class Cli:
-    def parse_args(self):
+    def parse_args(self, args: list = None):
         parser = argparse.ArgumentParser(description="HydroRoll 命令行工具")
 
         parser.add_argument("--new", action="store_true", help="创建一个 HydroRoll 规则包模板")
         parser.add_argument("--run", action="store_true", help="运行 HydroRoll 规则包")
-        # parser.add_argument("--gui", action="store_true", help="显示弹窗")
+        parser.add_argument("--gui", action="store_true", help="显示弹窗")
         parser.add_argument("--path", help="指定路径")
 
-        args = parser.parse_args()
+        args = parser.parse_args(args if args else sys.argv[1:])
 
-        # if args.gui:
-        #     messagebox.showinfo("提示", "这是一个弹窗！")
+        if args.gui:
+            logger.critical("选项[--gui]尚未被支持！")
+            sys.exit(1)
 
         if not args.path:
             path = Path(os.getcwd()).resolve()
@@ -28,12 +28,12 @@ class Cli:
             path = Path(args.path).resolve()
 
         if args.new and args.run:
-            print("无法确定的指令要求: 你同时指定了new与run指令。")
+            logger.error("无法确定的指令要求: 你同时指定了new与run指令。")
             sys.exit(1)
 
         if args.new:
             if path.exists():
-                print("指定的文件夹已经存在！")
+                logger.error("指定的文件夹已经存在！")
                 sys.exit(1)
 
             path.mkdir(parents=True, exist_ok=True)
@@ -41,4 +41,4 @@ class Cli:
             (path / "event.py").write_text(templates.EVENT)
             (path / "dice.py").write_text(templates.DICE)
 
-            print("HydroRoll 规则包模板已创建！")
+            logger.success("HydroRoll 规则包模板已创建！")