MoviePilot/app/application/storage.py

from typing import List, Optional

from app.schemas.system import StorageConf as _SchemaStorageConf
from app.application.configuration import get_configured_system_config
from app.schemas.types import SystemConfigKey


class StorageHelper:
    """
    存储帮助类
    """

    @staticmethod
    def get_storagies() -> List[_SchemaStorageConf]:
        """
        获取所有存储设置
        """
        storage_confs: List[dict] = get_configured_system_config().get(SystemConfigKey.Storages)
        if not storage_confs:
            return []
        return [_SchemaStorageConf(**s) for s in storage_confs]

    def get_storage(self, storage: str) -> Optional[_SchemaStorageConf]:
        """
        获取指定存储配置
        """
        storagies = self.get_storagies()
        for s in storagies:
            if s.type == storage:
                return s
        return None

    def set_storage(self, storage: str, conf: dict):
        """
        设置存储配置
        """
        storagies = self.get_storagies()
        if not storagies:
            storagies = [
                _SchemaStorageConf(
                    type=storage,
                    config=conf
                )
            ]
        else:
            for s in storagies:
                if s.type == storage:
                    s.config = conf
                    break
        get_configured_system_config().set(SystemConfigKey.Storages, [s.model_dump() for s in storagies])

    def add_storage(self, storage: str, name: str, conf: dict):
        """
        添加存储配置
        """
        storagies = self.get_storagies()
        if not storagies:
            storagies = [
                _SchemaStorageConf(
                    type=storage,
                    name=name,
                    config=conf
                )
            ]
        else:
            storagies.append(_SchemaStorageConf(
                type=storage,
                name=name,
                config=conf
            ))
        get_configured_system_config().set(SystemConfigKey.Storages, [s.model_dump() for s in storagies])

    def reset_storage(self, storage: str):
        """
        重置存储配置
        """
        storagies = self.get_storagies()
        for s in storagies:
            if s.type == storage:
                s.config = {}
                break
        get_configured_system_config().set(SystemConfigKey.Storages, [s.model_dump() for s in storagies])