531 lines
19 KiB
Python
531 lines
19 KiB
Python
#!/usr/bin/env python3
|
||
"""
|
||
ai-proj 与飞书项目同步脚本
|
||
|
||
功能：
|
||
- 初始化：创建飞书根目录、多维表格
|
||
- 全量同步：同步所有项目到飞书
|
||
- 增量同步：只同步新增项目
|
||
- 更新同步：更新现有项目的任务统计等信息
|
||
|
||
使用方法：
|
||
python aiproj_sync.py init # 首次初始化
|
||
python aiproj_sync.py sync # 增量同步（推荐日常使用）
|
||
python aiproj_sync.py sync-all # 全量同步
|
||
python aiproj_sync.py update # 更新统计信息
|
||
"""
|
||
|
||
import os
|
||
import sys
|
||
import json
|
||
import requests
|
||
from datetime import datetime
|
||
from typing import Optional, Dict, List, Any
|
||
|
||
# ============================================
|
||
# 配置
|
||
# ============================================
|
||
|
||
FEISHU_APP_ID = os.getenv("FEISHU_APP_ID", "cli_a9f29dca82b9dbef")
|
||
FEISHU_APP_SECRET = os.getenv("FEISHU_APP_SECRET", "sDfhjG7QT1S4gfHiMVYSygmPQPN1R2Ho")
|
||
|
||
# 默认协作者 open_id（邱栋梁@智云采购）
|
||
# 应用创建的文件夹所有者是应用本身，需要显式添加用户为协作者
|
||
DEFAULT_COLLABORATOR_OPENID = "ou_43784ff7c819ac000095fb52a4c3d1c7"
|
||
|
||
AIPROJ_API_BASE = "https://ai.pipexerp.com/api/v1"
|
||
AIPROJ_TOKEN = os.getenv("AIPROJ_TOKEN", "aiproj_pk_b455c91607414c22a0f3d8f09785969f1aa2144f33f1336fbb12450ecebfdb64")
|
||
|
||
CONFIG_PATH = os.path.expanduser("~/.config/aiproj-feishu-sync.json")
|
||
|
||
# ============================================
|
||
# 飞书 API
|
||
# ============================================
|
||
|
||
class FeishuAPI:
|
||
def __init__(self):
|
||
self.token = None
|
||
|
||
def get_token(self) -> str:
|
||
"""获取 tenant_access_token"""
|
||
if self.token:
|
||
return self.token
|
||
|
||
url = "https://open.feishu.cn/open-apis/auth/v3/tenant_access_token/internal"
|
||
response = requests.post(url, json={
|
||
"app_id": FEISHU_APP_ID,
|
||
"app_secret": FEISHU_APP_SECRET
|
||
})
|
||
data = response.json()
|
||
|
||
if data.get("code") == 0:
|
||
self.token = data["tenant_access_token"]
|
||
return self.token
|
||
else:
|
||
raise Exception(f"获取飞书 token 失败: {data}")
|
||
|
||
def _headers(self, content_type: bool = False) -> Dict:
|
||
headers = {"Authorization": f"Bearer {self.get_token()}"}
|
||
if content_type:
|
||
headers["Content-Type"] = "application/json"
|
||
return headers
|
||
|
||
def get_root_folder(self) -> str:
|
||
"""获取云空间根文件夹"""
|
||
url = "https://open.feishu.cn/open-apis/drive/explorer/v2/root_folder/meta"
|
||
response = requests.get(url, headers=self._headers())
|
||
data = response.json()
|
||
|
||
if data.get("code") == 0:
|
||
return data["data"]["token"]
|
||
raise Exception(f"获取根文件夹失败: {data}")
|
||
|
||
def create_folder(self, name: str, parent_token: str, add_collaborator: bool = True) -> str:
|
||
"""创建文件夹"""
|
||
url = "https://open.feishu.cn/open-apis/drive/v1/files/create_folder"
|
||
response = requests.post(url, headers=self._headers(True), json={
|
||
"name": name,
|
||
"folder_token": parent_token
|
||
})
|
||
data = response.json()
|
||
|
||
if data.get("code") == 0:
|
||
folder_token = data["data"]["token"]
|
||
# 自动添加协作者（应用创建的文件夹默认只有应用能访问）
|
||
if add_collaborator and DEFAULT_COLLABORATOR_OPENID:
|
||
self.add_collaborator(folder_token, "folder", DEFAULT_COLLABORATOR_OPENID)
|
||
return folder_token
|
||
raise Exception(f"创建文件夹失败: {data}")
|
||
|
||
def add_collaborator(self, file_token: str, file_type: str, user_open_id: str, perm: str = "full_access") -> bool:
|
||
"""添加协作者
|
||
|
||
应用通过 API 创建的文件/文件夹，所有者是应用本身。
|
||
组织内用户默认无法访问，需要显式添加为协作者。
|
||
|
||
Args:
|
||
file_token: 文件/文件夹 token
|
||
file_type: 类型 (folder, doc, sheet, bitable 等)
|
||
user_open_id: 用户 open_id
|
||
perm: 权限级别 (full_access, edit, view)
|
||
"""
|
||
url = f"https://open.feishu.cn/open-apis/drive/v1/permissions/{file_token}/members"
|
||
params = {"type": file_type, "need_notification": "false"}
|
||
payload = {
|
||
"member_type": "openid",
|
||
"member_id": user_open_id,
|
||
"perm": perm
|
||
}
|
||
response = requests.post(url, headers=self._headers(True), params=params, json=payload)
|
||
return response.json().get("code") == 0
|
||
|
||
def create_bitable(self, name: str, folder_token: str) -> Dict:
|
||
"""创建多维表格"""
|
||
url = "https://open.feishu.cn/open-apis/bitable/v1/apps"
|
||
response = requests.post(url, headers=self._headers(True), json={
|
||
"name": name,
|
||
"folder_token": folder_token
|
||
})
|
||
data = response.json()
|
||
|
||
if data.get("code") == 0:
|
||
return data["data"]["app"]
|
||
raise Exception(f"创建多维表格失败: {data}")
|
||
|
||
def get_bitable_tables(self, app_token: str) -> List[Dict]:
|
||
"""获取数据表列表"""
|
||
url = f"https://open.feishu.cn/open-apis/bitable/v1/apps/{app_token}/tables"
|
||
response = requests.get(url, headers=self._headers())
|
||
data = response.json()
|
||
|
||
if data.get("code") == 0:
|
||
return data["data"]["items"]
|
||
raise Exception(f"获取数据表失败: {data}")
|
||
|
||
def create_field(self, app_token: str, table_id: str, field: Dict) -> Optional[Dict]:
|
||
"""创建字段"""
|
||
url = f"https://open.feishu.cn/open-apis/bitable/v1/apps/{app_token}/tables/{table_id}/fields"
|
||
response = requests.post(url, headers=self._headers(True), json=field)
|
||
data = response.json()
|
||
|
||
if data.get("code") == 0:
|
||
return data["data"]["field"]
|
||
print(f" 字段创建失败: {field.get('field_name')} - {data.get('msg')}")
|
||
return None
|
||
|
||
def update_field(self, app_token: str, table_id: str, field_id: str, updates: Dict) -> bool:
|
||
"""更新字段"""
|
||
url = f"https://open.feishu.cn/open-apis/bitable/v1/apps/{app_token}/tables/{table_id}/fields/{field_id}"
|
||
response = requests.put(url, headers=self._headers(True), json=updates)
|
||
return response.json().get("code") == 0
|
||
|
||
def get_fields(self, app_token: str, table_id: str) -> List[Dict]:
|
||
"""获取字段列表"""
|
||
url = f"https://open.feishu.cn/open-apis/bitable/v1/apps/{app_token}/tables/{table_id}/fields"
|
||
response = requests.get(url, headers=self._headers())
|
||
data = response.json()
|
||
|
||
if data.get("code") == 0:
|
||
return data["data"]["items"]
|
||
return []
|
||
|
||
def add_record(self, app_token: str, table_id: str, fields: Dict) -> Optional[Dict]:
|
||
"""添加记录"""
|
||
url = f"https://open.feishu.cn/open-apis/bitable/v1/apps/{app_token}/tables/{table_id}/records"
|
||
response = requests.post(url, headers=self._headers(True), json={"fields": fields})
|
||
data = response.json()
|
||
|
||
if data.get("code") == 0:
|
||
return data["data"]["record"]
|
||
print(f" 记录添加失败: {data.get('msg')}")
|
||
return None
|
||
|
||
def get_records(self, app_token: str, table_id: str, filter_str: str = None) -> List[Dict]:
|
||
"""获取记录列表"""
|
||
url = f"https://open.feishu.cn/open-apis/bitable/v1/apps/{app_token}/tables/{table_id}/records"
|
||
params = {"page_size": 500}
|
||
if filter_str:
|
||
params["filter"] = filter_str
|
||
|
||
response = requests.get(url, headers=self._headers(), params=params)
|
||
data = response.json()
|
||
|
||
if data.get("code") == 0:
|
||
return data["data"].get("items", [])
|
||
return []
|
||
|
||
def update_record(self, app_token: str, table_id: str, record_id: str, fields: Dict) -> bool:
|
||
"""更新记录"""
|
||
url = f"https://open.feishu.cn/open-apis/bitable/v1/apps/{app_token}/tables/{table_id}/records/{record_id}"
|
||
response = requests.put(url, headers=self._headers(True), json={"fields": fields})
|
||
return response.json().get("code") == 0
|
||
|
||
# ============================================
|
||
# ai-proj API
|
||
# ============================================
|
||
|
||
class AIProjAPI:
|
||
@staticmethod
|
||
def get_projects() -> List[Dict]:
|
||
"""获取项目列表"""
|
||
url = f"{AIPROJ_API_BASE}/projects?page_size=100"
|
||
headers = {"Authorization": f"Bearer {AIPROJ_TOKEN}"}
|
||
response = requests.get(url, headers=headers)
|
||
data = response.json()
|
||
|
||
if data.get("success"):
|
||
return data["data"]["data"]
|
||
raise Exception(f"获取项目列表失败: {data}")
|
||
|
||
# ============================================
|
||
# 配置管理
|
||
# ============================================
|
||
|
||
def load_config() -> Optional[Dict]:
|
||
"""加载配置"""
|
||
if os.path.exists(CONFIG_PATH):
|
||
with open(CONFIG_PATH) as f:
|
||
return json.load(f)
|
||
return None
|
||
|
||
def save_config(config: Dict):
|
||
"""保存配置"""
|
||
os.makedirs(os.path.dirname(CONFIG_PATH), exist_ok=True)
|
||
with open(CONFIG_PATH, "w") as f:
|
||
json.dump(config, f, indent=2, ensure_ascii=False)
|
||
|
||
# ============================================
|
||
# 同步逻辑
|
||
# ============================================
|
||
|
||
def init():
|
||
"""初始化：创建飞书目录结构"""
|
||
print("=" * 50)
|
||
print("飞书 ai-proj 项目同步 - 初始化")
|
||
print("=" * 50)
|
||
|
||
if load_config():
|
||
print("\n已存在配置文件，是否重新初始化？(y/N)")
|
||
if input().lower() != 'y':
|
||
print("取消初始化")
|
||
return
|
||
|
||
feishu = FeishuAPI()
|
||
|
||
# 1. 获取根目录
|
||
print("\n[1/5] 获取云空间根目录...")
|
||
space_root = feishu.get_root_folder()
|
||
print(f" 根目录: {space_root}")
|
||
|
||
# 2. 创建 ai-proj 文件夹
|
||
print("\n[2/5] 创建 ai-proj 根文件夹...")
|
||
root_folder = feishu.create_folder("ai-proj", space_root)
|
||
print(f" 创建成功: {root_folder}")
|
||
|
||
# 3. 创建多维表格
|
||
print("\n[3/5] 创建项目目录多维表格...")
|
||
bitable = feishu.create_bitable("项目目录", root_folder)
|
||
app_token = bitable["app_token"]
|
||
print(f" 创建成功: {app_token}")
|
||
|
||
tables = feishu.get_bitable_tables(app_token)
|
||
table_id = tables[0]["table_id"]
|
||
|
||
# 4. 创建字段
|
||
print("\n[4/5] 创建数据表字段...")
|
||
|
||
# 先修改第一列名称
|
||
fields = feishu.get_fields(app_token, table_id)
|
||
if fields:
|
||
feishu.update_field(app_token, table_id, fields[0]["field_id"], {
|
||
"field_name": "项目名称",
|
||
"type": fields[0]["type"]
|
||
})
|
||
print(" 修改字段: 项目名称")
|
||
|
||
# 创建其他字段
|
||
field_definitions = [
|
||
{"field_name": "项目ID", "type": 2},
|
||
{"field_name": "项目编号", "type": 1},
|
||
{"field_name": "飞书文件夹", "type": 15},
|
||
{"field_name": "状态", "type": 3, "property": {"options": [
|
||
{"name": "active", "color": 0},
|
||
{"name": "on_hold", "color": 1},
|
||
{"name": "planning", "color": 2},
|
||
{"name": "completed", "color": 3},
|
||
{"name": "archived", "color": 4}
|
||
]}},
|
||
{"field_name": "优先级", "type": 3, "property": {"options": [
|
||
{"name": "high", "color": 0},
|
||
{"name": "medium", "color": 1},
|
||
{"name": "low", "color": 2}
|
||
]}},
|
||
{"field_name": "所属企业", "type": 1},
|
||
{"field_name": "任务总数", "type": 2},
|
||
{"field_name": "描述", "type": 1},
|
||
{"field_name": "创建时间", "type": 5},
|
||
{"field_name": "最后同步", "type": 5},
|
||
]
|
||
|
||
for field in field_definitions:
|
||
if feishu.create_field(app_token, table_id, field):
|
||
print(f" 创建字段: {field['field_name']}")
|
||
|
||
# 5. 保存配置
|
||
print("\n[5/5] 保存配置...")
|
||
config = {
|
||
"root_folder_token": root_folder,
|
||
"bitable_app_token": app_token,
|
||
"bitable_table_id": table_id,
|
||
"created_at": datetime.now().isoformat(),
|
||
"synced_projects": {}
|
||
}
|
||
save_config(config)
|
||
|
||
print("\n" + "=" * 50)
|
||
print("初始化完成!")
|
||
print("=" * 50)
|
||
print(f"\n根文件夹: https://feishu.cn/drive/folder/{root_folder}")
|
||
print(f"多维表格: https://feishu.cn/base/{app_token}")
|
||
print(f"\n配置文件: {CONFIG_PATH}")
|
||
print("\n运行 'python aiproj_sync.py sync' 同步项目")
|
||
|
||
def sync(full: bool = False):
|
||
"""同步项目"""
|
||
config = load_config()
|
||
if not config:
|
||
print("未找到配置，请先运行 init")
|
||
return
|
||
|
||
print("=" * 50)
|
||
print(f"飞书 ai-proj 项目同步 - {'全量' if full else '增量'}同步")
|
||
print("=" * 50)
|
||
|
||
feishu = FeishuAPI()
|
||
app_token = config["bitable_app_token"]
|
||
table_id = config["bitable_table_id"]
|
||
root_folder = config["root_folder_token"]
|
||
synced = config.get("synced_projects", {})
|
||
|
||
# 获取 ai-proj 项目
|
||
print("\n获取 ai-proj 项目列表...")
|
||
projects = AIProjAPI.get_projects()
|
||
print(f" 共 {len(projects)} 个项目")
|
||
|
||
# 筛选需要同步的项目
|
||
if full:
|
||
to_sync = projects
|
||
else:
|
||
to_sync = [p for p in projects if str(p["id"]) not in synced]
|
||
|
||
if not to_sync:
|
||
print("\n没有新项目需要同步")
|
||
return
|
||
|
||
print(f"\n需要同步 {len(to_sync)} 个项目...")
|
||
|
||
success = 0
|
||
for project in to_sync:
|
||
project_id = str(project["id"])
|
||
project_name = project["name"]
|
||
folder_name = f"{project_name}_{project_id}"
|
||
|
||
print(f"\n 处理: {project_name} (ID: {project_id})")
|
||
|
||
# 创建文件夹（如果不存在）
|
||
folder_token = synced.get(project_id, {}).get("folder_token")
|
||
if not folder_token:
|
||
try:
|
||
folder_token = feishu.create_folder(folder_name, root_folder)
|
||
print(f" 创建文件夹: {folder_token}")
|
||
except Exception as e:
|
||
print(f" 文件夹创建失败: {e}")
|
||
folder_token = ""
|
||
|
||
folder_url = f"https://feishu.cn/drive/folder/{folder_token}" if folder_token else ""
|
||
|
||
# 添加/更新记录
|
||
record_fields = {
|
||
"项目名称": project_name,
|
||
"项目ID": int(project_id),
|
||
"项目编号": project.get("project_number", ""),
|
||
"飞书文件夹": {"link": folder_url, "text": folder_name} if folder_url else None,
|
||
"状态": project.get("status", "active"),
|
||
"优先级": project.get("priority", "medium"),
|
||
"所属企业": project.get("company_name", ""),
|
||
"任务总数": project.get("task_count", 0),
|
||
"描述": (project.get("description", "") or "")[:500],
|
||
"创建时间": int(datetime.fromisoformat(
|
||
project["created_at"].replace("Z", "+00:00")
|
||
).timestamp() * 1000),
|
||
"最后同步": int(datetime.now().timestamp() * 1000)
|
||
}
|
||
|
||
record_id = synced.get(project_id, {}).get("record_id")
|
||
if record_id and not full:
|
||
# 更新现有记录
|
||
if feishu.update_record(app_token, table_id, record_id, record_fields):
|
||
print(f" 更新记录成功")
|
||
success += 1
|
||
else:
|
||
# 添加新记录
|
||
record = feishu.add_record(app_token, table_id, record_fields)
|
||
if record:
|
||
record_id = record["record_id"]
|
||
print(f" 添加记录成功")
|
||
success += 1
|
||
|
||
# 更新同步记录
|
||
synced[project_id] = {
|
||
"folder_token": folder_token,
|
||
"folder_url": folder_url,
|
||
"record_id": record_id,
|
||
"synced_at": datetime.now().isoformat()
|
||
}
|
||
|
||
# 保存配置
|
||
config["synced_projects"] = synced
|
||
config["last_sync"] = datetime.now().isoformat()
|
||
save_config(config)
|
||
|
||
print("\n" + "=" * 50)
|
||
print(f"同步完成: {success}/{len(to_sync)}")
|
||
print("=" * 50)
|
||
print(f"\n多维表格: https://feishu.cn/base/{app_token}")
|
||
|
||
def update_stats():
|
||
"""更新任务统计信息"""
|
||
config = load_config()
|
||
if not config:
|
||
print("未找到配置，请先运行 init")
|
||
return
|
||
|
||
print("=" * 50)
|
||
print("飞书 ai-proj 项目同步 - 更新统计")
|
||
print("=" * 50)
|
||
|
||
feishu = FeishuAPI()
|
||
app_token = config["bitable_app_token"]
|
||
table_id = config["bitable_table_id"]
|
||
synced = config.get("synced_projects", {})
|
||
|
||
# 获取最新项目数据
|
||
print("\n获取 ai-proj 项目列表...")
|
||
projects = AIProjAPI.get_projects()
|
||
project_map = {str(p["id"]): p for p in projects}
|
||
|
||
updated = 0
|
||
for project_id, sync_info in synced.items():
|
||
record_id = sync_info.get("record_id")
|
||
if not record_id:
|
||
continue
|
||
|
||
project = project_map.get(project_id)
|
||
if not project:
|
||
continue
|
||
|
||
# 只更新统计字段
|
||
update_fields = {
|
||
"任务总数": project.get("task_count", 0),
|
||
"状态": project.get("status", "active"),
|
||
"最后同步": int(datetime.now().timestamp() * 1000)
|
||
}
|
||
|
||
if feishu.update_record(app_token, table_id, record_id, update_fields):
|
||
print(f" 更新: {project['name']} - 任务数: {project.get('task_count', 0)}")
|
||
updated += 1
|
||
|
||
config["last_sync"] = datetime.now().isoformat()
|
||
save_config(config)
|
||
|
||
print(f"\n更新完成: {updated} 个项目")
|
||
|
||
def show_status():
|
||
"""显示同步状态"""
|
||
config = load_config()
|
||
if not config:
|
||
print("未初始化，请先运行: python aiproj_sync.py init")
|
||
return
|
||
|
||
print("=" * 50)
|
||
print("ai-proj 飞书同步状态")
|
||
print("=" * 50)
|
||
print(f"\n根文件夹: https://feishu.cn/drive/folder/{config['root_folder_token']}")
|
||
print(f"多维表格: https://feishu.cn/base/{config['bitable_app_token']}")
|
||
print(f"已同步项目: {len(config.get('synced_projects', {}))}")
|
||
print(f"上次同步: {config.get('last_sync', '从未')}")
|
||
print(f"配置文件: {CONFIG_PATH}")
|
||
|
||
# ============================================
|
||
# 主入口
|
||
# ============================================
|
||
|
||
def main():
|
||
if len(sys.argv) < 2:
|
||
print("用法: python aiproj_sync.py <command>")
|
||
print("\n命令:")
|
||
print(" init 首次初始化（创建目录和多维表格）")
|
||
print(" sync 增量同步（只同步新项目）")
|
||
print(" sync-all 全量同步（同步所有项目）")
|
||
print(" update 更新统计信息")
|
||
print(" status 查看同步状态")
|
||
return
|
||
|
||
command = sys.argv[1]
|
||
|
||
if command == "init":
|
||
init()
|
||
elif command == "sync":
|
||
sync(full=False)
|
||
elif command == "sync-all":
|
||
sync(full=True)
|
||
elif command == "update":
|
||
update_stats()
|
||
elif command == "status":
|
||
show_status()
|
||
else:
|
||
print(f"未知命令: {command}")
|
||
|
||
if __name__ == "__main__":
|
||
main()
|