From b8b769e03fc87e2647e8504845d67e6e02efedd2 Mon Sep 17 00:00:00 2001
From: Ovler <ovlertheyoung@gmail.com>
Date: Sat, 15 Jun 2024 15:28:26 +0000
Subject: [PATCH] fix: season and series are different

---
 biliarchiver/cli_tools/get_command.py | 33 ++++++++++++++++++++++++++-
 biliarchiver/rest_api/main.py         |  3 ++-
 2 files changed, 34 insertions(+), 2 deletions(-)

diff --git a/biliarchiver/cli_tools/get_command.py b/biliarchiver/cli_tools/get_command.py
index 3c80383..71bfda3 100644
--- a/biliarchiver/cli_tools/get_command.py
+++ b/biliarchiver/cli_tools/get_command.py
@@ -27,7 +27,38 @@ async def by_series(url_or_sid: str, truncate: int = int(1e10)) -> Path:
     client = AsyncClient(**api.dft_client_settings)
     print(_("正在获取 {sid} 的视频列表……").format(sid=sid))
     col_name, up_name, bvids = await api.get_list_info(client, sid)
-    filepath = f"bvids/by-sapce_fav_season/sid-{sid}-{int(time.time())}.txt"
+    filepath = f"bvids/by-sapce_fav_season/seriesid-{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)} 个视频")
+    count = len(bvids)
+    print(
+        ngettext(
+            "已获取 {}（{}）的一个视频",
+            "已获取 {}（{}）的 {count} 个视频",
+            count,
+        ).format(col_name, up_name, count=count)
+    )
+    print(_("存储到 {}").format(abs_filepath))
+    return Path(abs_filepath)
+
+
+async def by_season(url_or_sid: str, truncate: int = int(1e10)) -> Path:
+    """
+    truncate: do noting
+    """
+    sid = sid = (
+        re.search(r"sid=(\d+)", url_or_sid).groups()[0]
+        if url_or_sid.startswith("http")
+        else url_or_sid
+    )  # type: ignore
+    client = AsyncClient(**api.dft_client_settings)
+    print(_("正在获取 {sid} 的视频合集……").format(sid=sid))
+    col_name, up_name, bvids = await api.get_collect_info(client, sid)
+    filepath = f"bvids/by-sapce_fav_season/seasonid-{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:
diff --git a/biliarchiver/rest_api/main.py b/biliarchiver/rest_api/main.py
index 7fa0b37..974977b 100644
--- a/biliarchiver/rest_api/main.py
+++ b/biliarchiver/rest_api/main.py
@@ -15,7 +15,7 @@ except ImportError:
     raise
 
 
-from biliarchiver.cli_tools.get_command import by_favlist, by_series, by_up_videos
+from biliarchiver.cli_tools.get_command import by_favlist, by_series, by_up_videos, by_season
 from biliarchiver.rest_api.bilivid import BiliVideo, VideoStatus
 from biliarchiver.version import BILI_ARCHIVER_VERSION
 
@@ -133,6 +133,7 @@ async def perform_source_action_from_req(source_type: str, source_id: str):
         "up_videos": by_up_videos,
         "favlist": by_favlist,
         "series": by_series,
+        "season": by_season,
     }
 
     if source_type not in fun_mapping: