fix:添加UI界面,完善功能
This commit is contained in:
@@ -1,37 +1,71 @@
|
||||
# 尝试导入 toml 解析库 (兼容 Python 3.11+ 和旧版本)
|
||||
import os
|
||||
import sys
|
||||
|
||||
# 尝试导入 toml 解析库
|
||||
try:
|
||||
import tomllib as toml # Python 3.11+ 内置
|
||||
import tomllib as toml # Python 3.11+
|
||||
except ImportError:
|
||||
try:
|
||||
import tomli as toml # 需要 pip install tomli
|
||||
import tomli as toml # pip install tomli
|
||||
except ImportError:
|
||||
print("错误: 缺少 TOML 解析库。请运行: pip install tomli")
|
||||
sys.exit(1)
|
||||
|
||||
def get_main_path():
|
||||
"""
|
||||
获取程序运行的根目录
|
||||
兼容:
|
||||
1. PyInstaller 打包后的 .exe 环境
|
||||
2. 开发环境 (假设此脚本在子文件夹中,如 utils/)
|
||||
"""
|
||||
if getattr(sys, 'frozen', False):
|
||||
# --- 情况 A: 打包后的 exe ---
|
||||
# exe 就在根目录下,直接取 exe 所在目录
|
||||
return os.path.dirname(sys.executable)
|
||||
else:
|
||||
# --- 情况 B: 开发环境 (.py) ---
|
||||
# 1. 获取当前脚本的绝对路径 (例如: .../MyProject/utils/config_loader.py)
|
||||
current_file_path = os.path.abspath(__file__)
|
||||
|
||||
# 2. 获取当前脚本所在的文件夹 (例如: .../MyProject/utils)
|
||||
current_dir = os.path.dirname(current_file_path)
|
||||
|
||||
# 3. 【关键修改】再往上一层,获取项目根目录 (例如: .../MyProject)
|
||||
# 如果你的脚本藏得更深,就再套一层 os.path.dirname
|
||||
project_root = os.path.dirname(current_dir)
|
||||
|
||||
return project_root
|
||||
|
||||
# ==========================================
|
||||
# 1. 配置加载 (Config Loader)
|
||||
# ==========================================
|
||||
def load_config(config_path="config.toml"):
|
||||
"""读取 TOML 配置文件并转换为扁平字典"""
|
||||
def load_config(config_filename="config.toml"):
|
||||
"""读取 TOML 配置文件"""
|
||||
|
||||
# 1. 先获取正确的根目录
|
||||
main_dir = get_main_path()
|
||||
|
||||
# 2. 拼接配置文件的绝对路径 (防止在不同目录下运行脚本时找不到配置文件)
|
||||
config_path = os.path.join(main_dir, config_filename)
|
||||
|
||||
if not os.path.exists(config_path):
|
||||
print(f"错误: 找不到配置文件 {config_path}")
|
||||
print(f"错误: 在路径 {main_dir} 下找不到配置文件 {config_filename}")
|
||||
print(f"尝试寻找的完整路径是: {config_path}")
|
||||
sys.exit(1)
|
||||
|
||||
try:
|
||||
with open(config_path, "rb") as f:
|
||||
data = toml.load(f)
|
||||
|
||||
# 将 TOML 的层级结构映射回原本的扁平结构,保持代码其余部分不用大改
|
||||
|
||||
# 将 TOML 的层级结构映射回扁平结构
|
||||
# 关键点:所有的 os.path.join 都必须基于 main_dir (项目根目录)
|
||||
config = {
|
||||
"source_file": data["paths"]["source_file"],
|
||||
"output_folder": data["paths"]["output_folder"],
|
||||
"excel_file": data["paths"]["excel_file"],
|
||||
"image_folder": data["paths"]["image_folder"],
|
||||
"fonts_dir": data["paths"]["fonts_dir"],
|
||||
"root_path": main_dir, # 方便调试,把根目录也存进去
|
||||
"source_file": os.path.join(main_dir, data["paths"]["source_file"]),
|
||||
"output_folder": os.path.join(main_dir, data["paths"]["output_folder"]),
|
||||
"excel_file": os.path.join(main_dir, data["paths"]["excel_file"]),
|
||||
"image_folder": os.path.join(main_dir, data["paths"]["image_folder"]),
|
||||
"fonts_dir": os.path.join(main_dir, data["paths"]["fonts_dir"]),
|
||||
"class_name": data["class_info"]["class_name"],
|
||||
"teachers": data["class_info"]["teachers"],
|
||||
"default_comment": data["defaults"].get("default_comment", "暂无评语"),
|
||||
@@ -41,4 +75,4 @@ def load_config(config_path="config.toml"):
|
||||
return config
|
||||
except Exception as e:
|
||||
print(f"读取配置文件出错: {e}")
|
||||
sys.exit(1)
|
||||
sys.exit(1)
|
||||
Reference in New Issue
Block a user