feat: 支持弹幕下载

amtoaer · Dec 5, 2023 · 1686c1a · 1686c1a
1 parent de6eaeb
commit 1686c1a
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 16 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/processor.py b/processor.py
@@ -3,7 +3,7 @@
 from asyncio import Semaphore, create_subprocess_exec
 from asyncio.subprocess import DEVNULL
 
-from bilibili_api import favorite_list, video
+from bilibili_api import ass, favorite_list, video
 from bilibili_api.exceptions import ResponseCodeException
 from loguru import logger
 from tortoise import Tortoise
@@ -236,18 +236,16 @@ async def process_favorite_item(
                     fav_item.name,
                 )
         if process_subtitle:
-            pass
-            # # 写入字幕，上游库获取字幕有 bug，暂时不做实现
-            # if not await aexists(fav_item.subtitle_path):
-            #     await ass.make_ass_file_danmakus_protobuf(
-            #         v, 0, str(fav_item.subtitle_path.resolve())
-            #     )
-            # else:
-            #     logger.info(
-            #         "Subtitle of {} {} already exists, skipped.",
-            #         fav_item.bvid,
-            #         fav_item.name,
-            #     )
+            if not await aexists(fav_item.subtitle_path):
+                await ass.make_ass_file_danmakus_protobuf(
+                    v, 0, str(fav_item.subtitle_path.resolve())
+                )
+            else:
+                logger.info(
+                    "Subtitle of {} {} already exists, skipped.",
+                    fav_item.bvid,
+                    fav_item.name,
+                )
         if process_video:
             if await aexists(fav_item.video_path):
                 fav_item.downloaded = True