From 3e0bd7b10aa31870dae07787591a911f4d78cc5c Mon Sep 17 00:00:00 2001
From: yzqzss <yzqzss@yandex.com>
Date: Fri, 4 Aug 2023 19:17:19 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E8=8E=B7=E5=8F=96=E5=90=88=E9=9B=86?=
 =?UTF-8?q?=E6=88=96=E8=A7=86=E9=A2=91=E5=88=97=E8=A1=A8?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 biliarchiver/cli_tools/bili_get_bvids.py | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)

diff --git a/biliarchiver/cli_tools/bili_get_bvids.py b/biliarchiver/cli_tools/bili_get_bvids.py
index 0d08d1a..db469ba 100644
--- a/biliarchiver/cli_tools/bili_get_bvids.py
+++ b/biliarchiver/cli_tools/bili_get_bvids.py
@@ -39,9 +39,31 @@ def arg_parse():
     popular_series_group.add_argument('--popular_series-number', type=int, default=1, help='获取第几期（周） [default: 1]')
     popular_series_group.add_argument('--all-popular_series', action='store_true', help='自动获取全部的每周必看（增量）', dest='all_popular_series')
 
+    space_fav_season = parser.add_argument_group()
+    space_fav_season.title = 'space_fav_season'
+    space_fav_season.description = '获取合集或视频列表内视频'
+    space_fav_season.add_argument('--by-space_fav_season', type=str, help='合集或视频列表 sid (或 URL)', dest='by_space_fav_season', default=None)
+
     args = parser.parse_args()
     return args
 
+
+async def by_sapce_fav_season(url_or_sid: str) -> Path:
+    sid = sid = re.search(r'sid=(\d+)', url_or_sid).groups()[0] if url_or_sid.startswith('http') else url_or_sid
+    client = AsyncClient(**api.dft_client_settings)
+    print(f'正在获取 {sid} 的视频列表……')
+    col_name, up_name, bvids = await api.get_collect_info(client, sid)
+    filepath = f'bvids/by-sapce_fav_season/sid-{sid}-{int(time.time())}.txt'
+    os.makedirs(os.path.dirname(filepath), exist_ok=True)
+    abs_filepath = os.path.abspath(filepath)
+    with open(abs_filepath, 'w', encoding='utf-8') as f:
+        for bv_id in bvids:
+            f.write(f'{bv_id}' + '\n')
+    print(f'已获取 {col_name}（{up_name}）的 {len(bvids)} 个视频')
+    print(f'到 {abs_filepath}')
+    return Path(abs_filepath)
+
+
 def by_raning(rid: int) -> Path:
     bilibili_ranking_api = "https://api.bilibili.com/x/web-interface/ranking/v2"
     bilibili_ranking_params = {
@@ -190,6 +212,8 @@ async def _main():
                 by_popular_series_one(number)
         else:
             by_popular_series_one(args.popular_series_number)
+    if args.by_space_fav_season:
+        await by_sapce_fav_season(args.by_space_fav_season)
         
 
 def main():