add torrent manage

f

f

f

Author: shinonomeow

backend/src/module/api/__init__.py

@@ -8,6 +8,7 @@
 from .program import router as program_router
 from .rss import router as rss_router
 from .search import router as search_router
+from .torrent import router as torrent_router
 
 __all__ = ["v1","lifespan"]
 
@@ -20,3 +21,4 @@
 v1.include_router(config_router)
 v1.include_router(rss_router)
 v1.include_router(search_router)
+v1.include_router(torrent_router)

backend/src/module/api/bangumi.py

@@ -6,7 +6,7 @@
 
 from module.database import Database, engine
 from module.manager import BangumiManager
-from module.models import APIResponse, Bangumi, BangumiUpdate, ResponseModel
+from module.models import APIResponse, Bangumi, BangumiUpdate, ResponseModel,Torrent
 from module.security.api import get_current_user
 
 from .response import u_response
@@ -217,12 +217,37 @@ async def refresh_single_poster(bangumi_id: int):
 async def reset_all():
     with Database(engine) as db:
         db.bangumi.delete_all()
-        return JSONResponse(
-            status_code=200,
-            content={
-                "msg_en": "Reset all rules successfully.",
-                "msg_zh": "重置所有规则成功。",
-            },
-        )
+    return JSONResponse(
+        status_code=200,
+        content={
+            "msg_en": "Reset all rules successfully.",
+            "msg_zh": "重置所有规则成功。",
+        },
+    )
+
+
+# @router.get("/get_torrent", response_model=list[Torrent], dependencies=[Depends(get_current_user)])
+# async def manage_bangumi(_id:int):
+#     """
+#     管理对应 Bangumi 的规则
+#     """
+#     try:
+#         resp = await BangumiManager().fetch_all_bangumi_torrents(_id)
+#         return resp
+#     except Exception as e:
+#         logger.error(f"[Bangumi] Error managing bangumi: {e}")
+#         return []
+#
+# @router.get("/download_torrent", response_model=APIResponse, dependencies=[Depends(get_current_user)])
+# async def download_bangumi(_id:int, torrent:Torrent):
+#     """
+#     手动下载对应 Bangumi 的种子
+#     """
+#     try:
+#         resp = BangumiManager().download_torrent(_id,torrent)
+#         return resp
+#     except Exception as e:
+#         logger.error(f"[Bangumi] Error downloading bangumi: {e}")
+#         return []
 
 

backend/src/module/api/torrent.py

