65 lines
2.5 KiB
Python
65 lines
2.5 KiB
Python
#!/usr/bin/env python3
|
|
"""创建方案页面，挂到「📋 方案库」目录下"""
|
|
import json, sys, os, argparse
|
|
import subprocess
|
|
|
|
SCRIPTS_DIR = os.path.dirname(os.path.abspath(__file__))
|
|
SPACE_ID = os.environ.get('FEISHU_SPACE_ID', '7664817589230570761')
|
|
PLAN_PARENT = os.environ.get('FEISHU_PLAN_PARENT', 'LjXOwVZkriKmQwkjoixcnrSbnhd') # 📋 方案库
|
|
|
|
def get_token():
|
|
r = subprocess.run(['python3', os.path.join(SCRIPTS_DIR, 'init_token.py')],
|
|
capture_output=True, text=True)
|
|
return json.loads(r.stdout)['token']
|
|
|
|
def create_plan(title, content_file=None, parent=None):
|
|
import requests
|
|
token = get_token()
|
|
h = {'Authorization': 'Bearer ' + token, 'Content-Type': 'application/json'}
|
|
parent = parent or PLAN_PARENT
|
|
|
|
# 检查同名节点是否已存在（避免重复创建）
|
|
r = requests.get(
|
|
f'https://open.feishu.cn/open-apis/wiki/v2/spaces/{SPACE_ID}/nodes',
|
|
params={'parent_node_token': parent}, headers=h
|
|
)
|
|
kids = r.json().get('data', {}).get('items', [])
|
|
for k in kids:
|
|
if k.get('title') == title:
|
|
print(json.dumps({'node_token': k['node_token'], 'obj_token': k.get('obj_token'), 'exists': True}))
|
|
return k
|
|
|
|
# 创建新节点
|
|
r = requests.post(
|
|
f'https://open.feishu.cn/open-apis/wiki/v2/spaces/{SPACE_ID}/nodes',
|
|
headers=h,
|
|
json={'obj_type': 'docx', 'node_type': 'origin', 'parent_node_token': parent, 'title': title}
|
|
)
|
|
data = r.json()
|
|
if data.get('code') != 0:
|
|
raise Exception(f"Create plan node failed: {data.get('msg')}")
|
|
|
|
node = data['data']['node']
|
|
print(json.dumps({'node_token': node['node_token'], 'obj_token': node['obj_token'], 'exists': False}))
|
|
|
|
# 写入内容
|
|
if content_file:
|
|
subprocess.run(
|
|
['python3', os.path.join(SCRIPTS_DIR, 'wiki_write.py'), '--node', node['node_token'], '--file', content_file],
|
|
check=True
|
|
)
|
|
return node
|
|
|
|
if __name__ == '__main__':
|
|
p = argparse.ArgumentParser(description='创建飞书方案库页面')
|
|
p.add_argument('--title', required=True, help='方案标题')
|
|
p.add_argument('--content', default=None, help='内容 markdown 文件路径')
|
|
p.add_argument('--parent', default=None, help='父节点 token（默认方案库）')
|
|
args = p.parse_args()
|
|
|
|
try:
|
|
create_plan(args.title, args.content, args.parent)
|
|
except Exception as e:
|
|
print(f'Error: {e}', file=sys.stderr)
|
|
sys.exit(1)
|