butterbot-python 3.1.0.dev1__py3-none-any.whl

butterbot/sources/bilibili/data/dynamic_data.py

@@ -0,0 +1,362 @@
+from dataclasses import dataclass
+from typing import Optional
+
+from butterbot.core.data import BaseDataMixin
+
+from .dto import (
+    ArticleDto,
+    AuthorDto,
+    DynamicDTO,
+    LiveRcmdDto,
+    MusicDto,
+    StatDto,
+    VideoDto,
+)
+
+
+def get_max_id(date: dict) -> int | None:
+    """获取最新动态的索引ID
+    Args:
+        date (dict): 原始动态数据
+    Returns:
+        max_id (int): 索引id，如果数据不存在则返回None
+
+    """
+    # 数据校验：检查 items 键是否存在
+    if not date or "items" not in date:
+        return None
+
+    list_data = date["items"]
+
+    # 数据校验：检查 items 是否为空
+    if not list_data:
+        return None
+
+    max_timestamp: int = 0
+    max_id: int | None = None
+
+    for index, item in enumerate(list_data):
+        # 数据校验：检查 item 是否为字典
+        if not isinstance(item, dict):
+            continue
+
+        # 数据校验：检查嵌套的 key 是否存在
+        modules = item.get("modules")
+        if not modules or not isinstance(modules, dict):
+            continue
+
+        module_author = modules.get("module_author")
+        if not module_author or not isinstance(module_author, dict):
+            continue
+
+        pub_ts = module_author.get("pub_ts")
+        if pub_ts is None:
+            continue
+
+        try:
+            timestamp = int(pub_ts)
+            if timestamp > max_timestamp:
+                max_timestamp = timestamp
+                max_id = index
+        except (ValueError, TypeError):
+            # 如果转换失败，跳过该项
+            continue
+
+    return max_id
+
+
+@dataclass(frozen=True)
+class AuthorData(BaseDataMixin):
+    """
+    作者信息数据
+    """
+
+    uid: int  # UP主UID
+    name: str  # UP主昵称
+    face: str  # UP主头像URL
+
+    @classmethod
+    def from_dto(cls, author: AuthorDto) -> "AuthorData":
+        """从AuthorDto构造AuthorData实例"""
+        return cls(
+            uid=author.uid,
+            name=author.name,
+            face=author.face,
+        )
+
+    @property
+    def jump_url(self) -> str:
+        """作者空间跳转链接"""
+        return "https://space.bilibili.com/%s" % self.uid
+
+
+@dataclass(frozen=True)
+class StatData(BaseDataMixin):
+    """
+    动态统计信息数据
+    """
+
+    comment_count: int  # 评论数
+    like_count: int  # 点赞数
+    forward_count: int  # 转发数
+
+    @classmethod
+    def from_dto(cls, stat: StatDto) -> "StatData":
+        """从StatDto构造StatData实例"""
+        return cls(
+            comment_count=stat.comment_count,
+            like_count=stat.like_count,
+            forward_count=stat.forward_count,
+        )
+
+
+@dataclass(frozen=True)
+class VideoData(BaseDataMixin):
+    """
+    视频信息数据
+    """
+
+    av_id: str  # 视频AV号
+    bv_id: str  # 视频BV号
+    title: str  # 视频标题
+    cover: str  # 视频封面
+    desc: str  # 视频简介
+    duration_text: str  # 视频时长
+    dynamic_text: str  # 动态文本
+    play_count: str  # 播放数
+    danmaku_count: str  # 弹幕数
+
+    @classmethod
+    def from_dto(cls, video: VideoDto) -> "VideoData":
+        """从VideoDto构造VideoData实例"""
+        return cls(
+            av_id=video.av_id,
+            bv_id=video.bv_id,
+            title=video.title,
+            cover=video.cover,
+            desc=video.desc,
+            duration_text=video.duration_text,
+            dynamic_text=video.dynamic_text,
+            play_count=video.play_count,
+            danmaku_count=video.danmaku_count,
+        )
+
+    @property
+    def jump_url(self) -> str:
+        """视频跳转链接"""
+        return "https://www.bilibili.com/video/%s/" % self.bv_id
+
+
+@dataclass(frozen=True)
+class MusicData(BaseDataMixin):
+    """
+    音乐信息数据
+    """
+
+    music_id: str  # 音乐ID
+    title: str  # 音乐标题
+    cover: str  # 音乐封面
+    label: str  # 音乐标签（作者/歌手）
+    dynamic_text: str  # 动态文本
+
+    @classmethod
+    def from_dto(cls, music: MusicDto) -> "MusicData":
+        """从MusicDto构造MusicData实例"""
+        return cls(
+            music_id=music.music_id,
+            title=music.title,
+            cover=music.cover,
+            label=music.label,
+            dynamic_text=music.dynamic_text,
+        )
+
+    @property
+    def jump_url(self) -> str:
+        """音乐跳转链接"""
+        return "https://www.bilibili.com/audio/au%s" % self.music_id
+
+
+@dataclass(frozen=True)
+class ArticleData(BaseDataMixin):
+    """
+    专栏信息数据
+    """
+
+    title: str  # 专栏标题
+    summary: str  # 专栏摘要
+    has_more: bool  # 是否有更多内容
+    article_id: int  # 专栏ID
+
+    @classmethod
+    def from_dto(cls, article: ArticleDto) -> "ArticleData":
+        """从ArticleDto构造ArticleData实例"""
+        return cls(
+            title=article.title,
+            summary=article.summary,
+            has_more=article.has_more,
+            article_id=article.id,
+        )
+
+    @property
+    def jump_url(self) -> str:
+        """专栏跳转链接"""
+        return "https://www.bilibili.com/opus/%s" % self.article_id
+
+
+@dataclass(frozen=True)
+class LiveRcmdData(BaseDataMixin):
+    """
+    直播推荐信息数据
+    """
+
+    room_id: int  # 直播间ID
+    live_status: int  # 直播状态 1:直播中
+    title: str  # 直播间标题
+    cover: str  # 直播间封面
+    online: int  # 在线人数
+    area_id: int  # 直播分区ID
+    area_name: str  # 直播分区
+    parent_area_id: int  # 直播父分区ID
+    parent_area_name: str  # 直播父分区
+    live_start_time: int  # 开播时间戳
+    watched_num: int | None  # 观看人数
+    switch: bool | None  # 观看榜开关
+    text_small: str | None  # 小文本
+    text_large: str | None  # 大文本
+
+    @classmethod
+    def from_dto(cls, live_rcmd: LiveRcmdDto) -> "LiveRcmdData":
+        """从LiveRcmdDto构造LiveRcmdData实例"""
+        return cls(
+            room_id=live_rcmd.room_id,
+            live_status=live_rcmd.live_status,
+            title=live_rcmd.title,
+            cover=live_rcmd.cover,
+            online=live_rcmd.online,
+            area_id=live_rcmd.area_id,
+            area_name=live_rcmd.area_name,
+            parent_area_id=live_rcmd.parent_area_id,
+            parent_area_name=live_rcmd.parent_area_name,
+            live_start_time=live_rcmd.live_start_time,
+            watched_num=live_rcmd.watched_num,
+            switch=live_rcmd.switch,
+            text_small=live_rcmd.text_small,
+            text_large=live_rcmd.text_large,
+        )
+
+    @property
+    def jump_url(self) -> str:
+        """直播间跳转链接"""
+        return "https://live.bilibili.com/%s" % self.room_id
+
+
+@dataclass(frozen=True)
+class DynamicData(BaseDataMixin):
+    """
+    动态数据
+    """
+
+    dynamic_id: str  # 动态ID
+    dynamic_type: str  # 动态类型
+    visible: bool  # 动态显示状态(false时被折叠)
+    pub_time: str  # 发布时间
+    pub_ts: int  # 发布时间戳
+    author: AuthorData  # 作者信息
+    stat: StatData | None  # 统计信息
+    tag: str | None  # 标签（如置顶）
+    text: str | None  # 文字内容
+    pics_url: list[str] | None  # 图片列表
+    video: VideoData | None  # 视频信息
+    music: MusicData | None  # 音乐信息
+    article: ArticleData | None  # 专栏信息
+    live_rcmd: LiveRcmdData | None  # 直播推荐信息
+    forward_orig: Optional["DynamicData"]  # 转发的原动态
+
+    @classmethod
+    def from_dto(cls, dto: DynamicDTO) -> "DynamicData":
+        """从DTO对象构造DynamicData实例
+
+        Args:
+            dto: DynamicDTO对象
+
+        Returns:
+            DynamicData实例
+        """
+        # 构造作者数据
+        author_data = AuthorData.from_dto(dto.author)
+
+        # 构造统计数据
+        stat_data = StatData.from_dto(dto.stat) if dto.stat else None
+
+        # 构造视频数据
+        video_data = VideoData.from_dto(dto.video) if dto.video else None
+
+        # 构造音乐数据
+        music_data = MusicData.from_dto(dto.music) if dto.music else None
+
+        # 构造专栏数据
+        article_data = ArticleData.from_dto(dto.article) if dto.article else None
+
+        # 构造直播推荐数据
+        live_rcmd_data = LiveRcmdData.from_dto(dto.live_rcmd) if dto.live_rcmd else None
+
+        # 递归构造转发动态数据
+        forward_orig_data = cls.from_dto(dto.forward_orig) if dto.forward_orig else None
+
+        return cls(
+            dynamic_id=dto.dynamic_id,
+            dynamic_type=dto.dynamic_type,
+            visible=dto.visible,
+            pub_time=dto.pub_time,
+            pub_ts=dto.pub_ts,
+            author=author_data,
+            stat=stat_data,
+            tag=dto.tag,
+            text=dto.text,
+            pics_url=list(dto.pics_url) if dto.pics_url else None,
+            video=video_data,
+            music=music_data,
+            article=article_data,
+            live_rcmd=live_rcmd_data,
+            forward_orig=forward_orig_data,
+        )
+
+    @property
+    def jump_url(self) -> str:
+        """动态跳转链接"""
+        return "https://t.bilibili.com/%s" % self.dynamic_id
+
+    @property
+    def is_live(self) -> bool:
+        """判断动态是否为直播推荐类型"""
+        return self.dynamic_type == "DYNAMIC_TYPE_LIVE_RCMD"
+
+    @property
+    def is_video(self) -> bool:
+        """判断动态是否为视频类型"""
+        return self.dynamic_type == "DYNAMIC_TYPE_AV"
+
+    @property
+    def is_music(self) -> bool:
+        """判断动态是否为音乐类型"""
+        return self.dynamic_type == "DYNAMIC_TYPE_MUSIC"
+
+    @property
+    def is_article(self) -> bool:
+        """判断动态是否为专栏类型"""
+        return self.dynamic_type == "DYNAMIC_TYPE_ARTICLE"
+
+    @property
+    def is_word(self) -> bool:
+        """判断动态是否为纯文字类型"""
+        return self.dynamic_type == "DYNAMIC_TYPE_WORD"
+
+    @property
+    def is_draw(self) -> bool:
+        """判断动态是否为图文类型"""
+        return self.dynamic_type == "DYNAMIC_TYPE_DRAW"
+
+    @property
+    def is_forward(self) -> bool:
+        """判断动态是否为转发类型"""
+        return self.dynamic_type == "DYNAMIC_TYPE_FORWARD"