@@ -0,0 +1,104 @@
+import logging
+
+from fastapi import APIRouter, Depends, HTTPException
+from fastapi.responses import JSONResponse
+from sqlalchemy.util.concurrency import asyncio
+
+from module.database import Database, engine
+from module.manager import BangumiManager, TorrentManager
+from module.models import APIResponse, Bangumi, BangumiUpdate, ResponseModel, Torrent
+from module.security.api import get_current_user
+
+from .response import u_response
+
+router = APIRouter(prefix="/torrent", tags=["torrent"])
+logger = logging.getLogger(__name__)
+
+
+@router.get(
+    "/get_all",
+    response_model=list[Torrent],
+    dependencies=[Depends(get_current_user)],
+)
+async def manage_bangumi(_id: int):
+    """
+    管理对应 Bangumi 的规则
+    """
+    try:
+        resp = await TorrentManager().fetch_all_bangumi_torrents(_id)
+        return resp
+    except Exception as e:
+        logger.error(f"[Bangumi] Error managing bangumi: {e}")
+        return []
+
+
+@router.post(
+    "/delete",
+    response_model=APIResponse,
+    dependencies=[Depends(get_current_user)],
+)
+async def delete_torrent(url: str):
+    """
+    删除对应的种子
+    """
+    try:
+        resp = await TorrentManager().delete_torrent(url)
+        if resp:
+            resp = ResponseModel(
+                status_code=200,
+                status=True,
+                msg_en=f"Delete torrent for {url}",
+                msg_zh=f"删除 {url} 的种子",
+            )
+        else:
+            resp = ResponseModel(
+                status_code=406,
+                status=False,
+                msg_en=f"Can't find torrent with url {url}",
+                msg_zh=f"无法找到 url 为 {url} 的种子",
+            )
+        return resp
+    except Exception as e:
+        logger.error(f"[Bangumi] Error deleting torrent: {e}")
+        return ResponseModel(
+            status_code=500,
+            status=False,
+            msg_en="Internal server error",
+            msg_zh="服务器内部错误",
+        )
+
+
+@router.post(
+    "/download",
+    response_model=APIResponse,
+    dependencies=[Depends(get_current_user)],
+)
+async def download_bangumi(_id: int, torrent: Torrent):
+    """
+    手动下载对应 Bangumi 的种子
+    """
+    try:
+        resp = TorrentManager().download_torrent(_id, torrent)
+        if resp:
+            resp = ResponseModel(
+                status_code=200,
+                status=True,
+                msg_en=f"Download torrent for {_id}",
+                msg_zh=f"下载 {_id} 的种子",
+            )
+        else:
+            resp = ResponseModel(
+                status_code=406,
+                status=False,
+                msg_en=f"Can't find id {_id}",
+                msg_zh=f"无法找到 id {_id}",
+            )
+        return resp
+    except Exception as e:
+        logger.error(f"[Bangumi] Error downloading bangumi: {e}")
+        return ResponseModel(
+            status_code=500,
+            status=False,
+            msg_en="Internal server error",
+            msg_zh="服务器内部错误",
+        )

backend/src/module/database/combine.py

@@ -1,6 +1,6 @@
 import logging
 
-from sqlmodel import Session, SQLModel, and_, delete, false, or_, select, text
+from sqlmodel import Session, SQLModel, and_, false, select, text
 
 from module.models import Bangumi, RSSItem, Torrent, User
 

backend/src/module/database/torrent.py

@@ -119,6 +119,7 @@ def delete_by_url(self, url: str) -> bool:
             )
             self.session.commit()
             return True
+        return False
 
     def delete_by_duid(self, duid: str) -> bool:
         stmt = select(Torrent).where(Torrent.download_uid == duid)

backend/src/module/downloader/download_client.py

@@ -249,7 +249,6 @@ async def move_torrent(self, hashes: list[str] | str, location: str) -> bool:
         if not await self.wait_for_login():
             return False  # 登录失败时返回False
         try:
-            #TODO: 好像是用 | 连起来就行,但现在好像用不上了
             result = await self.downloader.move(hashes=hashes, new_location=location)
             if result:
                 logger.info(f"[Downloader] Move torrents {hashes} to {location}")

backend/src/module/manager/bangumi.py

@@ -4,8 +4,10 @@
 from module.conf import settings
 from module.database import Database, engine
 from module.downloader import Client as DownlondClient
+from module.downloader import download_queue
 from module.manager.torrent import TorrentManager
 from module.models import Bangumi, BangumiUpdate, Torrent
+from module.network import RequestContent
 from module.parser import MikanParser, TmdbParser
 from module.utils import gen_save_path
 from module.utils.events import Event, EventType, event_bus
@@ -158,6 +160,10 @@ def search_one(self, _id: int | str):
                 return data
 
 
+
+
+
 if __name__ == "__main__":
     manager = TorrentManager()
-    asyncio.run(manager.refresh_poster())
+    manager = BangumiManager()
+    asyncio.run(manager.fetch_all_bangumi_torrents(1))

backend/src/module/manager/renamer.py

@@ -248,6 +248,8 @@ async def rename_torrent(self, torrent: Torrent, bangumi: Bangumi | None = None)
             return
         if bangumi is None:
             with Database() as db:
