From ac505f51d3dc2fa9d3a5b2e4357d18ab7bda81f1 Mon Sep 17 00:00:00 2001
From: Azide <rukuy@qq.com>
Date: Fri, 23 Sep 2022 23:17:58 +0800
Subject: [PATCH] =?UTF-8?q?fix(platform-bilibili/bilibililive):=E4=B8=BA?=
 =?UTF-8?q?=E7=9B=B8=E5=85=B3=E6=8E=A5=E5=8F=A3=E6=B7=BB=E5=8A=A0UA,=20?=
 =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E6=8E=A5=E5=8F=A3=E4=BF=A1=E6=81=AF=E8=8E=B7?=
 =?UTF-8?q?=E5=8F=96=E5=A4=B1=E8=B4=A5=E7=9A=84=E9=97=AE=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/plugins/nonebot_bison/platform/bilibili.py | 15 +++++++++++++--
 1 file changed, 13 insertions(+), 2 deletions(-)

diff --git a/src/plugins/nonebot_bison/platform/bilibili.py b/src/plugins/nonebot_bison/platform/bilibili.py
index 7d5a763..71e07ac 100644
--- a/src/plugins/nonebot_bison/platform/bilibili.py
+++ b/src/plugins/nonebot_bison/platform/bilibili.py
@@ -27,11 +27,16 @@ class Bilibili(NewMessage):
     name = "B站"
     has_target = True
     parse_target_promot = "请输入用户主页的链接"
+    UA = {
+        "User-Agent": "Mozilla/5.0 (Windows NT 6.3; WOW64; Trident/7.0; rv:11.0) like Gecko",
+    }
 
     async def get_target_name(self, target: Target) -> Optional[str]:
         async with http_client() as client:
             res = await client.get(
-                "https://api.bilibili.com/x/space/acc/info", params={"mid": target}
+                "https://api.bilibili.com/x/space/acc/info",
+                params={"mid": target},
+                headers=self.UA,
             )
             res_data = json.loads(res.text)
             if res_data["code"]:
@@ -171,11 +176,16 @@ class Bilibililive(StatusChange):
     schedule_kw = {"seconds": 10}
     name = "Bilibili直播"
     has_target = True
+    UA = {
+        "User-Agent": "Mozilla/5.0 (Windows NT 6.3; WOW64; Trident/7.0; rv:11.0) like Gecko",
+    }
 
     async def get_target_name(self, target: Target) -> Optional[str]:
         async with http_client() as client:
             res = await client.get(
-                "https://api.bilibili.com/x/space/acc/info", params={"mid": target}
+                "https://api.bilibili.com/x/space/acc/info",
+                params={"mid": target},
+                headers=self.UA,
             )
             res_data = json.loads(res.text)
             if res_data["code"]:
@@ -188,6 +198,7 @@ class Bilibililive(StatusChange):
             res = await client.get(
                 "https://api.bilibili.com/x/space/acc/info",
                 params=params,
+                headers=self.UA,
                 timeout=4.0,
             )
             res_dict = json.loads(res.text)