Merge pull request #60 from Sichongzou/main

添加bilibili开播提醒

README.md

@@ -38,6 +38,8 @@
   - 专栏
   - 转发
   - 纯文字
+- Bilibili 直播
+  - 开播提醒
 - RSS
   - 富文本转换为纯文本
   - 提取出所有图片

src/plugins/nonebot_bison/platform/bilibili.py

@@ -5,7 +5,7 @@ from typing import Any, Optional
 from ..post import Post
 from ..types import Category, RawPost, Tag, Target
 from ..utils import http_client
-from .platform import CategoryNotSupport, NewMessage
+from .platform import CategoryNotSupport, NewMessage, StatusChange
 
 
 class Bilibili(NewMessage):
@@ -155,3 +155,73 @@ class Bilibili(NewMessage):
         else:
             raise CategoryNotSupport(post_type)
         return Post("bilibili", text=text, url=url, pics=pic, target_name=target_name)
+
+
+class Bilibililive(StatusChange):
+    # Author : Sichongzou
+    # Date : 2022-5-18 8:54
+    # Description : bilibili开播提醒
+    # E-mail : 1557157806@qq.com
+    categories = {}
+    platform_name = "bilibili-live"
+    enable_tag = True
+    enabled = True
+    is_common = True
+    schedule_type = "interval"
+    schedule_kw = {"seconds": 10}
+    name = "Bilibili直播"
+    has_target = True
+
+    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}
+            )
+            res_data = json.loads(res.text)
+            if res_data["code"]:
+                return None
+            return res_data["data"]["name"]
+
+    async def get_status(self, target: Target):
+        async with http_client() as client:
+            params = {"mid": target}
+            res = await client.get(
+                "https://api.bilibili.com/x/space/acc/info",
+                params=params,
+                timeout=4.0,
+            )
+            res_dict = json.loads(res.text)
+            if res_dict["code"] == 0:
+                info = {}
+                info["uid"] = res_dict["data"]["mid"]
+                info["uname"] = res_dict["data"]["name"]
+                info["live_state"] = res_dict["data"]["live_room"]["liveStatus"]
+                info["room_id"] = res_dict["data"]["live_room"]["roomid"]
+                info["title"] = res_dict["data"]["live_room"]["title"]
+                info["cover"] = res_dict["data"]["live_room"]["cover"]
+                return info
+            else:
+                return []
+
+    def compare_status(self, target: Target, old_status, new_status) -> list[RawPost]:
+        if (
+            new_status["live_state"] != old_status["live_state"]
+            and new_status["live_state"] == 1
+        ):
+            return [new_status]
+        else:
+            return []
+
+    async def parse(self, raw_post: RawPost) -> Post:
+        url = "https://live.bilibili.com/{}".format(raw_post["room_id"])
+        pic = [raw_post["cover"]]
+        target_name = raw_post["uname"]
+        title = raw_post["title"]
+        return Post(
+            self.name,
+            text=title,
+            url=url,
+            pics=pic,
+            target_name=target_name,
+            compress=True,
+        )

tests/platforms/static/bili_live_status.json

