下载控制

app/api/endpoints/download.py

@@ -10,9 +10,30 @@ router = APIRouter()
 
 
 @router.get("/", summary="正在下载", response_model=List[schemas.DownloadingTorrent])
-async def read_subscribes(
+async def read_downloading(
         _: schemas.TokenPayload = Depends(verify_token)) -> Any:
     """
     查询正在下载的任务
     """
     return DownloadChain().downloading()
+
+
+@router.put("/{hashString}", summary="开始/暂停", response_model=schemas.Response)
+async def set_downloading(
+        hashString: str,
+        oper: str,
+        _: schemas.TokenPayload = Depends(verify_token)) -> Any:
+    """
+    控制下载任务
+    """
+    return DownloadChain().set_downloading(hashString, oper)
+
+
+@router.delete("/{hashString}", summary="删除下载任务", response_model=schemas.Response)
+async def remove_downloading(
+        hashString: str,
+        _: schemas.TokenPayload = Depends(verify_token)) -> Any:
+    """
+    控制下载任务
+    """
+    return DownloadChain().remove_downloading(hashString)

app/chain/__init__.py

@@ -253,6 +253,22 @@ class ChainBase(AbstractSingleton, metaclass=Singleton):
         """
         return self.run_module("remove_torrents", hashs=hashs)
 
+    def start_torrents(self, hashs: Union[list, str]) -> bool:
+        """
+        开始下载
+        :param hashs:  种子Hash
+        :return: bool
+        """
+        return self.run_module("start_torrent", hash_str=hashs)
+
+    def stop_torrents(self, hashs: Union[list, str]) -> bool:
+        """
+        停止下载
+        :param hashs:  种子Hash
+        :return: bool
+        """
+        return self.run_module("stop_torrent", hashs=hashs)
+
     def media_exists(self, mediainfo: MediaInfo) -> Optional[ExistMediaInfo]:
         """
         判断媒体文件是否存在

app/chain/download.py

@@ -552,3 +552,19 @@ class DownloadChain(ChainBase):
                 }
             ret_torrents.append(torrent)
         return ret_torrents
+
+    def set_downloading(self, hash_str, oper: str) -> bool:
+        """
+        控制下载任务 start/pause
+        """
+        if oper == "start":
+            return self.start_torrents(hashs=[hash_str])
+        elif oper == "pause":
+            return self.stop_torrents(hashs=[hash_str])
+        return False
+
+    def remove_downloading(self, hash_str: str) -> bool:
+        """
+        删除下载任务
+        """
+        return self.remove_torrents(hashs=[hash_str])

app/modules/qbittorrent/__init__.py

@@ -141,6 +141,7 @@ class QbittorrentModule(_ModuleBase):
                     season_episode=meta.season_episode,
                     progress=torrent.get('progress'),
                     size=torrent.get('total_size'),
+                    state="downloading" if torrent.get('state') == "downloading" else "paused",
                     dlspeed=StringUtils.str_filesize(torrent.get('dlspeed')),
                     upspeed=StringUtils.str_filesize(torrent.get('upspeed')),
                 ))
@@ -167,3 +168,19 @@ class QbittorrentModule(_ModuleBase):
         :return: bool
         """
         return self.qbittorrent.delete_torrents(delete_file=True, ids=hashs)
+
+    def start_torrents(self, hashs: Union[list, str]) -> bool:
+        """
+        开始下载
+        :param hashs:  种子Hash
+        :return: bool
+        """
+        return self.qbittorrent.start_torrents(ids=hashs)
+
+    def stop_torrents(self, hashs: Union[list, str]) -> bool:
+        """
+        停止下载
+        :param hashs:  种子Hash
+        :return: bool
+        """
+        return self.qbittorrent.start_torrents(ids=hashs)

app/modules/transmission/__init__.py

@@ -124,6 +124,7 @@ class TransmissionModule(_ModuleBase):
                     season_episode=meta.season_episode,
                     progress=torrent.progress,
                     size=torrent.total_size,
+                    state="paused" if torrent.status == "stopped" else "downloading",
                     dlspeed=StringUtils.str_filesize(torrent.download_speed),
                     ulspeed=StringUtils.str_filesize(torrent.upload_speed),
                 ))
@@ -151,3 +152,19 @@ class TransmissionModule(_ModuleBase):
         :return: bool
         """
         return self.transmission.delete_torrents(delete_file=True, ids=hashs)
+
+    def start_torrents(self, hashs: Union[list, str]) -> bool:
+        """
+        开始下载
+        :param hashs:  种子Hash
+        :return: bool
+        """
+        return self.transmission.start_torrents(ids=hashs)
+
+    def stop_torrents(self, hashs: Union[list, str]) -> bool:
+        """
+        停止下载
+        :param hashs:  种子Hash
+        :return: bool
+        """
+        return self.transmission.start_torrents(ids=hashs)