feat：飞书文档补全
This commit is contained in:
Evilom
@@ -0,0 +1,60 @@
|
||||
"""共享配置：dws 路径解析 + 群组定义"""
|
||||
|
||||
import os
|
||||
import shutil
|
||||
import platform
|
||||
|
||||
SCRIPT_DIR = os.path.dirname(os.path.abspath(__file__))
|
||||
PROJECT_ROOT = os.path.dirname(SCRIPT_DIR)
|
||||
TOOLS_DIR = os.path.join(PROJECT_ROOT, "tools")
|
||||
|
||||
# dws 路径查找优先级:
|
||||
# 1. 环境变量 DWS_PATH
|
||||
# 2. 项目 tools/ 目录（从 GitHub 下载的独立版本）
|
||||
# 3. 悟空内置路径（向后兼容）
|
||||
# 4. 系统 PATH
|
||||
|
||||
_WUKONG_PATHS = [
|
||||
os.path.expanduser(r"~\.real\.bin\dws\bin\dws.exe"),
|
||||
r"C:\Program Files\Wukong\0.9.51-26052503\bin\dws.exe",
|
||||
]
|
||||
|
||||
|
||||
def _find_dws():
|
||||
# 1) 显式指定
|
||||
env = os.environ.get("DWS_PATH")
|
||||
if env and os.path.isfile(env):
|
||||
return env
|
||||
|
||||
# 2) 项目 tools/ 目录
|
||||
ext = ".exe" if platform.system() == "Windows" else ""
|
||||
local = os.path.join(TOOLS_DIR, "dws" + ext)
|
||||
if os.path.isfile(local):
|
||||
return local
|
||||
|
||||
# 3) 悟空内置（向后兼容）
|
||||
for p in _WUKONG_PATHS:
|
||||
if os.path.isfile(p):
|
||||
return p
|
||||
|
||||
# 4) 系统 PATH
|
||||
found = shutil.which("dws")
|
||||
if found:
|
||||
return found
|
||||
|
||||
raise FileNotFoundError(
|
||||
"找不到 dws CLI。请执行以下任一操作：\n"
|
||||
" 1) 运行: python scripts/setup_dws.py (从 GitHub 自动下载)\n"
|
||||
" 2) 手动下载: https://github.com/DingTalk-Real-AI/dingtalk-workspace-cli/releases/latest\n"
|
||||
" 将 dws.exe 放到 tools/ 目录\n"
|
||||
" 3) 设置环境变量: DWS_PATH=<你的 dws 路径>"
|
||||
)
|
||||
|
||||
|
||||
DWS = _find_dws()
|
||||
|
||||
# 钉钉群组 ID
|
||||
GROUPS = {
|
||||
"创新组": "cidMuM+itt5PeY7xNSWsv3M0g==",
|
||||
"dc战略问题研究院": "cidoUneRB4Db8TAXaTrKxkQAw==",
|
||||
}
|
||||
@@ -7,11 +7,7 @@ import re
|
||||
import os
|
||||
from datetime import datetime, timedelta
|
||||
|
||||
DWS = r"C:\Users\admin\.real\.bin\dws\bin\dws.exe"
|
||||
GROUPS = {
|
||||
"创新组": "cidMuM+itt5PeY7xNSWsv3M0g==",
|
||||
"dc战略问题研究院": "cidoUneRB4Db8TAXaTrKxkQAw=="
|
||||
}
|
||||
from config import DWS, GROUPS
|
||||
|
||||
def run_dws(args):
|
||||
try:
|
||||
@@ -58,3 +54,4 @@ def main():
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
|
||||
|
||||
@@ -4,7 +4,7 @@ sys.stdout = io.TextIOWrapper(sys.stdout.buffer, encoding="utf-8")
|
||||
SCRIPT_DIR = os.path.dirname(os.path.abspath(__file__))
|
||||
PROJECT_ROOT = os.path.dirname(SCRIPT_DIR)
|
||||
|
||||
DWS = r"C:\Users\admin\.real\.bin\dws\bin\dws.exe"
|
||||
from config import DWS
|
||||
|
||||
def fetch(group_id, time_str, forward="true"):
|
||||
cmd = [DWS, "chat", "message", "list", "--group", group_id, "--time", time_str, "--forward", forward, "--format", "json", "--limit", "200"]
|
||||
@@ -73,3 +73,4 @@ with open(os.path.join(PROJECT_ROOT, "data", "raw-messages", "all_messages_combi
|
||||
json.dump(output, f, ensure_ascii=False, indent=2)
|
||||
print("Saved!")
|
||||
|
||||
|
||||
|
||||
@@ -0,0 +1,139 @@
|
||||
"""从 GitHub Releases 下载 dws CLI 到项目 tools/ 目录"""
|
||||
|
||||
import os
|
||||
import sys
|
||||
import json
|
||||
import platform
|
||||
import zipfile
|
||||
import io
|
||||
import shutil
|
||||
|
||||
try:
|
||||
from urllib.request import urlopen, Request
|
||||
except ImportError:
|
||||
print("需要 Python 3")
|
||||
sys.exit(1)
|
||||
|
||||
SCRIPT_DIR = os.path.dirname(os.path.abspath(__file__))
|
||||
PROJECT_ROOT = os.path.dirname(SCRIPT_DIR)
|
||||
TOOLS_DIR = os.path.join(PROJECT_ROOT, "tools")
|
||||
|
||||
REPO = "DingTalk-Real-AI/dingtalk-workspace-cli"
|
||||
API_URL = f"https://api.github.com/repos/{REPO}/releases/latest"
|
||||
|
||||
|
||||
def get_asset_name():
|
||||
"""根据当前平台返回对应的 release asset 文件名"""
|
||||
system = platform.system().lower() # windows / darwin / linux
|
||||
machine = platform.machine().lower() # amd64 / arm64 / x86_64
|
||||
|
||||
if machine in ("x86_64", "amd64"):
|
||||
arch = "amd64"
|
||||
elif machine in ("aarch64", "arm64"):
|
||||
arch = "arm64"
|
||||
else:
|
||||
print(f"不支持的架构: {machine}")
|
||||
sys.exit(1)
|
||||
|
||||
if system == "windows":
|
||||
return f"dws-windows-{arch}.zip"
|
||||
elif system == "darwin":
|
||||
return f"dws-darwin-{arch}.tar.gz"
|
||||
elif system == "linux":
|
||||
return f"dws-linux-{arch}.tar.gz"
|
||||
else:
|
||||
print(f"不支持的操作系统: {system}")
|
||||
sys.exit(1)
|
||||
|
||||
|
||||
def download_and_extract(asset_name, download_url):
|
||||
"""下载并解压 dws 到 tools/ 目录"""
|
||||
os.makedirs(TOOLS_DIR, exist_ok=True)
|
||||
|
||||
print(f"下载 {asset_name} ...")
|
||||
req = Request(download_url, headers={"Accept": "application/octet-stream", "User-Agent": "feishu-docs-summary/setup"})
|
||||
with urlopen(req) as resp:
|
||||
data = resp.read()
|
||||
|
||||
ext = ".exe" if platform.system() == "Windows" else ""
|
||||
target = os.path.join(TOOLS_DIR, f"dws{ext}")
|
||||
|
||||
if asset_name.endswith(".zip"):
|
||||
with zipfile.ZipFile(io.BytesIO(data)) as zf:
|
||||
# 找到 dws 可执行文件
|
||||
for name in zf.namelist():
|
||||
basename = os.path.basename(name)
|
||||
if basename == "dws" or basename == "dws.exe":
|
||||
with zf.open(name) as src, open(target, "wb") as dst:
|
||||
dst.write(src.read())
|
||||
break
|
||||
else:
|
||||
print("zip 中未找到 dws 可执行文件")
|
||||
sys.exit(1)
|
||||
elif asset_name.endswith(".tar.gz"):
|
||||
import tarfile
|
||||
with tarfile.open(fileobj=io.BytesIO(data), mode="r:gz") as tf:
|
||||
for member in tf.getmembers():
|
||||
if os.path.basename(member.name) == "dws":
|
||||
src = tf.extractfile(member)
|
||||
with open(target, "wb") as dst:
|
||||
dst.write(src.read())
|
||||
break
|
||||
else:
|
||||
print("tar.gz 中未找到 dws 可执行文件")
|
||||
sys.exit(1)
|
||||
|
||||
# Unix 系统加执行权限
|
||||
if platform.system() != "Windows":
|
||||
os.chmod(target, 0o755)
|
||||
|
||||
size_kb = os.path.getsize(target) / 1024
|
||||
print(f"已保存到: {target} ({size_kb:.0f} KB)")
|
||||
return target
|
||||
|
||||
|
||||
def main():
|
||||
# 检查是否已存在
|
||||
ext = ".exe" if platform.system() == "Windows" else ""
|
||||
existing = os.path.join(TOOLS_DIR, f"dws{ext}")
|
||||
if os.path.isfile(existing):
|
||||
print(f"dws 已存在: {existing}")
|
||||
print("如需更新，请先删除该文件后重新运行此脚本")
|
||||
return
|
||||
|
||||
# 获取最新 release 信息
|
||||
print(f"查询最新版本: {API_URL}")
|
||||
req = Request(API_URL, headers={"Accept": "application/vnd.github.v3+json", "User-Agent": "feishu-docs-summary/setup"})
|
||||
with urlopen(req) as resp:
|
||||
release = json.loads(resp.read())
|
||||
|
||||
tag = release["tag_name"]
|
||||
print(f"最新版本: {tag}")
|
||||
# 检查 rate limit
|
||||
remaining = resp.headers.get("X-RateLimit-Remaining", "?")
|
||||
print(f"GitHub API 剩余请求: {remaining}")
|
||||
|
||||
asset_name = get_asset_name()
|
||||
print(f"目标平台: {asset_name}")
|
||||
|
||||
# 找到下载链接
|
||||
download_url = None
|
||||
for asset in release.get("assets", []):
|
||||
if asset["name"] == asset_name:
|
||||
download_url = asset["browser_download_url"]
|
||||
break
|
||||
|
||||
if not download_url:
|
||||
print(f"未找到对应平台的 release: {asset_name}")
|
||||
sys.exit(1)
|
||||
|
||||
path = download_and_extract(asset_name, download_url)
|
||||
print("\n安装完成！验证:")
|
||||
print(f" {path} version")
|
||||
print(f" {path} auth")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
|
||||
|
||||
Reference in New Issue
Block a user