[teamai] Push 1 resource(s) from root

2026-09-10 16:51:20 +08:00
parent f3388c50e2
commit 766efe4076
15 changed files with 1284 additions and 0 deletions
@@ -0,0 +1,158 @@
+#!/usr/bin/env python3
+"""写入 Wiki 页面内容。支持 Markdown 文件（含表格）和 Block JSON。
+表格块（block_type=31）自动走 descendant API 创建嵌套结构。
+"""
+import json, sys, os, argparse
+import subprocess
+
+SPACE_ID = os.environ.get('FEISHU_SPACE_ID', '7664817589230570761')
+
+def get_token():
+    r = subprocess.run(['python3', os.path.join(os.path.dirname(__file__), 'init_token.py')],
+                       capture_output=True, text=True)
+    return json.loads(r.stdout)['token']
+
+def strip_table_meta(block):
+    """Remove __table/__rows/__cols internal markers."""
+    out = {}
+    for k, v in block.items():
+        if not k.startswith('__'):
+            out[k] = v
+    return out
+
+def write_blocks(node_token, blocks):
+    import requests
+    token = get_token()
+    h = {'Authorization': f'Bearer {token}', 'Content-Type': 'application/json'}
+
+    # Get doc_id
+    r = requests.get(
+        'https://open.feishu.cn/open-apis/wiki/v2/spaces/get_node',
+        params={'token': node_token}, headers=h
+    )
+    nd = r.json()
+    if nd.get('code') != 0:
+        raise Exception(f"Get node failed: {nd.get('msg')}")
+    doc_id = nd['data']['node']['obj_token']
+
+    # Clear existing content
+    r = requests.get(
+        f'https://open.feishu.cn/open-apis/docx/v1/documents/{doc_id}/blocks/{doc_id}',
+        headers=h
+    )
+    kids = r.json().get('data', {}).get('block', {}).get('children', [])
+    if kids:
+        cids = [c if isinstance(c, str) else c['block_id'] for c in kids]
+        requests.delete(
+            f'https://open.feishu.cn/open-apis/docx/v1/documents/{doc_id}/blocks/{doc_id}/children/batch_delete',
+            headers=h, json={'start_index': 0, 'end_index': len(cids)}
+        )
+
+    # Write blocks in source order: normal blocks interleaved with tables
+    batch_size = 25
+
+    def write_normal_batch(batch):
+        for idx in range(0, len(batch), batch_size):
+            chunk = batch[idx:idx + batch_size]
+            r = requests.post(
+                f'https://open.feishu.cn/open-apis/docx/v1/documents/{doc_id}/blocks/{doc_id}/children',
+                headers=h, json={'children': chunk, 'index': -1}
+            )
+            if r.json().get('code') != 0:
+                raise Exception(f"Write batch failed: {r.json().get('msg')}")
+
+    def write_table_group(group):
+        table_block = group[0]
+        descendants = [strip_table_meta(b) for b in group]
+        table_bid = table_block.get('block_id', table_block.get('children', [''])[0])
+        payload = {
+            'index': -1,
+            'children_id': [table_bid],
+            'descendants': descendants
+        }
+        r = requests.post(
+            f'https://open.feishu.cn/open-apis/docx/v1/documents/{doc_id}/blocks/{doc_id}/descendant?document_revision_id=-1',
+            headers=h, json=payload
+        )
+        rd = r.json()
+        if rd.get('code') != 0:
+            raise Exception(f"Table descendant API failed: {rd.get('msg')}")
+
+    normal_batch = []
+    i = 0
+    while i < len(blocks):
+        b = blocks[i]
+        if b.get('__table'):
+            # Flush pending normal blocks before the table to keep source order
+            if normal_batch:
+                write_normal_batch(normal_batch)
+                normal_batch = []
+            # Collect this table group: table + cells + cell content
+            group = []
+            j = i
+            while j < len(blocks):
+                bj = blocks[j]
+                bid = bj.get('block_id', '')
+                if bj.get('__table') and j != i:
+                    break
+                if bj.get('block_type') == 31 and not bj.get('__table'):
+                    break
+                if not bj.get('__table') and not bid.startswith('ecc_t'):
+                    if bj.get('block_type') == 2 and bid:
+                        elements = bj.get('text', {}).get('elements', [])
+                        if elements and all(e.get('text_run', {}).get('content', '') == '' for e in elements):
+                            pass  # empty spacer included
+                        else:
+                            break
+                    else:
+                        break
+                group.append(bj)
+                j += 1
+            write_table_group(group)
+            i = j
+        else:
+            normal_batch.append(b)
+            i += 1
+
+    # Flush remaining normal blocks
+    if normal_batch:
+        write_normal_batch(normal_batch)
+
+    return True
+
+if __name__ == '__main__':
+    p = argparse.ArgumentParser()
+    p.add_argument('--node', required=True, help='Node token')
+    p.add_argument('--file', default=None, help='Content file (JSON block array or Markdown)')
+    p.add_argument('--json', default=None, help='Inline JSON block array')
+    p.add_argument('--markdown', action='store_true', help='Treat --file as Markdown')
+    args = p.parse_args()
+
+    try:
+        if args.json:
+            data = json.loads(args.json)
+        elif args.file:
+            if args.markdown or args.file.endswith('.md'):
+                r = subprocess.run(
+                    ['python3', os.path.join(os.path.dirname(__file__), 'md_to_blocks.py'), args.file],
+                    capture_output=True, text=True
+                )
+                if r.returncode != 0:
+                    raise Exception(f"md_to_blocks failed: {r.stderr}")
+                data = json.loads(r.stdout)
+            else:
+                with open(args.file) as f:
+                    data = json.load(f)
+        else:
+            data = json.load(sys.stdin)
+
+        if isinstance(data, list) and not isinstance(data, str):
+            blocks = data
+        else:
+            blocks = data
+
+        write_blocks(args.node, blocks)
+        print('OK')
+    except Exception as e:
+        print(f'Error: {e}', file=sys.stderr)
+        sys.exit(1)