add disable torrent

Author: shinonomeow

backend/src/module/api/torrent.py

@@ -68,6 +68,29 @@ async def delete_torrent(url: str):
         )
 
 
+@router.post("/disable",response_model=APIResponse,dependencies=[Depends(get_current_user)])
+async def disable_torrent(url:str,name,_id:int):
+    """
+    禁用对应的种子
+    """
+    try:
+        await TorrentManager().disable_torrent(url,name, _id)
+        return ResponseModel(
+            status_code=200,
+            status=True,
+            msg_en=f"Successfully disabled torrent with url {url}",
+            msg_zh=f"成功禁用 url 为 {url} 的种子",
+        )
+    except Exception as e:
+        logger.error(f"[Bangumi] Error disabling torrent: {e}")
+        return ResponseModel(
+            status_code=500,
+            status=False,
+            msg_en="Internal server error",
+            msg_zh="服务器内部错误",
+        )
+
+
 @router.post(
     "/download",
     response_model=APIResponse,

backend/src/module/manager/torrent.py

@@ -91,3 +91,19 @@ async def fetch_all_bangumi_torrents(self,_id:int)->list[Torrent]:
                     break
         exist_torrents.sort(key=lambda x: x.name, reverse=True)
         return exist_torrents
+
+    async def disable_torrent(self,url,name,_id:int)->bool:
+        with Database(engine) as db:
+            bangumi = db.bangumi.search_id(_id)
+            if not bangumi:
+                return False
+            torrent = db.torrent.search_by_url(url)
+            if not torrent:
+                torrent = Torrent(url=url,name=name)
+            torrent.downloaded = True
+            torrent.renamed = True
+            torrent.bangumi_official_title = bangumi.official_title
+            torrent.bangumi_season = bangumi.season
+            torrent.rss_link = bangumi.rss_link
+            db.torrent.add(torrent)
+            return True

webui/src/api/bangumi.ts

@@ -186,4 +186,22 @@ export const apiBangumi = {
     );
     return data;
   },
+
+  /**
+   * 禁用指定的 torrent（将其标记为已下载和已重命名）
+   * @param url - torrent url
+   * @param name - torrent name
+   * @param bangumiId - bangumi id
+   * @returns axios 请求返回的数据
+   */
+  async disableTorrent(url: string, name: string, bangumiId: number) {
+    const { data } = await axios.post<ApiSuccess>(
+      `api/v1/torrent/disable`,
+      null,
+      {
+        params: { url, name, _id: bangumiId }
+      }
+    );
+    return data;
+  },
 };

webui/src/components/ab-torrent-manage.vue

@@ -1,5 +1,5 @@
 <script lang="ts" setup>
-import { Download, Delete, Refresh } from '@icon-park/vue-next';
+import { Download, Delete, Refresh, Forbid } from '@icon-park/vue-next';
 import { NModal, NButton, NDataTable, NTag } from 'naive-ui';
 import type { DataTableColumns } from 'naive-ui';
 import type { BangumiRule } from '#/bangumi';
@@ -65,7 +65,7 @@ const columns = computed((): DataTableColumns<Torrent> => [
   {
     title: t('torrent.actions'),
     key: 'actions',
-    width: 140,
+    width: 180,
     align: 'center',
     render(row: Torrent) {
       return h(
@@ -82,6 +82,16 @@ const columns = computed((): DataTableColumns<Torrent> => [
             },
             { default: () => h(Download, { size: 16 }) }
           ),
+          h(
+            NButton,
+            {
+              size: 'small',
+              type: 'warning',
+              disabled: row.downloaded && row.renamed,
+              onClick: () => handleDisable(row),
+            },
+            { default: () => h(Forbid, { size: 16 }) }
+          ),
           h(
             NButton,
             {
@@ -121,6 +131,17 @@ function handleDownload(torrent: Torrent) {
   }).execute(props.bangumi.id, torrent);
 }
 
+function handleDisable(torrent: Torrent) {
+  if (!props.bangumi?.id) return;
+  
+  useApi(apiBangumi.disableTorrent, {
+    showMessage: true,
+    onSuccess() {
+      fetchTorrents(); // 刷新列表
+    },
+  }).execute(torrent.url, torrent.name, props.bangumi.id);
+}
+
 function handleDelete(torrent: Torrent) {
   useApi(apiBangumi.deleteTorrent, {
     showMessage: true,