mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-09-06 07:56:52 +08:00
refactor: 推进后端分层架构治理
This commit is contained in:
@@ -0,0 +1,201 @@
|
||||
"""中心服务订阅和工作流分享用例。"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import json
|
||||
from collections.abc import Awaitable, Callable
|
||||
from typing import Any, Optional
|
||||
|
||||
from app.schemas.media import resolve_media_identity
|
||||
|
||||
|
||||
class ServerSharingService:
|
||||
"""协调本地订阅、工作流读取与中心服务分享传输。"""
|
||||
|
||||
SUBSCRIBE_FIELDS = frozenset({
|
||||
"share_title", "share_comment", "share_user", "share_uid", "name",
|
||||
"year", "type", "keyword", "media_source", "media_id", "music_type",
|
||||
"total_tracks", "season", "poster", "backdrop", "vote", "description",
|
||||
"genre_ids", "include", "exclude", "quality", "resolution", "effect",
|
||||
"total_episode", "custom_words", "media_category", "episode_group",
|
||||
"date",
|
||||
})
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
*,
|
||||
subscribe_provider: Callable[[int], Any],
|
||||
async_subscribe_provider: Callable[[int], Awaitable[Any]],
|
||||
workflow_provider: Callable[[int], Any],
|
||||
async_workflow_provider: Callable[[int], Awaitable[Any]],
|
||||
user_uuid_provider: Callable[[], str],
|
||||
subscribe_sender: Callable[[dict], Any],
|
||||
async_subscribe_sender: Callable[[dict], Awaitable[Any]],
|
||||
workflow_sender: Callable[[dict], Any],
|
||||
async_workflow_sender: Callable[[dict], Awaitable[Any]],
|
||||
response_handler: Callable[[Any, Callable[[], None]], tuple[bool, str]],
|
||||
subscribe_cache_clearer: Callable[[], None],
|
||||
workflow_cache_clearer: Callable[[], None],
|
||||
) -> None:
|
||||
"""保存本地数据端口、中心服务传输端口和缓存失效端口。"""
|
||||
self._subscribe_provider = subscribe_provider
|
||||
self._async_subscribe_provider = async_subscribe_provider
|
||||
self._workflow_provider = workflow_provider
|
||||
self._async_workflow_provider = async_workflow_provider
|
||||
self._user_uuid_provider = user_uuid_provider
|
||||
self._subscribe_sender = subscribe_sender
|
||||
self._async_subscribe_sender = async_subscribe_sender
|
||||
self._workflow_sender = workflow_sender
|
||||
self._async_workflow_sender = async_workflow_sender
|
||||
self._response_handler = response_handler
|
||||
self._subscribe_cache_clearer = subscribe_cache_clearer
|
||||
self._workflow_cache_clearer = workflow_cache_clearer
|
||||
|
||||
def build_subscribe_payload(self, item: Optional[dict]) -> Optional[dict]:
|
||||
"""构造订阅分享载荷并隔离本地字段和旧专用 ID。"""
|
||||
if not isinstance(item, dict):
|
||||
return None
|
||||
media_source, media_id = resolve_media_identity(media=item)
|
||||
if not media_source or not media_id:
|
||||
return None
|
||||
payload = {
|
||||
key: value
|
||||
for key, value in item.items()
|
||||
if key in self.SUBSCRIBE_FIELDS
|
||||
}
|
||||
payload["media_source"] = str(media_source)
|
||||
payload["media_id"] = media_id
|
||||
return payload
|
||||
|
||||
@staticmethod
|
||||
def prepare_workflow(workflow: Any) -> dict:
|
||||
"""移除本地字段并把动作和流程编码为中心服务兼容格式。"""
|
||||
workflow_dict = workflow.to_dict()
|
||||
workflow_dict.pop("id", None)
|
||||
workflow_dict.pop("context", None)
|
||||
workflow_dict["actions"] = json.dumps(workflow_dict["actions"] or [])
|
||||
workflow_dict["flows"] = json.dumps(workflow_dict["flows"] or [])
|
||||
return workflow_dict
|
||||
|
||||
@staticmethod
|
||||
def validate_workflow(workflow: Any) -> tuple[bool, str]:
|
||||
"""验证工作流存在且同时包含动作与流程。"""
|
||||
if not workflow:
|
||||
return False, "工作流不存在"
|
||||
if not workflow.actions or not workflow.flows:
|
||||
return False, "请分享有动作和流程的工作流"
|
||||
return True, ""
|
||||
|
||||
def share_subscribe(
|
||||
self,
|
||||
*,
|
||||
enabled: bool,
|
||||
subscribe_id: int,
|
||||
share_title: str,
|
||||
share_comment: str,
|
||||
share_user: str,
|
||||
) -> tuple[bool, str]:
|
||||
"""同步读取并分享指定订阅。"""
|
||||
if not enabled:
|
||||
return False, "当前没有开启订阅数据共享功能"
|
||||
subscribe = self._subscribe_provider(subscribe_id)
|
||||
if not subscribe:
|
||||
return False, "订阅不存在"
|
||||
payload = self.build_subscribe_payload({
|
||||
"share_title": share_title,
|
||||
"share_comment": share_comment,
|
||||
"share_user": share_user,
|
||||
"share_uid": self._user_uuid_provider(),
|
||||
**subscribe.to_dict(),
|
||||
})
|
||||
if not payload:
|
||||
return False, "订阅媒体身份不完整"
|
||||
return self._response_handler(
|
||||
self._subscribe_sender(payload),
|
||||
self._subscribe_cache_clearer,
|
||||
)
|
||||
|
||||
async def async_share_subscribe(
|
||||
self,
|
||||
*,
|
||||
enabled: bool,
|
||||
subscribe_id: int,
|
||||
share_title: str,
|
||||
share_comment: str,
|
||||
share_user: str,
|
||||
) -> tuple[bool, str]:
|
||||
"""异步读取并分享指定订阅。"""
|
||||
if not enabled:
|
||||
return False, "当前没有开启订阅数据共享功能"
|
||||
subscribe = await self._async_subscribe_provider(subscribe_id)
|
||||
if not subscribe:
|
||||
return False, "订阅不存在"
|
||||
payload = self.build_subscribe_payload({
|
||||
"share_title": share_title,
|
||||
"share_comment": share_comment,
|
||||
"share_user": share_user,
|
||||
"share_uid": self._user_uuid_provider(),
|
||||
**subscribe.to_dict(),
|
||||
})
|
||||
if not payload:
|
||||
return False, "订阅媒体身份不完整"
|
||||
return self._response_handler(
|
||||
await self._async_subscribe_sender(payload),
|
||||
self._subscribe_cache_clearer,
|
||||
)
|
||||
|
||||
def share_workflow(
|
||||
self,
|
||||
*,
|
||||
enabled: bool,
|
||||
workflow_id: int,
|
||||
share_title: str,
|
||||
share_comment: str,
|
||||
share_user: str,
|
||||
) -> tuple[bool, str]:
|
||||
"""同步读取并分享指定工作流。"""
|
||||
if not enabled:
|
||||
return False, "当前没有开启工作流数据共享功能"
|
||||
workflow = self._workflow_provider(workflow_id)
|
||||
valid, message = self.validate_workflow(workflow)
|
||||
if not valid:
|
||||
return False, message
|
||||
payload = {
|
||||
"share_title": share_title,
|
||||
"share_comment": share_comment,
|
||||
"share_user": share_user,
|
||||
"share_uid": self._user_uuid_provider(),
|
||||
**self.prepare_workflow(workflow),
|
||||
}
|
||||
return self._response_handler(
|
||||
self._workflow_sender(payload),
|
||||
self._workflow_cache_clearer,
|
||||
)
|
||||
|
||||
async def async_share_workflow(
|
||||
self,
|
||||
*,
|
||||
enabled: bool,
|
||||
workflow_id: int,
|
||||
share_title: str,
|
||||
share_comment: str,
|
||||
share_user: str,
|
||||
) -> tuple[bool, str]:
|
||||
"""异步读取并分享指定工作流。"""
|
||||
if not enabled:
|
||||
return False, "当前没有开启工作流数据共享功能"
|
||||
workflow = await self._async_workflow_provider(workflow_id)
|
||||
valid, message = self.validate_workflow(workflow)
|
||||
if not valid:
|
||||
return False, message
|
||||
payload = {
|
||||
"share_title": share_title,
|
||||
"share_comment": share_comment,
|
||||
"share_user": share_user,
|
||||
"share_uid": self._user_uuid_provider(),
|
||||
**self.prepare_workflow(workflow),
|
||||
}
|
||||
return self._response_handler(
|
||||
await self._async_workflow_sender(payload),
|
||||
self._workflow_cache_clearer,
|
||||
)
|
||||
Reference in New Issue
Block a user