Merge pull request #2777 from Aqr-K/dev-transtype

app/api/endpoints/storage.py

@@ -205,3 +205,14 @@ def usage(name: str, _: User = Depends(get_current_active_superuser)) -> Any:
     if ret:
         return ret
     return schemas.StorageUsage()
+
+
+@router.get("/transtype/{name}", summary="支持的整理方式获取", response_model=schemas.StorageTransType)
+def transtype(name: str, _: User = Depends(get_current_active_superuser)) -> Any:
+    """
+    查询支持的整理方式
+    """
+    ret = StorageChain().support_transtype(name)
+    if ret:
+        return schemas.StorageTransType(transtype=ret)
+    return schemas.StorageTransType()

app/chain/storage.py

@@ -85,3 +85,9 @@ class StorageChain(ChainBase):
         存储使用情况
         """
         return self.run_module("storage_usage", storage=storage)
+
+    def support_transtype(self, storage: str) -> Optional[str]:
+        """
+        获取支持的整理方式
+        """
+        return self.run_module("support_transtype", storage=storage)

app/modules/filemanager/__init__.py

@@ -89,6 +89,16 @@ class FileManagerModule(_ModuleBase):
     def init_setting(self) -> Tuple[str, Union[str, bool]]:
         pass
 
+    def support_transtype(self, storage: str) -> Optional[Dict[str, str]]:
+        """
+        支持的整理方式
+        """
+        storage_oper = self.__get_storage_oper(storage)
+        if not storage_oper:
+            logger.error(f"不支持 {storage} 的整理方式获取")
+            return None
+        return storage_oper.support_transtype()
+
     def recommend_name(self, meta: MetaBase, mediainfo: MediaInfo) -> Optional[str]:
         """
         获取重命名后的名称

app/modules/filemanager/storages/u115.py

@@ -73,7 +73,7 @@ class U115Pan(StorageBase, metaclass=Singleton):
             return None
         return Credential.from_dict(cookie_dict.dict().get("config"))
 
-    def __save_credentail(self, credential: Credential):
+    def __save_credential(self, credential: Credential):
         """
         设置115认证参数
         """
@@ -117,7 +117,7 @@ class U115Pan(StorageBase, metaclass=Singleton):
             status = self.cloud.qrcode_poll(self._session)
             if status == QrcodeStatus.Done:
                 # 确认完成，保存认证信息
-                self.__save_credentail(self.cloud.export_credentail())
+                self.__save_credential(self.cloud.export_credentail())
                 result = {
                     "status": 1,
                     "tip": "登录成功！"

app/schemas/file.py

@@ -41,3 +41,8 @@ class StorageUsage(BaseModel):
     total: float = 0.0
     # 剩余空间
     available: float = 0.0
+
+
+class StorageTransType(BaseModel):
+    # 传输类型
+    transtype: Optional[dict] = {}