refactor: 重构项目配置管理,统一使用.env配置
- 新增config.py统一读取.env配置,移除硬编码路径和参数 - 重构collect_jrxml.py支持命令行参数和环境变量配置源目录 - 新增.env.example示例配置文件,整理所有可配置项 - 重构down_embedding_model.py、import_to_chroma.py等所有脚本使用统一配置 - 新增Windows一键部署脚本setup.bat - 修正jrxml_banch_chunker.py的文件名拼写错误
This commit is contained in:
+15
-3
@@ -2,10 +2,12 @@
|
||||
"""
|
||||
JRXML 文件收集脚本
|
||||
从指定目录递归查找所有 .jrxml 文件并复制到项目的 jrxml_source 目录
|
||||
源目录和目标目录通过 .env / config.py 配置
|
||||
"""
|
||||
|
||||
import os
|
||||
import shutil
|
||||
from config import JRXML_SOURCE_DIR
|
||||
|
||||
def collect_jrxml_files(source_dir: str, target_dir: str) -> int:
|
||||
"""
|
||||
@@ -53,12 +55,22 @@ def collect_jrxml_files(source_dir: str, target_dir: str) -> int:
|
||||
return copied_count
|
||||
|
||||
if __name__ == "__main__":
|
||||
SOURCE_DIR = r"C:\Users\zy187\JaspersoftWorkspace\JasperReportsSamples"
|
||||
TARGET_DIR = os.path.join(os.path.dirname(__file__), "jrxml_source")
|
||||
import sys
|
||||
|
||||
if len(sys.argv) >= 2:
|
||||
SOURCE_DIR = sys.argv[1]
|
||||
else:
|
||||
SOURCE_DIR = os.environ.get(
|
||||
"JRXML_COLLECT_SOURCE",
|
||||
r"C:\Users\zy187\JaspersoftWorkspace\JasperReportsSamples"
|
||||
)
|
||||
|
||||
TARGET_DIR = str(JRXML_SOURCE_DIR)
|
||||
|
||||
if not os.path.exists(SOURCE_DIR):
|
||||
print(f"错误:源目录不存在 - {SOURCE_DIR}")
|
||||
print("请检查路径是否正确")
|
||||
print("请检查路径是否正确,或通过命令行参数指定:")
|
||||
print(f" python collect_jrxml.py <源目录路径>")
|
||||
exit(1)
|
||||
|
||||
collect_jrxml_files(SOURCE_DIR, TARGET_DIR)
|
||||
Reference in New Issue
Block a user