75 lines
2.1 KiB
Python
75 lines
2.1 KiB
Python
from __future__ import annotations
|
||
|
||
import sys
|
||
import argparse
|
||
import ctypes
|
||
import os
|
||
|
||
# 添加当前目录到 Python 路径
|
||
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
|
||
|
||
# Windows DPI 感知设置 - 必须在导入 tkinter 之前设置
|
||
if sys.platform == 'win32':
|
||
try:
|
||
# 设置 DPI 感知模式（支持 Windows 8.1 及以上）
|
||
ctypes.windll.shcore.SetProcessDpiAwareness(1) # 系统 DPI 感知
|
||
except Exception:
|
||
try:
|
||
# 兼容旧版本 Windows
|
||
ctypes.windll.user32.SetProcessDPIAware()
|
||
except Exception:
|
||
pass
|
||
|
||
from config_loader import get_config
|
||
from step1 import Step1FeastCoding
|
||
from step2 import Step2Converter
|
||
|
||
|
||
def run_business_flow(project_name: str | None = None) -> int:
|
||
"""运行完整的业务流程（Step1 + Step2）"""
|
||
config = get_config()
|
||
|
||
chrome_path = config.chrome_path
|
||
step1_config = config.step1_config
|
||
|
||
username = step1_config.get("username", "wangpeng")
|
||
password = step1_config.get("password", "Feastcoding@123")
|
||
|
||
try:
|
||
# Step 1
|
||
step1 = Step1FeastCoding(chrome_path, username, password)
|
||
if project_name:
|
||
step1.project_name = project_name
|
||
step1.run()
|
||
|
||
# Step 2
|
||
step2 = Step2Converter()
|
||
step2.run()
|
||
|
||
print("[INFO] 完整流程执行成功！")
|
||
return 0
|
||
except Exception as e:
|
||
print(f"[ERROR] 流程执行失败: {e}")
|
||
return 1
|
||
|
||
|
||
def main() -> int:
|
||
# 解析命令行参数
|
||
parser = argparse.ArgumentParser(description="自动化工具")
|
||
parser.add_argument("--cli", action="store_true", help="使用命令行模式（默认启动 GUI）")
|
||
parser.add_argument("--project", type=str, help="设置项目名称")
|
||
args = parser.parse_args()
|
||
|
||
# 如果没有指定 --cli 参数，默认启动 GUI
|
||
if not args.cli:
|
||
from gui import main as gui_main
|
||
gui_main()
|
||
return 0
|
||
|
||
# 否则运行命令行模式
|
||
return run_business_flow(project_name=args.project)
|
||
|
||
|
||
if __name__ == "__main__":
|
||
raise SystemExit(main())
|