From c69df36b982b9431a29eb754355637bcb1abd955 Mon Sep 17 00:00:00 2001
From: jxxghp <jxxghp@gmail.com>
Date: Wed, 25 Dec 2024 18:11:57 +0800
Subject: [PATCH] add transfer queue api

---
 app/api/endpoints/transfer.py | 11 ++++++++++-
 app/chain/transfer.py         |  6 ++++++
 2 files changed, 16 insertions(+), 1 deletion(-)

diff --git a/app/api/endpoints/transfer.py b/app/api/endpoints/transfer.py
index 48409b77..c3bb75d9 100644
--- a/app/api/endpoints/transfer.py
+++ b/app/api/endpoints/transfer.py
@@ -1,5 +1,5 @@
 from pathlib import Path
-from typing import Any
+from typing import Any, List
 
 from fastapi import APIRouter, Depends
 from sqlalchemy.orm import Session
@@ -47,6 +47,15 @@ def query_name(path: str, filetype: str,
     })
 
 
+@router.get("/queue", summary="查询整理队列", response_model=List[dict])
+def query_queue(_: schemas.TokenPayload = Depends(verify_token)) -> Any:
+    """
+    查询整理队列
+    :param _: Token校验
+    """
+    return TransferChain().get_queue_tasks()
+
+
 @router.post("/manual", summary="手动转移", response_model=schemas.Response)
 def manual_transfer(transer_item: ManualTransferItem,
                     db: Session = Depends(get_db),
diff --git a/app/chain/transfer.py b/app/chain/transfer.py
index eec63e98..1efabc29 100644
--- a/app/chain/transfer.py
+++ b/app/chain/transfer.py
@@ -471,6 +471,12 @@ class TransferChain(ChainBase, metaclass=Singleton):
 
         return transferinfo.success, transferinfo.message
 
+    def get_queue_tasks(self) -> List[dict]:
+        """
+        获取整理任务列表
+        """
+        return self.jobview.list_jobs()
+
     def recommend_name(self, meta: MetaBase, mediainfo: MediaInfo) -> Optional[str]:
         """
         获取重命名后的名称