From 5b0a76e6832615f0275dd8deb1f1bf7c6e5eab3f Mon Sep 17 00:00:00 2001 From: 苏向夜 Date: Sun, 10 Dec 2023 21:16:08 +0800 Subject: :sparkles: 使用logger替代print MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/hydrorollcore/cli.py | 20 ++++++++++---------- 1 file changed, 10 insertions(+), 10 deletions(-) (limited to 'src') 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 规则包模板已创建！") -- cgit v1.2.3-70-g09d2