mirror of
https://github.com/suyiiyii/nonebot-bison.git
synced 2025-06-04 02:26:11 +08:00
af72b6c3d0
* ✨ 使用新接口 * ♻️ 调整刷新逻辑 * 🐛 调整刷新逻辑 * ♻️ 将单个哔哩哔哩文件拆开 * 🐛 修修补补边界情况 * ✨ 添加UID:xxx匹配 * ✅ 调整测试中的导入 * ✅ 调整测试的断言 * 🐛 添加unicode字符的escape * ✨ 不再主动刷新cookies * 🔀 适配新版Site * 🐛 解析live_rcmd中的json string * 🚨 make ruff happy * 🐛 调整并测试bilibili retry函数 * ✅ 修正测试 * ♻️ 按review意见调整 * ♻️ 清理一些遗留的复杂写法 * ♻️ 移出函数内的NameTuple * 🔇 删除不必要的日志输出 Co-authored-by: felinae98 <731499577@qq.com> * Update nonebot_bison/platform/bilibili/scheduler.py * Update scheduler.py --------- Co-authored-by: felinae98 <731499577@qq.com>
81 lines
2.3 KiB
Python
81 lines
2.3 KiB
Python
from random import randint
|
|
from typing_extensions import override
|
|
|
|
from httpx import AsyncClient
|
|
from nonebot import logger, require
|
|
from playwright.async_api import Cookie
|
|
|
|
from nonebot_bison.types import Target
|
|
from nonebot_bison.utils import Site, ClientManager, http_client
|
|
|
|
require("nonebot_plugin_htmlrender")
|
|
from nonebot_plugin_htmlrender import get_browser
|
|
|
|
|
|
class BilibiliClientManager(ClientManager):
|
|
_client: AsyncClient
|
|
_inited: bool = False
|
|
|
|
def __init__(self) -> None:
|
|
self._client = http_client()
|
|
|
|
async def _get_cookies(self) -> list[Cookie]:
|
|
browser = await get_browser()
|
|
async with await browser.new_page() as page:
|
|
await page.goto(f"https://space.bilibili.com/{randint(1, 1000)}/dynamic")
|
|
await page.wait_for_load_state("load")
|
|
cookies = await page.context.cookies()
|
|
|
|
return cookies
|
|
|
|
async def _reset_client_cookies(self, cookies: list[Cookie]):
|
|
for cookie in cookies:
|
|
self._client.cookies.set(
|
|
name=cookie.get("name", ""),
|
|
value=cookie.get("value", ""),
|
|
domain=cookie.get("domain", ""),
|
|
path=cookie.get("path", "/"),
|
|
)
|
|
|
|
@override
|
|
async def refresh_client(self):
|
|
cookies = await self._get_cookies()
|
|
await self._reset_client_cookies(cookies)
|
|
logger.debug("刷新B站客户端的cookie")
|
|
|
|
@override
|
|
async def get_client(self, target: Target | None) -> AsyncClient:
|
|
if not self._inited:
|
|
logger.debug("初始化B站客户端")
|
|
await self.refresh_client()
|
|
self._inited = True
|
|
return self._client
|
|
|
|
@override
|
|
async def get_client_for_static(self) -> AsyncClient:
|
|
return http_client()
|
|
|
|
@override
|
|
async def get_query_name_client(self) -> AsyncClient:
|
|
return http_client()
|
|
|
|
|
|
class BilibiliSite(Site):
|
|
name = "bilibili.com"
|
|
schedule_setting = {"seconds": 30}
|
|
schedule_type = "interval"
|
|
client_mgr = BilibiliClientManager
|
|
require_browser = True
|
|
|
|
|
|
class BililiveSite(Site):
|
|
name = "live.bilibili.com"
|
|
schedule_setting = {"seconds": 5}
|
|
schedule_type = "interval"
|
|
|
|
|
|
class BiliBangumiSite(Site):
|
|
name = "bilibili.com/bangumi"
|
|
schedule_setting = {"seconds": 30}
|
|
schedule_type = "interval"
|