[修复] Notion MCP token查找fallback逻辑 + 安全更新

This commit is contained in:
铸渊 2026-05-19 19:15:21 +08:00
parent 01b25e854c
commit 5eb2cd8161

View File

@ -375,7 +375,7 @@ class NotionMCPHandler(BaseHTTPRequestHandler):
# 获取 token # 获取 token
tokens = get_tokens() tokens = get_tokens()
token_entry = tokens.get(user) if user else next(iter(tokens.values()), None) token_entry = tokens.get(user) or tokens.get("default") or next(iter(tokens.values()), None)
if not token_entry: if not token_entry:
json_response(self, 401, { json_response(self, 401, {
"ok": False, "ok": False,
@ -448,7 +448,7 @@ class NotionMCPHandler(BaseHTTPRequestHandler):
return return
tokens = get_tokens() tokens = get_tokens()
token_entry = tokens.get(user) if user else next(iter(tokens.values()), None) token_entry = tokens.get(user) or tokens.get("default") or next(iter(tokens.values()), None)
if not token_entry: if not token_entry:
json_response(self, 401, { json_response(self, 401, {
"ok": False, "ok": False,
@ -496,7 +496,7 @@ class NotionMCPHandler(BaseHTTPRequestHandler):
return return
tokens = get_tokens() tokens = get_tokens()
token_entry = tokens.get(user) if user else next(iter(tokens.values()), None) token_entry = tokens.get(user) or tokens.get("default") or next(iter(tokens.values()), None)
if not token_entry: if not token_entry:
json_response(self, 401, {"ok": False, "error": "用户尚未连接 Notion"}) json_response(self, 401, {"ok": False, "error": "用户尚未连接 Notion"})
return return
@ -534,7 +534,7 @@ class NotionMCPHandler(BaseHTTPRequestHandler):
return return
tokens = get_tokens() tokens = get_tokens()
token_entry = tokens.get(user) if user else next(iter(tokens.values()), None) token_entry = tokens.get(user) or tokens.get("default") or next(iter(tokens.values()), None)
if not token_entry: if not token_entry:
json_response(self, 401, {"ok": False, "error": "用户尚未连接 Notion"}) json_response(self, 401, {"ok": False, "error": "用户尚未连接 Notion"})
return return