From 2beb23a1d5f06c4f9126c6e3f4c5489cf089e161 Mon Sep 17 00:00:00 2001
From: woni <320753691@qq.com>
Date: Tue, 16 Apr 2024 16:42:51 +0800
Subject: [PATCH] adjust node token

---
 app/__init__.py     | 10 +++++++++-
 app/task.py         | 16 ++++++++++++----
 app/utils/feishu.py |  3 ++-
 3 files changed, 23 insertions(+), 6 deletions(-)

diff --git a/app/__init__.py b/app/__init__.py
index a936bbe..5995fe0 100644
--- a/app/__init__.py
+++ b/app/__init__.py
@@ -124,7 +124,15 @@ def do_bitable_record_add(data) -> None:
     print(f"target_action: {target_action.get('action')}")
     print(action_data)
 
-    process_new_demand.delay(app_id, app_secret, action_data)
+    table_id = "tblESHsFZwtiFROB"
+
+    process_new_demand.delay(
+        app_id,
+        app_secret,
+        table_id,
+        record_id,
+        action_data,
+    )
 
 
 handler = (
diff --git a/app/task.py b/app/task.py
index b96d05c..57d3837 100644
--- a/app/task.py
+++ b/app/task.py
@@ -9,6 +9,7 @@ from celery import Task
 from app.utils.utils import *
 from app.utils.yd_utils import get_token, get_activity_info
 from app.utils.feishu import create_doc, get_doc_block, create_doc_block
+from app.utils.utils import update_record
 
 
 @shared_task(include_result=False)
@@ -127,7 +128,7 @@ def sync_xbot_activity_info():
 
 
 @shared_task(include_result=False)
-def process_new_demand(app_id, app_secret, data):
+def process_new_demand(app_id, app_secret, table_id, record_id, data):
 
     title = data.get("title")
     background = data.get("background")
@@ -143,6 +144,7 @@ def process_new_demand(app_id, app_secret, data):
     )
 
     space_id = "7268552249415319554"
+    bitable_token = os.environ["OFFICE_XBOT_BITABLE_TOKEN"]
 
     # 放官方应用下
     if category == "opt5xydmUE":
@@ -151,7 +153,13 @@ def process_new_demand(app_id, app_secret, data):
     if category == "optRTwYcra":
         parent_node_token = "SvWCwlWJ8iuFKpkXp1rcd9u2n3c"
 
-    doc_id = create_doc(client, space_id, parent_node_token, title)
-    block_id = get_doc_block(client, doc_id)
+    doc_node_toke = create_doc(client, space_id, parent_node_token, title)
+    block_id = get_doc_block(client, doc_node_toke)
 
-    create_doc_block(client, doc_id, block_id, background, features)
+    create_doc_block(client, doc_node_toke, block_id, background, features)
+
+    doc_node_url = f"https://ying-dao.feishu.cn/wiki/{doc_node_toke}"
+
+    update_record(
+        client, bitable_token, table_id, record_id, {"需求文档链接": doc_node_url}
+    )
diff --git a/app/utils/feishu.py b/app/utils/feishu.py
index c79ae93..38e8351 100644
--- a/app/utils/feishu.py
+++ b/app/utils/feishu.py
@@ -112,7 +112,8 @@ def create_doc(client, space_id, parent_node_token, title):
         )
         return
 
-    return response.data.node.obj_token
+    print(lark.JSON.marshal(response.data))
+    return response.data.node.node_token
 
 
 def get_doc_block(client, document_id):