83 lines
3.2 KiB
Python
83 lines
3.2 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
# @Author : relakkes@gmail.com
|
||
|
|
# @Time : 2024/1/14 19:34
|
||
|
|
# @Desc :
|
||
|
|
|
||
|
|
from typing import List
|
||
|
|
|
||
|
|
import config
|
||
|
|
|
||
|
|
from .bilibili_store_impl import *
|
||
|
|
|
||
|
|
|
||
|
|
class BiliStoreFactory:
|
||
|
|
STORES = {
|
||
|
|
"csv": BiliCsvStoreImplement,
|
||
|
|
"db": BiliDbStoreImplement,
|
||
|
|
"json": BiliJsonStoreImplement
|
||
|
|
}
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def create_store() -> AbstractStore:
|
||
|
|
store_class = BiliStoreFactory.STORES.get(config.SAVE_DATA_OPTION)
|
||
|
|
if not store_class:
|
||
|
|
raise ValueError(
|
||
|
|
"[BiliStoreFactory.create_store] Invalid save option only supported csv or db or json ...")
|
||
|
|
return store_class()
|
||
|
|
|
||
|
|
|
||
|
|
async def update_bilibili_video(video_item: Dict):
|
||
|
|
video_item_view: Dict = video_item.get("View")
|
||
|
|
video_user_info: Dict = video_item_view.get("owner")
|
||
|
|
video_item_stat: Dict = video_item_view.get("stat")
|
||
|
|
video_id = str(video_item_view.get("aid"))
|
||
|
|
save_content_item = {
|
||
|
|
"video_id": video_id,
|
||
|
|
"video_type": "video",
|
||
|
|
"title": video_item_view.get("title", "")[:500],
|
||
|
|
"desc": video_item_view.get("desc", "")[:500],
|
||
|
|
"create_time": video_item_view.get("pubdate"),
|
||
|
|
"user_id": str(video_user_info.get("mid")),
|
||
|
|
"nickname": video_user_info.get("name"),
|
||
|
|
"avatar": video_user_info.get("face", ""),
|
||
|
|
"liked_count": str(video_item_stat.get("like", "")),
|
||
|
|
"video_play_count": str(video_item_stat.get("view", "")),
|
||
|
|
"video_danmaku": str(video_item_stat.get("danmaku", "")),
|
||
|
|
"video_comment": str(video_item_stat.get("reply", "")),
|
||
|
|
"last_modify_ts": utils.get_current_timestamp(),
|
||
|
|
"video_url": f"https://www.bilibili.com/video/av{video_id}",
|
||
|
|
"video_cover_url": video_item_view.get("pic", ""),
|
||
|
|
}
|
||
|
|
utils.logger.info(
|
||
|
|
f"[store.bilibili.update_bilibili_video] bilibili video id:{video_id}, title:{save_content_item.get('title')}")
|
||
|
|
await BiliStoreFactory.create_store().store_content(content_item=save_content_item)
|
||
|
|
|
||
|
|
|
||
|
|
async def batch_update_bilibili_video_comments(video_id: str, comments: List[Dict]):
|
||
|
|
if not comments:
|
||
|
|
return
|
||
|
|
for comment_item in comments:
|
||
|
|
await update_bilibili_video_comment(video_id, comment_item)
|
||
|
|
|
||
|
|
|
||
|
|
async def update_bilibili_video_comment(video_id: str, comment_item: Dict):
|
||
|
|
comment_id = str(comment_item.get("rpid"))
|
||
|
|
parent_comment_id = str(comment_item.get("parent", 0))
|
||
|
|
content: Dict = comment_item.get("content")
|
||
|
|
user_info: Dict = comment_item.get("member")
|
||
|
|
save_comment_item = {
|
||
|
|
"comment_id": comment_id,
|
||
|
|
"parent_comment_id": parent_comment_id,
|
||
|
|
"create_time": comment_item.get("ctime"),
|
||
|
|
"video_id": str(video_id),
|
||
|
|
"content": content.get("message"),
|
||
|
|
"user_id": user_info.get("mid"),
|
||
|
|
"nickname": user_info.get("uname"),
|
||
|
|
"avatar": user_info.get("avatar"),
|
||
|
|
"sub_comment_count": str(comment_item.get("rcount", 0)),
|
||
|
|
"last_modify_ts": utils.get_current_timestamp(),
|
||
|
|
}
|
||
|
|
utils.logger.info(
|
||
|
|
f"[store.bilibili.update_bilibili_video_comment] Bilibili video comment: {comment_id}, content: {save_comment_item.get('content')}")
|
||
|
|
await BiliStoreFactory.create_store().store_comment(comment_item=save_comment_item)
|