@@ -0,0 +1,114 @@
+{
+    "code": 0,
+    "message": "0",
+    "ttl": 1,
+    "data": {
+        "mid": 13164144,
+        "name": "魔法Zc目录",
+        "sex": "男",
+        "face": "http://i0.hdslb.com/bfs/face/a84fa10f90f7060d0336384954ee1cde7a8e9bc6.jpg",
+        "face_nft": 0,
+        "sign": "每日18:00~22:00欢乐直播！请勿在任何乌有相关内容中刷Zc，尊重角色；商务合作qq271374252",
+        "rank": 10000,
+        "level": 6,
+        "jointime": 0,
+        "moral": 0,
+        "silence": 0,
+        "coins": 0,
+        "fans_badge": true,
+        "fans_medal": {
+            "show": false,
+            "wear": false,
+            "medal": null
+        },
+        "official": {
+            "role": 1,
+            "title": "bilibili 2021百大UP主、知名游戏UP主、直播高能主播",
+            "desc": "",
+            "type": 0
+        },
+        "vip": {
+            "type": 2,
+            "status": 1,
+            "due_date": 1702051200000,
+            "vip_pay_type": 0,
+            "theme_type": 0,
+            "label": {
+                "path": "",
+                "text": "年度大会员",
+                "label_theme": "annual_vip",
+                "text_color": "#FFFFFF",
+                "bg_style": 1,
+                "bg_color": "#FB7299",
+                "border_color": ""
+            },
+            "avatar_subscript": 1,
+            "nickname_color": "#FB7299",
+            "role": 3,
+            "avatar_subscript_url": "http://i0.hdslb.com/bfs/vip/icon_Certification_big_member_22_3x.png"
+        },
+        "pendant": {
+            "pid": 3399,
+            "name": "2233幻星集",
+            "image": "http://i0.hdslb.com/bfs/garb/item/20c07ded13498a5b12db99660c766ddd92ecfe31.png",
+            "expire": 0,
+            "image_enhance": "http://i0.hdslb.com/bfs/garb/item/20c07ded13498a5b12db99660c766ddd92ecfe31.png",
+            "image_enhance_frame": ""
+        },
+        "nameplate": {
+            "nid": 1,
+            "name": "黄金殿堂",
+            "image": "http://i2.hdslb.com/bfs/face/82896ff40fcb4e7c7259cb98056975830cb55695.png",
+            "image_small": "http://i0.hdslb.com/bfs/face/627e342851dfda6fe7380c2fa0cbd7fae2e61533.png",
+            "level": "稀有勋章",
+            "condition": "单个自制视频总播放数\u003e=100万"
+        },
+        "user_honour_info": {
+            "mid": 0,
+            "colour": null,
+            "tags": []
+        },
+        "is_followed": true,
+        "top_photo": "http://i2.hdslb.com/bfs/space/853fea2728651588a2cdef0a1e586bcefff8e3d8.png",
+        "theme": {},
+        "sys_notice": {},
+        "live_room": {
+            "roomStatus": 1,
+            "liveStatus": 0,
+            "url": "https://live.bilibili.com/3044248?broadcast_type=0\u0026is_room_feed=1",
+            "title": "【Zc】早朝危机合约！",
+            "cover": "http://i0.hdslb.com/bfs/live/new_room_cover/cf7d4d3b2f336c6dba299644c3af952c5db82612.jpg",
+            "roomid": 3044248,
+            "roundStatus": 1,
+            "broadcast_type": 0,
+            "watched_show": {
+                "switch": true,
+                "num": 13753,
+                "text_small": "1.3万",
+                "text_large": "1.3万人看过",
+                "icon": "https://i0.hdslb.com/bfs/live/a725a9e61242ef44d764ac911691a7ce07f36c1d.png",
+                "icon_location": "",
+                "icon_web": "https://i0.hdslb.com/bfs/live/8d9d0f33ef8bf6f308742752d13dd0df731df19c.png"
+            }
+        },
+        "birthday": "07-21",
+        "school": {
+            "name": ""
+        },
+        "profession": {
+            "name": "",
+            "department": "",
+            "title": "",
+            "is_show": 0
+        },
+        "tags": [
+            "评论区UP主",
+            "目标是星辰大海"
+        ],
+        "series": {
+            "user_upgrade_status": 3,
+            "show_upgrade_window": false
+        },
+        "is_senior_member": 1
+    }
+}

tests/platforms/test_bilibili_live.py

@@ -0,0 +1,44 @@
+from datetime import datetime
+
+import feedparser
+import pytest
+import respx
+from httpx import Response
+from nonebug.app import App
+from pytz import timezone
+
+from .utils import get_file, get_json
+
+
+@pytest.fixture
+def bili_live(app: App):
+    from nonebot_bison.platform import platform_manager
+
+    return platform_manager["bilibili-live"]
+
+
+@pytest.mark.asyncio
+@respx.mock
+async def test_fetch_bilibili_live_status(bili_live, dummy_user_subinfo):
+    mock_bili_live_status = get_json("bili_live_status.json")
+
+    bili_live_router = respx.get(
+        "https://api.bilibili.com/x/space/acc/info?mid=13164144"
+    )
+    bili_live_router.mock(return_value=Response(200, json=mock_bili_live_status))
+    target = "13164144"
+    res = await bili_live.fetch_new_post(target, [dummy_user_subinfo])
+    assert bili_live_router.called
+    assert len(res) == 0
+    mock_bili_live_status["data"]["live_room"]["liveStatus"] = 1
+    bili_live_router.mock(return_value=Response(200, json=mock_bili_live_status))
+    res2 = await bili_live.fetch_new_post(target, [dummy_user_subinfo])
+    post = res2[0][1][0]
+    assert post.target_type == "Bilibili直播"
+    assert post.text == "【Zc】早朝危机合约！"
+    assert post.url == "https://live.bilibili.com/3044248"
+    assert post.target_name == "魔法Zc目录"
+    assert post.pics == [
+        "http://i0.hdslb.com/bfs/live/new_room_cover/cf7d4d3b2f336c6dba299644c3af952c5db82612.jpg"
+    ]
+    assert post.compress == True