fix storage api

app/modules/filemanager/__init__.py

@@ -16,7 +16,7 @@ from app.helper.module import ModuleHelper
 from app.log import logger
 from app.modules import _ModuleBase
 from app.modules.filemanager.storage import StorageBase
-from app.schemas import TransferInfo, ExistMediaInfo, TmdbEpisode, TransferDirectoryConf, FileItem
+from app.schemas import TransferInfo, ExistMediaInfo, TmdbEpisode, TransferDirectoryConf, FileItem, StorageUsage
 from app.schemas.types import MediaType
 from app.utils.system import SystemUtils
 
@@ -193,6 +193,16 @@ class FileManagerModule(_ModuleBase):
             return None
         return storage_oper.snapshot(path)
 
+    def storage_usage(self, storage: str) -> Optional[StorageUsage]:
+        """
+        存储使用情况
+        """
+        storage_oper = self.__get_storage_oper(storage)
+        if not storage_oper:
+            logger.error(f"不支持 {storage} 的存储使用情况")
+            return None
+        return storage_oper.usage()
+
     def transfer(self, fileitem: FileItem, meta: MetaBase, mediainfo: MediaInfo,
                  transfer_type: str, target_storage: str = None, target_path: Path = None,
                  episodes_info: List[TmdbEpisode] = None,

app/modules/filemanager/storage/__init__.py

@@ -138,6 +138,13 @@ class StorageBase(metaclass=ABCMeta):
         """
         pass
 
+    @abstractmethod
+    def usage(self) -> Optional[schemas.StorageUsage]:
+        """
+        存储使用情况
+        """
+        pass
+
     def snapshot(self, path: Path) -> Dict[str, float]:
         """
         快照文件系统，输出所有层级文件信息（不含目录）

app/modules/filemanager/storage/alipan.py

@@ -744,3 +744,9 @@ class AliPan(StorageBase):
         软链接文件
         """
         pass
+
+    def usage(self) -> Optional[schemas.StorageUsage]:
+        """
+        存储使用情况
+        """
+        pass

app/modules/filemanager/storage/local.py

@@ -3,6 +3,7 @@ from pathlib import Path
 from typing import Optional, List
 
 from app import schemas
+from app.helper.directory import DirectoryHelper
 from app.log import logger
 from app.modules.filemanager.storage import StorageBase
 from app.schemas.types import StorageSchema
@@ -231,3 +232,14 @@ class LocalStorage(StorageBase):
             logger.error(f"移动文件失败：{message}")
             return False
         return True
+
+    def usage(self) -> Optional[schemas.StorageUsage]:
+        """
+        存储使用情况
+        """
+        library_dirs = DirectoryHelper().get_local_library_dirs()
+        total_storage, free_storage = SystemUtils.space_usage([Path(d.library_path) for d in library_dirs])
+        return schemas.StorageUsage(
+            total=total_storage,
+            available=free_storage
+        )

app/modules/filemanager/storage/rclone.py

@@ -296,3 +296,9 @@ class Rclone(StorageBase):
 
     def softlink(self, fileitm: schemas.FileItem, target_file: Path) -> bool:
         pass
+
+    def usage(self) -> Optional[schemas.StorageUsage]:
+        """
+        存储使用情况
+        """
+        pass

app/modules/filemanager/storage/u115.py

@@ -375,3 +375,9 @@ class U115Pan(StorageBase, metaclass=Singleton):
 
     def softlink(self, fileitm: schemas.FileItem, target_file: Path) -> bool:
         pass
+
+    def usage(self) -> Optional[schemas.StorageUsage]:
+        """
+        存储使用情况
+        """
+        pass