feat: 支持bilibili视频下载

store/bilibili/__init__.py

@@ -8,7 +8,7 @@ from typing import List
 import config
 
 from .bilibili_store_impl import *
-
+from .bilibilli_store_video import *
 
 class BiliStoreFactory:
     STORES = {
@@ -80,3 +80,15 @@ async def update_bilibili_video_comment(video_id: str, comment_item: Dict):
     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)
+
+
+async def store_video(aid, video_content, extension_file_name):
+    """
+    video video storage implementation
+    Args:
+        aid:
+        video_content:
+        extension_file_name:
+    """
+    await BilibiliVideo().store_video(
+        {"aid": aid, "video_content": video_content, "extension_file_name": extension_file_name})

store/bilibili/bilibilli_store_video.py

@@ -0,0 +1,53 @@
+# -*- coding: utf-8 -*-
+# @Author  : helloteemo
+# @Time    : 2024/7/12 20:01
+# @Desc    : bilibili图片保存
+import pathlib
+from typing import Dict
+
+import aiofiles
+
+from base.base_crawler import AbstractStoreImage
+from tools import utils
+
+
+class BilibiliVideo(AbstractStoreImage):
+    video_store_path: str = "data/bilibili/videos"
+
+    async def store_video(self, video_content_item: Dict):
+        """
+        store content
+        Args:
+            content_item:
+
+        Returns:
+
+        """
+        await self.save_video(video_content_item.get("aid"), video_content_item.get("video_content"),
+                              video_content_item.get("extension_file_name"))
+
+    def make_save_file_name(self, aid: str, extension_file_name: str) -> str:
+        """
+        make save file name by store type
+        Args:
+            aid: aid
+        Returns:
+
+        """
+        return f"{self.video_store_path}/{aid}/{extension_file_name}"
+
+    async def save_video(self, aid: int, video_content: str, extension_file_name="mp4"):
+        """
+        save video to local
+        Args:
+            aid: aid
+            video_content: video content
+
+        Returns:
+
+        """
+        pathlib.Path(self.video_store_path + "/" + str(aid)).mkdir(parents=True, exist_ok=True)
+        save_file_name = self.make_save_file_name(str(aid), extension_file_name)
+        async with aiofiles.open(save_file_name, 'wb') as f:
+            await f.write(video_content)
+            utils.logger.info(f"[BilibiliVideoImplement.save_video] save save_video {save_file_name} success ...")