mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 06:12:53 +08:00
250 lines
9.6 KiB
Python
250 lines
9.6 KiB
Python
|
from services.log import logger
|
|||
|
|
from services.db_context import db
|
|||
|
|
from datetime import datetime
|
|||
|
|
from typing import Optional, List
|
|||
|
|
|
|||
|
|
|
|||
|
|
class BilibiliSub(db.Model):
|
|||
|
|
__tablename__ = "bilibili_sub"
|
|||
|
|
|
|||
|
|
id = db.Column(db.Integer(), primary_key=True)
|
|||
|
|
sub_id = db.Column(db.Integer(), nullable=False)
|
|||
|
|
sub_type = db.Column(db.String(), nullable=False)
|
|||
|
|
# 订阅用户
|
|||
|
|
sub_users = db.Column(db.String(), nullable=False)
|
|||
|
|
# 直播
|
|||
|
|
live_short_id = db.Column(db.Integer())
|
|||
|
|
live_status = db.Column(db.Integer)
|
|||
|
|
# 主播/UP
|
|||
|
|
uid = db.Column(db.BigInteger())
|
|||
|
|
uname = db.Column(db.String())
|
|||
|
|
latest_video_created = db.Column(db.BigInteger()) # 视频上传时间
|
|||
|
|
dynamic_upload_time = db.Column(db.BigInteger(), default=0) # 动态发布时间
|
|||
|
|
# 番剧
|
|||
|
|
season_name = db.Column(db.String())
|
|||
|
|
season_id = db.Column(db.Integer())
|
|||
|
|
season_current_episode = db.Column(db.String())
|
|||
|
|
season_update_time = db.Column(db.DateTime())
|
|||
|
|
|
|||
|
|
_idx1 = db.Index("bilibili_sub_idx1", "sub_id", "sub_type", unique=True)
|
|||
|
|
|
|||
|
|
@classmethod
|
|||
|
|
async def add_bilibili_sub(
|
|||
|
|
cls,
|
|||
|
|
sub_id: int,
|
|||
|
|
sub_type: str,
|
|||
|
|
sub_user: str,
|
|||
|
|
*,
|
|||
|
|
live_short_id: Optional[int] = None,
|
|||
|
|
live_status: Optional[int] = None,
|
|||
|
|
dynamic_upload_time: Optional[int] = None,
|
|||
|
|
uid: Optional[int] = None,
|
|||
|
|
uname: Optional[str] = None,
|
|||
|
|
latest_video_created: Optional[int] = None,
|
|||
|
|
season_name: Optional[str] = None,
|
|||
|
|
season_id: Optional[int] = None,
|
|||
|
|
season_current_episode: Optional[str] = None,
|
|||
|
|
season_update_time: Optional[datetime] = None,
|
|||
|
|
) -> bool:
|
|||
|
|
"""
|
|||
|
|
说明：
|
|||
|
|
添加订阅
|
|||
|
|
参数：
|
|||
|
|
:param sub_id: 订阅名称，房间号，番剧号等
|
|||
|
|
:param sub_type: 订阅类型
|
|||
|
|
:param sub_user: 订阅此条目的用户
|
|||
|
|
:param live_short_id: 直接短 id
|
|||
|
|
:param live_status: 主播开播状态
|
|||
|
|
:param dynamic_upload_time: 主播/UP最新动态时间
|
|||
|
|
:param uid: 主播/UP uid
|
|||
|
|
:param uname: 用户名称
|
|||
|
|
:param latest_video_created: 最新视频上传时间
|
|||
|
|
:param season_name: 番剧名称
|
|||
|
|
:param season_id: 番剧 season_id
|
|||
|
|
:param season_current_episode: 番剧最新集数
|
|||
|
|
:param season_update_time: 番剧更新时间
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
async with db.transaction():
|
|||
|
|
query = (
|
|||
|
|
await cls.query.where(cls.sub_id == sub_id)
|
|||
|
|
.with_for_update()
|
|||
|
|
.gino.first()
|
|||
|
|
)
|
|||
|
|
sub_user = sub_user if sub_user[-1] == "," else f"{sub_user},"
|
|||
|
|
if query:
|
|||
|
|
if sub_user not in query.sub_users:
|
|||
|
|
sub_users = query.sub_users + sub_user
|
|||
|
|
await query.update(sub_users=sub_users).apply()
|
|||
|
|
else:
|
|||
|
|
sub = await cls.create(
|
|||
|
|
sub_id=sub_id, sub_type=sub_type, sub_users=sub_user
|
|||
|
|
)
|
|||
|
|
await sub.update(
|
|||
|
|
live_short_id=live_short_id
|
|||
|
|
if live_short_id
|
|||
|
|
else sub.live_short_id,
|
|||
|
|
live_status=live_status if live_status else sub.live_status,
|
|||
|
|
dynamic_upload_time=dynamic_upload_time
|
|||
|
|
if dynamic_upload_time
|
|||
|
|
else sub.dynamic_upload_time,
|
|||
|
|
uid=uid if uid else sub.uid,
|
|||
|
|
uname=uname if uname else sub.uname,
|
|||
|
|
latest_video_created=latest_video_created
|
|||
|
|
if latest_video_created
|
|||
|
|
else sub.latest_video_created,
|
|||
|
|
season_update_time=season_update_time
|
|||
|
|
if season_update_time
|
|||
|
|
else sub.season_update_time,
|
|||
|
|
season_current_episode=season_current_episode
|
|||
|
|
if season_current_episode
|
|||
|
|
else sub.season_current_episode,
|
|||
|
|
season_id=season_id if season_id else sub.season_id,
|
|||
|
|
season_name=season_name if season_name else sub.season_name,
|
|||
|
|
).apply()
|
|||
|
|
return True
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.info(f"bilibili_sub 添加订阅错误 {type(e)}: {e}")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
@classmethod
|
|||
|
|
async def delete_bilibili_sub(cls, sub_id: int, sub_user: str) -> bool:
|
|||
|
|
"""
|
|||
|
|
说明：
|
|||
|
|
删除订阅
|
|||
|
|
参数：
|
|||
|
|
:param sub_id: 订阅名称
|
|||
|
|
:param sub_user: 删除此条目的用户
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
async with db.transaction():
|
|||
|
|
query = (
|
|||
|
|
await cls.query.where(
|
|||
|
|
(cls.sub_id == sub_id) & (cls.sub_users.contains(sub_user))
|
|||
|
|
)
|
|||
|
|
.with_for_update()
|
|||
|
|
.gino.first()
|
|||
|
|
)
|
|||
|
|
if not query:
|
|||
|
|
return False
|
|||
|
|
await query.update(
|
|||
|
|
sub_users=query.sub_users.replace(f"{sub_user},", "")
|
|||
|
|
).apply()
|
|||
|
|
if not query.sub_users.strip():
|
|||
|
|
await query.delete()
|
|||
|
|
return True
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.info(f"bilibili_sub 删除订阅错误 {type(e)}: {e}")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
@classmethod
|
|||
|
|
async def get_sub(cls, sub_id: int) -> Optional["BilibiliSub"]:
|
|||
|
|
"""
|
|||
|
|
说明：
|
|||
|
|
获取订阅对象
|
|||
|
|
参数：
|
|||
|
|
:param sub_id: 订阅 id
|
|||
|
|
"""
|
|||
|
|
return await cls.query.where(cls.sub_id == sub_id).gino.first()
|
|||
|
|
|
|||
|
|
@classmethod
|
|||
|
|
async def get_sub_data(cls, id_: str) -> List["BilibiliSub"]:
|
|||
|
|
"""
|
|||
|
|
获取 id_ 订阅的所有内容
|
|||
|
|
:param id_: id
|
|||
|
|
"""
|
|||
|
|
query = cls.query.where(cls.sub_users.contains(id_))
|
|||
|
|
return await query.gino.all()
|
|||
|
|
|
|||
|
|
@classmethod
|
|||
|
|
async def update_sub_info(
|
|||
|
|
cls,
|
|||
|
|
sub_id: int,
|
|||
|
|
*,
|
|||
|
|
live_short_id: Optional[int] = None,
|
|||
|
|
live_status: Optional[int] = None,
|
|||
|
|
dynamic_upload_time: Optional[int] = None,
|
|||
|
|
uid: Optional[int] = None,
|
|||
|
|
uname: Optional[str] = None,
|
|||
|
|
latest_video_created: Optional[int] = None,
|
|||
|
|
season_name: Optional[str] = None,
|
|||
|
|
season_id: Optional[int] = None,
|
|||
|
|
season_current_episode: Optional[str] = None,
|
|||
|
|
season_update_time: Optional[datetime] = None,
|
|||
|
|
) -> bool:
|
|||
|
|
"""
|
|||
|
|
说明：
|
|||
|
|
更新订阅信息
|
|||
|
|
参数：
|
|||
|
|
:param sub_id: 订阅名称，房间号，番剧号等
|
|||
|
|
:param live_short_id: 直接短 id
|
|||
|
|
:param live_status: 主播开播状态
|
|||
|
|
:param dynamic_upload_time: 主播/UP最新动态时间
|
|||
|
|
:param uid: 主播/UP uid
|
|||
|
|
:param uname: 用户名称
|
|||
|
|
:param latest_video_created: 最新视频上传时间
|
|||
|
|
:param season_name: 番剧名称
|
|||
|
|
:param season_id: 番剧 season_id
|
|||
|
|
:param season_current_episode: 番剧最新集数
|
|||
|
|
:param season_update_time: 番剧更新时间
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
async with db.transaction():
|
|||
|
|
sub = (
|
|||
|
|
await cls.query.where(cls.sub_id == sub_id)
|
|||
|
|
.with_for_update()
|
|||
|
|
.gino.first()
|
|||
|
|
)
|
|||
|
|
if sub:
|
|||
|
|
await sub.update(
|
|||
|
|
live_short_id=live_short_id
|
|||
|
|
if live_short_id is not None
|
|||
|
|
else sub.live_short_id,
|
|||
|
|
live_status=live_status
|
|||
|
|
if live_status is not None
|
|||
|
|
else sub.live_status,
|
|||
|
|
dynamic_upload_time=dynamic_upload_time
|
|||
|
|
if dynamic_upload_time is not None
|
|||
|
|
else sub.dynamic_upload_time,
|
|||
|
|
uid=uid if uid is not None else sub.uid,
|
|||
|
|
uname=uname if uname is not None else sub.uname,
|
|||
|
|
latest_video_created=latest_video_created
|
|||
|
|
if latest_video_created is not None
|
|||
|
|
else sub.latest_video_created,
|
|||
|
|
season_update_time=season_update_time
|
|||
|
|
if season_update_time is not None
|
|||
|
|
else sub.season_update_time,
|
|||
|
|
season_current_episode=season_current_episode
|
|||
|
|
if season_current_episode is not None
|
|||
|
|
else sub.season_current_episode,
|
|||
|
|
season_id=season_id if season_id is not None else sub.season_id,
|
|||
|
|
season_name=season_name
|
|||
|
|
if season_name is not None
|
|||
|
|
else sub.season_name,
|
|||
|
|
).apply()
|
|||
|
|
return True
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.info(f"bilibili_sub 更新订阅错误 {type(e)}: {e}")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
@classmethod
|
|||
|
|
async def get_all_sub_data(
|
|||
|
|
cls,
|
|||
|
|
) -> "List[BilibiliSub], List[BilibiliSub], List[BilibiliSub]":
|
|||
|
|
"""
|
|||
|
|
说明：
|
|||
|
|
分类获取所有数据
|
|||
|
|
"""
|
|||
|
|
live_data = []
|
|||
|
|
up_data = []
|
|||
|
|
season_data = []
|
|||
|
|
query = await cls.query.gino.all()
|
|||
|
|
for x in query:
|
|||
|
|
if x.sub_type == "live":
|
|||
|
|
live_data.append(x)
|
|||
|
|
if x.sub_type == "up":
|
|||
|
|
up_data.append(x)
|
|||
|
|
if x.sub_type == "season":
|
|||
|
|
season_data.append(x)
|
|||
|
|
return live_data, up_data, season_data
|