Evilom
124 lines
3.9 KiB
Python
124 lines
3.9 KiB
Python
"""共享路径和配置工具"""
|
|
|
|
import os
|
|
import sys
|
|
import json
|
|
|
|
def find_project_root():
|
|
"""向上查找项目根目录（包含 data/ 目录的最顶层）"""
|
|
# 从当前脚本位置开始
|
|
script_dir = os.path.dirname(os.path.abspath(__file__))
|
|
|
|
# 向上查找，找到包含 data/ 目录的目录
|
|
d = script_dir
|
|
for _ in range(5): # 最多向上5级
|
|
if os.path.isdir(os.path.join(d, "data")):
|
|
return d
|
|
parent = os.path.dirname(d)
|
|
if parent == d:
|
|
break
|
|
d = parent
|
|
|
|
# 如果找不到，假设项目根在 scripts/ 的上一级
|
|
return os.path.dirname(os.path.dirname(script_dir))
|
|
|
|
def find_dws():
|
|
"""查找 dws CLI"""
|
|
import shutil
|
|
|
|
# 1. 环境变量
|
|
env = os.environ.get("DWS_PATH")
|
|
if env and os.path.isfile(env):
|
|
return env
|
|
|
|
# 2. 项目 tools/ 目录
|
|
root = find_project_root()
|
|
local = os.path.join(root, "tools", "dws.exe")
|
|
if os.path.isfile(local):
|
|
return local
|
|
|
|
# 3. 悟空内置
|
|
candidates = [
|
|
os.path.expanduser(r"~\.real\.bin\dws\bin\dws.exe"),
|
|
r"C:\Program Files\Wukong\0.9.51-26052503\bin\dws.exe",
|
|
]
|
|
for p in candidates:
|
|
if os.path.isfile(p):
|
|
return p
|
|
|
|
# 4. PATH
|
|
found = shutil.which("dws")
|
|
if found:
|
|
return found
|
|
|
|
print("ERROR: 找不到 dws CLI，请设置 DWS_PATH 环境变量或安装悟空", file=sys.stderr)
|
|
sys.exit(1)
|
|
|
|
def find_lark_cli():
|
|
"""查找 lark-cli"""
|
|
import shutil
|
|
root = find_project_root()
|
|
ext = ".exe" if os.name == "nt" else ""
|
|
|
|
local = os.path.join(root, "tools", "lark-cli", "lark-cli" + ext)
|
|
if os.path.isfile(local):
|
|
return local
|
|
|
|
found = shutil.which("lark-cli")
|
|
if found:
|
|
return found
|
|
print("ERROR: 找不到 lark-cli，请运行 npm install -g @larksuite/cli", file=sys.stderr)
|
|
sys.exit(1)
|
|
|
|
def load_groups():
|
|
"""读取群配置，优先使用 data/group_config.json，失败时返回默认群。"""
|
|
default = {
|
|
"dc战略问题研究院": "cidoUneRB4Db8TAXaTrKxkQAw==",
|
|
"创新组": "cidMuM+itt5PeY7xNSWsv3M0g==",
|
|
"天龙计划X小游戏矩阵": "cid/Vb3hSjdv4sIIs8Fp7sC4g==",
|
|
}
|
|
path = os.path.join(find_project_root(), "data", "group_config.json")
|
|
if not os.path.isfile(path):
|
|
return default
|
|
try:
|
|
with open(path, "r", encoding="utf-8") as f:
|
|
raw = json.load(f).get("groups", {})
|
|
groups = {}
|
|
for name, value in raw.items():
|
|
if isinstance(value, dict):
|
|
cid = value.get("cid") or value.get("id")
|
|
else:
|
|
cid = value
|
|
if cid:
|
|
groups[name] = cid
|
|
return groups or default
|
|
except Exception as e:
|
|
print(f"WARN: 读取群配置失败，使用默认配置: {e}", file=sys.stderr)
|
|
return default
|
|
|
|
def normalize_messages(messages):
|
|
"""兼容 list 和按群 dict 两种历史消息文件格式。"""
|
|
if isinstance(messages, dict):
|
|
return messages
|
|
grouped = {}
|
|
if not isinstance(messages, list):
|
|
return grouped
|
|
for msg in messages:
|
|
if not isinstance(msg, dict):
|
|
continue
|
|
group = msg.get("group") or msg.get("groupName") or msg.get("openConversationId") or "未分组"
|
|
grouped.setdefault(group, []).append(msg)
|
|
return grouped
|
|
|
|
# 常用路径
|
|
PROJECT_ROOT = find_project_root()
|
|
DATA_DIR = os.path.join(PROJECT_ROOT, "data")
|
|
RAW_DIR = os.path.join(DATA_DIR, "raw-messages")
|
|
LINKS_DIR = os.path.join(DATA_DIR, "links")
|
|
OUTPUT_DIR = os.path.join(PROJECT_ROOT, "output")
|
|
DOCS_DIR = os.path.join(OUTPUT_DIR, "feishu-docs")
|
|
REPORTS_DIR = os.path.join(OUTPUT_DIR, "reports")
|
|
DOWNLOAD_DIR = os.path.join(OUTPUT_DIR, "downloaded-files")
|
|
OBSIDIAN_DIR = os.path.join(OUTPUT_DIR, "obsidian-vault")
|
|
KG_DIR = os.path.join(OUTPUT_DIR, "knowledge-graph")
|