butterbot/sources/bilibili/data/live_room_data.py

@@ -0,0 +1,162 @@
+from dataclasses import dataclass
+
+from butterbot.core.data import BaseDataMixin
+
+from .dto import (
+    AnchorInfoDto,
+    LiveRoomDTO,
+    NoticeBoardDto,
+    RoomInfoDto,
+    WatchedShowDto,
+)
+
+
+@dataclass(frozen=True)
+class RoomInfoData(BaseDataMixin):
+    """直播间信息数据"""
+
+    uid: int  # 用户uid
+    room_id: int  # 房间号
+    title: str  # 直播间标题
+    cover_url: str  # 直播间封面url
+    background_url: str  # 直播间背景图url
+    description: str  # 主播简介
+    tags: list[str]  # 直播间标签列表
+    live_status: int  # 直播状态 0：未开播 1：直播中 2：轮播中
+    live_start_time: int  # 直播开始时间戳
+    parent_area_name: str  # 直播间父分区
+    parent_area_id: int  # 直播间父分区ID
+    area_name: str  # 直播间子分区
+    area_id: int  # 直播间子分区ID
+    keyframe_url: str  # 直播间关键帧url
+    online: int  # 直播间当前在线人数
+
+    @classmethod
+    def from_dto(cls, room_info: RoomInfoDto) -> "RoomInfoData":
+        """从RoomInfoDto构造RoomInfoData实例"""
+        return cls(
+            uid=room_info.uid,
+            room_id=room_info.room_id,
+            title=room_info.title,
+            cover_url=room_info.cover_url,
+            background_url=room_info.background_url,
+            description=room_info.description,
+            tags=list(room_info.tags),
+            live_status=room_info.live_status,
+            live_start_time=room_info.live_start_time,
+            parent_area_name=room_info.parent_area_name,
+            parent_area_id=room_info.parent_area_id,
+            area_name=room_info.area_name,
+            area_id=room_info.area_id,
+            keyframe_url=room_info.keyframe_url,
+            online=room_info.online,
+        )
+
+    @property
+    def jump_url(self) -> str:
+        """直播间跳转链接"""
+        return "https://live.bilibili.com/%s" % self.room_id
+
+
+@dataclass(frozen=True)
+class AnchorInfoData(BaseDataMixin):
+    """主播信息数据"""
+
+    name: str  # 主播昵称
+    face_url: str  # 主播头像url
+    gender: str  # 主播性别
+    official_info: str  # 主播官方信息(认证信息)
+    fanclub_name: str  # 粉丝牌名称
+    fanclub_num: int  # 粉丝团人数
+    live_level: int  # 主播等级
+    live_score: int  # 直播分数
+    live_upgrade_score: int  # 升级所需分数
+
+    @classmethod
+    def from_dto(cls, anchor_info: AnchorInfoDto) -> "AnchorInfoData":
+        """从AnchorInfoDto构造AnchorInfoData实例"""
+        return cls(
+            name=anchor_info.name,
+            face_url=anchor_info.face_url,
+            gender=anchor_info.gender,
+            official_info=anchor_info.official_info,
+            fanclub_name=anchor_info.fanclub_name,
+            fanclub_num=anchor_info.fanclub_num,
+            live_level=anchor_info.live_level,
+            live_score=anchor_info.live_score,
+            live_upgrade_score=anchor_info.live_upgrade_score,
+        )
+
+
+@dataclass(frozen=True)
+class WatchedShowData(BaseDataMixin):
+    """观看榜信息数据"""
+
+    switch: bool  # 观看榜开关
+    num: int  # 观看人数/人气值
+    text_small: str  # 小文本
+    text_large: str  # 大文本
+
+    @classmethod
+    def from_dto(cls, watched_show: WatchedShowDto) -> "WatchedShowData":
+        """从WatchedShowDto构造WatchedShowData实例"""
+        return cls(
+            switch=watched_show.switch,
+            num=watched_show.num,
+            text_small=watched_show.text_small,
+            text_large=watched_show.text_large,
+        )
+
+
+@dataclass(frozen=True)
+class NoticeBoardData(BaseDataMixin):
+    """公告栏信息数据"""
+
+    content: str  # 公告内容
+    ctime: str  # 公告发布时间
+
+    @classmethod
+    def from_dto(cls, notice_board: NoticeBoardDto) -> "NoticeBoardData":
+        """从NoticeBoardDto构造NoticeBoardData实例"""
+        return cls(content=notice_board.content, ctime=notice_board.ctime)
+
+
+@dataclass(frozen=True)
+class LiveRoomData(BaseDataMixin):
+    """直播间数据"""
+
+    room_info: RoomInfoData  # 直播间信息
+    anchor_info: AnchorInfoData  # 主播信息
+    watched_show: WatchedShowData  # 观看榜信息
+    notice_board: NoticeBoardData | None  # 公告栏信息
+
+    @classmethod
+    def from_dto(cls, dto: LiveRoomDTO) -> "LiveRoomData":
+        """从DTO对象构造LiveRoomData实例
+
+        Args:
+            dto: LiveRoomDTO对象
+
+        Returns:
+            LiveRoomData实例
+        """
+        # 构造直播间信息数据
+        room_info_data = RoomInfoData.from_dto(dto.room_info)
+
+        # 构造主播信息数据
+        anchor_info_data = AnchorInfoData.from_dto(dto.anchor_info)
+
+        # 构造观看榜信息数据
+        watched_show_data = WatchedShowData.from_dto(dto.watched_show)
+
+        # 构造公告栏信息数据
+        notice_board_data = (
+            NoticeBoardData.from_dto(dto.notice_board) if dto.notice_board else None
+        )
+
+        return cls(
+            room_info=room_info_data,
+            anchor_info=anchor_info_data,
+            watched_show=watched_show_data,
+            notice_board=notice_board_data,
+        )

