126 lines
4.2 KiB
Python
126 lines
4.2 KiB
Python
import time
|
|
import json
|
|
|
|
from itertools import chain
|
|
|
|
from celery import shared_task
|
|
from celery import Task
|
|
|
|
from app.utils.utils import *
|
|
from app.utils.yd_utils import get_token, get_activity_info
|
|
|
|
|
|
@shared_task(include_result=False)
|
|
def process_message_callback(
|
|
app_id, app_secret, bitable_token, table_id, cache_dir, data
|
|
) -> None:
|
|
client = lark.Client.builder().app_id(app_id).app_secret(app_secret).build()
|
|
|
|
record = {}
|
|
images = []
|
|
images_token = []
|
|
message_content = ""
|
|
# message_type = data.event.message.message_type
|
|
# content = json.loads(data.event.message.content)
|
|
# sender_id = data.event.sender.sender_id.open_id
|
|
# message_id = data.event.message.message_id
|
|
|
|
message_type = data.get("message_type")
|
|
content = data.get("content")
|
|
sender_id = data.get("sender_id")
|
|
message_id = data.get("message_id")
|
|
|
|
if message_type == "post":
|
|
content = content["content"]
|
|
for item in chain.from_iterable(content):
|
|
if item.get("tag") == "text":
|
|
message_content += item["text"]
|
|
|
|
if item.get("tag") == "img":
|
|
images.append(item["image_key"])
|
|
|
|
if message_type == "text":
|
|
message_content = content["text"].replace("@_user_1", "").strip()
|
|
|
|
record["问题概述"] = message_content
|
|
record["创建人"] = [{"id": sender_id}]
|
|
|
|
# 对 image_key 处理并并上传至多维表格
|
|
|
|
for image in images:
|
|
image_path = get_message_resource(client, message_id, image, "image", cache_dir)
|
|
print("下载文件成功:", image, image_path)
|
|
file_token = upload_media(client, image_path, "bitable_image", bitable_token)
|
|
print("上传文件成功:", file_token)
|
|
images_token.append(file_token)
|
|
|
|
if images:
|
|
record["附件"] = [{"file_token": image} for image in images_token]
|
|
|
|
record_id = add_record(client, bitable_token, table_id, record)
|
|
|
|
time.sleep(1)
|
|
|
|
update_data = {"创建人": {"id": sender_id}}
|
|
update_record(client, bitable_token, table_id, record_id, update_data)
|
|
|
|
# content = {"text": "反馈的问题已经记录到多维表格, 稍后有小伙伴跟进"}
|
|
# send_message(receive_id, content)
|
|
|
|
# 回复群组的消息
|
|
content = f'{{"text": "反馈的问题已经记录到多维表格, 稍后有小伙伴跟进"}}'
|
|
send_group_message(client, message_id, content)
|
|
|
|
|
|
@shared_task()
|
|
def sync_xbot_activity_info():
|
|
app_id = os.environ["APP_ID"]
|
|
app_secret = os.environ["APP_SECRET"]
|
|
|
|
bitable_token = os.environ["OFFICE_XBOT_BITABLE_TOKEN"]
|
|
table_id = os.environ["OFFICE_XBOT_TABLE_ID"]
|
|
view_id = os.environ["OFFICE_XBOT_VIEW_ID"]
|
|
|
|
xbot_username = os.environ["XBOT_USERNAME"]
|
|
xbot_password = os.environ["XBOT_PASSWORD"]
|
|
|
|
client = lark.Client.builder().app_id(app_id).app_secret(app_secret).build()
|
|
|
|
token = get_token(xbot_username, xbot_password)
|
|
|
|
handle_list = []
|
|
|
|
# 同步应用数据
|
|
# office_app_uuids = get_office_app_uuids()
|
|
|
|
# print(f"office_activity_uuids count: {len(office_app_uuids)}")
|
|
|
|
# 同步指令数据
|
|
office_activity_uuids = get_activity_info(token)
|
|
print(f"office_activity_uuids count: {len(office_activity_uuids)}")
|
|
|
|
for item in office_activity_uuids:
|
|
robot_uuid = item.get("robotUuid")
|
|
package_view_name = item.get("packageViewName")
|
|
print("=====================================")
|
|
print(f"同步信息: {robot_uuid}, {package_view_name}")
|
|
record_id = get_filtered_record_by_uuid(
|
|
client, bitable_token, table_id, view_id, "UUID", robot_uuid
|
|
)
|
|
if record_id:
|
|
print(f"{robot_uuid}已同步多维表格中")
|
|
update_record(
|
|
client, bitable_token, table_id, record_id, {"名称": package_view_name}
|
|
)
|
|
else:
|
|
handle_list.append(robot_uuid)
|
|
print(f"{robot_uuid}: {package_view_name} 未同步多维表格中")
|
|
add_record(
|
|
client,
|
|
bitable_token,
|
|
table_id,
|
|
{"UUID": robot_uuid, "名称": package_view_name, "分类": "官方指令"},
|
|
)
|
|
print(f"新增的数量: {len(handle_list)}")
|
|
print("新增的uuid: ", handle_list)
|