84 lines
3.8 KiB
Python
84 lines
3.8 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
"""
|
||
|
|
notion-mcp-connect · 冰朔/铸渊/之之 通用的 Notion 连接工具
|
||
|
|
铸渊 ICE-GL-ZY001 + Mavis · D166 · 2026-07-06
|
||
|
|
|
||
|
|
功能:
|
||
|
|
search <query> 搜索页面/数据库
|
||
|
|
get <page_id> 获取页面详情
|
||
|
|
blocks <page_id> 列出页面的 blocks
|
||
|
|
create <parent> <title> 创建子页面
|
||
|
|
append <page_id> <text> 追加 paragraph block
|
||
|
|
check-todo <block_id> 勾选/取消 to_do block
|
||
|
|
check 检查 token 有效性
|
||
|
|
query <db_id> 查询数据库
|
||
|
|
|
||
|
|
环境: NOTION_TOKEN=ntn_... 或命令行第二个参数
|
||
|
|
API version: 2022-06-28 (默认)
|
||
|
|
"""
|
||
|
|
import json, os, sys, urllib.request, urllib.error
|
||
|
|
|
||
|
|
NOTION_TOKEN = os.environ.get("NOTION_TOKEN", "")
|
||
|
|
NOTION_VERSION = os.environ.get("NOTION_VERSION", "2022-06-28")
|
||
|
|
NOTION_BASE_URL = os.environ.get("NOTION_BASE_URL", "https://api.notion.com")
|
||
|
|
|
||
|
|
class NotionMCP:
|
||
|
|
def __init__(self, token):
|
||
|
|
if not token:
|
||
|
|
raise ValueError("需要 NOTION_TOKEN 环境变量或第二个参数")
|
||
|
|
self.token, self.base_url, self.version = token, NOTION_BASE_URL, NOTION_VERSION
|
||
|
|
|
||
|
|
def _req(self, method, path, body=None):
|
||
|
|
url = f"{self.base_url}{path}"
|
||
|
|
data = json.dumps(body).encode() if body else None
|
||
|
|
req = urllib.request.Request(url, data=data, method=method,
|
||
|
|
headers={"Authorization": f"Bearer {self.token}", "Notion-Version": self.version, "Content-Type": "application/json"})
|
||
|
|
try:
|
||
|
|
with urllib.request.urlopen(req, timeout=30) as r:
|
||
|
|
return json.loads(r.read().decode())
|
||
|
|
except urllib.error.HTTPError as e:
|
||
|
|
return {"error": True, "status": e.code, "body": e.read().decode()[:500]}
|
||
|
|
|
||
|
|
def check(self): return self._req("GET", "/v1/users/me")
|
||
|
|
def search(self, q, f=None, n=10):
|
||
|
|
b = {"query": q, "page_size": n}
|
||
|
|
if f: b["filter"] = {"value": f, "property": "object"}
|
||
|
|
return self._req("POST", "/v1/search", b)
|
||
|
|
def get_page(self, pid): return self._req("GET", f"/v1/pages/{pid}")
|
||
|
|
def get_blocks(self, pid): return self._req("GET", f"/v1/blocks/{pid}/children?page_size=100")
|
||
|
|
def create_page(self, parent, title, ptype="page_id"):
|
||
|
|
return self._req("POST", "/v1/pages", {"parent": {ptype: parent}, "properties": {"title": [{"type": "text", "text": {"content": title}}]}})
|
||
|
|
def append_block(self, pid, text, btype="paragraph"):
|
||
|
|
return self._req("PATCH", f"/v1/blocks/{pid}/children", {"children": [{"object": "block", "type": btype, btype: {"rich_text": [{"type": "text", "text": {"content": text}}]}}]})
|
||
|
|
def update_todo(self, bid, checked):
|
||
|
|
return self._req("PATCH", f"/v1/blocks/{bid}", {"to_do": {"checked": checked}})
|
||
|
|
def query_db(self, did, f=None, n=100):
|
||
|
|
b = {"page_size": n}
|
||
|
|
if f: b["filter"] = f
|
||
|
|
return self._req("POST", f"/v1/databases/{did}/query", b)
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
if len(sys.argv) < 2 or sys.argv[1] in ("-h", "--help"):
|
||
|
|
print(__doc__); return 0
|
||
|
|
cmd = sys.argv[1]
|
||
|
|
token = NOTION_TOKEN
|
||
|
|
ao = 2
|
||
|
|
if len(sys.argv) > 2 and sys.argv[2].startswith("ntn_"):
|
||
|
|
token, ao = sys.argv[2], 3
|
||
|
|
m = NotionMCP(token)
|
||
|
|
methods = {"check": m.check, "search": lambda: m.search(sys.argv[ao]),
|
||
|
|
"get": lambda: m.get_page(sys.argv[ao]),
|
||
|
|
"blocks": lambda: m.get_blocks(sys.argv[ao]),
|
||
|
|
"create": lambda: m.create_page(sys.argv[ao], sys.argv[ao+1]),
|
||
|
|
"append": lambda: m.append_block(sys.argv[ao], sys.argv[ao+1]),
|
||
|
|
"check-todo": lambda: m.update_todo(sys.argv[ao], sys.argv[ao+1].lower()=="true" if len(sys.argv)>ao+1 else True),
|
||
|
|
"query": lambda: m.query_db(sys.argv[ao])}
|
||
|
|
if cmd not in methods:
|
||
|
|
print(f"未知命令: {cmd}"); print(__doc__); return 1
|
||
|
|
print(json.dumps(methods[cmd](), ensure_ascii=False, indent=2))
|
||
|
|
return 0
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
sys.exit(main())
|