+                #NOTE: 这里会更新一下 season 和 official_title
+                # 可能是因为更新了番剧信息导致的
                 bangumi = db.torrent_to_bangumi(torrent)
                 if not bangumi:
                     bangumi = Bangumi(

backend/src/module/manager/torrent.py

@@ -2,7 +2,9 @@
 
 from module.database import Database, engine
 from module.downloader import Client as DownlondClient
-from module.models import Bangumi
+from module.downloader import download_queue
+from module.models import Bangumi, Torrent
+from module.network import RequestContent
 
 logger = logging.getLogger(__name__)
 
@@ -36,3 +38,56 @@ async def delete_torrents(self, data: Bangumi) -> bool:
             logger.info(f"Delete rule and torrents for {data.official_title}")
             return True
         return False
+
+    async def delete_torrent(self,url:str)->bool:
+        with Database(engine) as db:
+            torrent =  db.torrent.search_by_url(url)
+            if torrent and torrent.download_uid:
+                res = await DownlondClient.delete_torrent(torrent.download_uid)
+            res = db.torrent.delete_by_url(url)
+            return res
+
+    def download_torrent(self, _id:int,torrent: Torrent) -> bool:
+        """下载种子
+        Args:
+            bangumi: 番剧信息
+        """
+        with Database() as db:
+            bangumi = db.bangumi.search_id(int(_id))
+            if not bangumi:
+                logger.error(f"[Manager] Can't find data with {_id}")
+                return False
+
+        # 构造一个 torrent
+        torrent.bangumi_official_title = bangumi.official_title
+        torrent.bangumi_season = bangumi.season
+        torrent.rss_link = bangumi.rss_link
+
+        download_queue.add(torrent, bangumi)
+        return True
+
+    async def fetch_all_bangumi_torrents(self,_id:int)->list[Torrent]:
+        # 先拉一下 rss
+        # 然后从数据库找
+        # 最后返回一个 torrent list
+        with Database() as db:
+            bangumi = db.bangumi.search_id(int(_id))
+            if not bangumi:
+                logger.error(f"[Manager] Can't find data with {_id}")
+                return []
+            exist_torrents: list[Torrent] = db.find_torrent_by_bangumi(bangumi)
+            url = bangumi.rss_link
+        existing_urls = {torrent.url for torrent in exist_torrents}
+        async with RequestContent() as req:
+            torrents = await req.get_torrents(url)
+        title_raws = bangumi.title_raw.split(",")
+        for torrent in torrents:
+            if torrent.url in existing_urls:
+                continue
+            for title in title_raws:
+                if title in torrent.name:
+                    exist_torrents.append(torrent)
+                    existing_urls.add(torrent.url)
+                    break
+        exist_torrents.sort(key=lambda x: x.name, reverse=True)
+        return exist_torrents

backend/src/module/models/torrent.py

@@ -1,4 +1,4 @@
-from datetime import datetime
+from datetime import datetime, timezone
 
 from pydantic import BaseModel
 from sqlmodel import Field, SQLModel
@@ -9,14 +9,13 @@
 class Torrent(SQLModel, table=True):
     url: str = Field(primary_key=True, alias="url")
     name: str = Field(default="", alias="name")
-    created_at: datetime = Field(default_factory=datetime.utcnow, index=True)
+    created_at: datetime = Field(default_factory=lambda: datetime.now(timezone.utc), index=True)
     downloaded: bool = Field(default=False, alias="downloaded")
     renamed: bool = Field(default=False, alias="renamed")
     download_uid: str | None = Field(default=None, alias="duid", index=True)
     bangumi_official_title: str = Field(default="", index=True, alias="bangumi_title")
     bangumi_season: int = Field(default=1, index=True, alias="bangumi_season")
     rss_link: str = Field(default="", alias="ruid", index=True)
-    # TODO: 添加外键字段 rss_id 和 bangumi_id 替代当前的字符串引用
     homepage: str | None = Field(default=None, alias="homepage")