butterbot/sources/bilibili/data/video_part.py

@@ -0,0 +1,46 @@
+from dataclasses import dataclass
+
+from butterbot.core.data import BaseDataMixin
+
+from .dto import (
+    VideoPartDto,
+)
+
+
+@dataclass(frozen=True)
+class VideoPartData(BaseDataMixin):
+    """
+    视频字幕数据
+    """
+
+    start_timestamp: int  # 开始时间戳
+    end_timestamp: int  # 结束时间戳
+    content: str  # 字幕内容
+
+    @classmethod
+    def from_dto_list(cls, dto_list: list[VideoPartDto]) -> list["VideoPartData"]:
+        """从DTO对象列表构造VideoPartData实例列表
+
+        Args:
+            dto_list: VideoPartDto对象列表
+
+        Returns:
+            VideoPartData实例列表
+        """
+        return [cls.from_dto(dto) for dto in dto_list]
+
+    @classmethod
+    def from_dto(cls, dto: VideoPartDto) -> "VideoPartData":
+        """从DTO对象构造VideoPartData实例
+
+        Args:
+            dto: VideoPartDto对象
+
+        Returns:
+            VideoPartData实例
+        """
+        return cls(
+            start_timestamp=dto.start_timestamp,
+            end_timestamp=dto.end_timestamp,
+            content=dto.content,
+        )

butterbot/sources/bilibili/source/__init__.py

@@ -0,0 +1,15 @@
+from .bili_danmaku_source import (
+    BiliDanmakuSource,
+)
+from .bili_dynamic_source import (
+    BiliDynamicSource,
+)
+from .bili_live_source import (
+    BiliLiveSource,
+)
+
+__all__ = [
+    "BiliDanmakuSource",
+    "BiliDynamicSource",
+    "BiliLiveSource",
+]