mirror of
https://git.datalinker.icu/comfyanonymous/ComfyUI
synced 2026-08-16 08:43:30 +08:00
106 lines
3.2 KiB
Python
106 lines
3.2 KiB
Python
import os
|
|
from notion_client import Client
|
|
from datetime import datetime
|
|
|
|
# 设置你的 Notion API Token 和 数据库 ID
|
|
NOTION_TOKEN = "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"
|
|
NOTION_DATABASE_ID = "230de91f217680388daed107820a5dc8"
|
|
|
|
if not NOTION_TOKEN or not NOTION_DATABASE_ID:
|
|
print("请设置 NOTION_TOKEN 和 NOTION_DATABASE_ID 环境变量。")
|
|
exit()
|
|
|
|
# 初始化 Notion 客户端
|
|
notion = Client(auth=NOTION_TOKEN)
|
|
|
|
def get_all_records_from_database():
|
|
"""
|
|
从 Notion 数据库获取所有记录的 ID。
|
|
"""
|
|
records_id = []
|
|
has_more = True
|
|
next_cursor = None
|
|
|
|
print(f"正在从数据库 '{NOTION_DATABASE_ID}' 获取记录...")
|
|
while has_more:
|
|
try:
|
|
# 查询数据库
|
|
# 注意: Notion API 对查询结果有分页限制 (通常每页 100 条)
|
|
response = notion.databases.query(
|
|
database_id=NOTION_DATABASE_ID,
|
|
start_cursor=next_cursor
|
|
)
|
|
|
|
for page in response["results"]:
|
|
records_id.append(page["id"])
|
|
|
|
has_more = response["has_more"]
|
|
next_cursor = response["next_cursor"]
|
|
|
|
except Exception as e:
|
|
print(f"获取记录失败: {e}")
|
|
break
|
|
|
|
print(f"找到 {len(records_id)} 条记录。")
|
|
return records_id
|
|
|
|
def archive_record(page_id):
|
|
"""
|
|
归档（删除）指定的 Notion 页面（记录）。
|
|
"""
|
|
try:
|
|
notion.pages.update(
|
|
page_id=page_id,
|
|
archived=True
|
|
)
|
|
print(f"成功归档页面: {page_id}")
|
|
except Exception as e:
|
|
print(f"归档页面 {page_id} 失败: {e}")
|
|
|
|
def delete_all_records():
|
|
"""
|
|
删除数据库中所有现有记录，然后插入新的记录。
|
|
|
|
Args:
|
|
new_record_data (list of dict): 包含要插入的新记录数据的列表，
|
|
每个字典应包含 'name', 'tags', 'status' 等键。
|
|
"""
|
|
print("\n--- 步骤 1: 删除（归档）所有现有记录 ---")
|
|
record_ids_to_delete = get_all_records_from_database()
|
|
|
|
if not record_ids_to_delete:
|
|
print("数据库中没有需要删除的记录。")
|
|
else:
|
|
for page_id in record_ids_to_delete:
|
|
archive_record(page_id)
|
|
print("所有现有记录已归档。")
|
|
|
|
def add_record_to_notion_database(url):
|
|
"""
|
|
向 Notion 数据库插入一条新记录。
|
|
|
|
"""
|
|
current_datetime = datetime.now()
|
|
formatted_datetime = current_datetime.strftime("%Y-%m-%d %H:%M:%S")
|
|
|
|
try:
|
|
new_page = notion.pages.create(
|
|
parent={"database_id": NOTION_DATABASE_ID},
|
|
properties={
|
|
"更新时间": {
|
|
"title": [
|
|
{
|
|
"text": {"content": formatted_datetime}
|
|
}
|
|
]
|
|
},
|
|
"地址": {
|
|
"url": url # <<--- 传递一个简单的 URL 字符串
|
|
}
|
|
# 根据你的数据库结构添加更多属性
|
|
},
|
|
)
|
|
print(f"成功插入记录: {url}")
|
|
except Exception as e:
|
|
print(f"插入记录失败: {e}")
|