✨ 初步实现携带cookie请求

nonebot_bison/platform/rss.py

@@ -9,6 +9,7 @@ from bs4 import BeautifulSoup as bs
 from ..post import Post
 from .platform import NewMessage
 from ..types import Target, RawPost
+from ..utils.site import create_cookie_client_manager
 from ..utils import Site, text_fletten, text_similarity
 
 
@@ -16,6 +17,7 @@ class RssSite(Site):
     name = "rss"
     schedule_type = "interval"
     schedule_setting = {"seconds": 30}
+    client_mgr = create_cookie_client_manager("rss")
 
 
 class RssPost(Post):
@@ -63,7 +65,7 @@ class Rss(NewMessage):
         return post.id
 
     async def get_sub_list(self, target: Target) -> list[RawPost]:
-        client = await self.ctx.get_client()
+        client = await self.ctx.get_client(target)
         res = await client.get(target, timeout=10.0)
         feed = feedparser.parse(res)
         entries = feed.entries

nonebot_bison/utils/site.py

@@ -1,9 +1,12 @@
+import json
 from typing import Literal
 from abc import ABC, abstractmethod
 
+import httpx
 from httpx import AsyncClient
 
 from ..types import Target
+from ..config import config
 from .http import http_client
 
 
@@ -35,6 +38,45 @@ class DefaultClientManager(ClientManager):
         pass
 
 
+class CookieClientManager(ClientManager):
+    _platform_name: str
+
+    async def _choose_cookie(self, target: Target) -> dict[str, str]:
+        if not target:
+            return {}
+        cookies = await config.get_cookie_by_target(target, self._platform_name)
+        if not cookies:
+            return {}
+        cookie = sorted(cookies, key=lambda x: x.last_usage, reverse=True)[0]
+        return json.loads(cookie.content)
+
+    async def get_client(self, target: Target | None) -> AsyncClient:
+        client = http_client()
+        cookie = await self._choose_cookie(target)
+        cookies = httpx.Cookies()
+        if cookie:
+            cookies.update(cookie)
+        client.cookies = cookies
+        return client
+
+    async def get_client_for_static(self) -> AsyncClient:
+        pass
+
+    async def get_query_name_client(self) -> AsyncClient:
+        pass
+
+    async def refresh_client(self):
+        pass
+
+
+def create_cookie_client_manager(platform_name: str) -> type[CookieClientManager]:
+    return type(
+        "CookieClientManager",
+        (CookieClientManager,),
+        {"_platform_name": platform_name},
+    )
+
+
 class Site:
     schedule_type: Literal["date", "interval", "cron"]
     schedule_setting: dict