mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-08-29 03:56:43 +08:00
65 lines
2.3 KiB
Python
65 lines
2.3 KiB
Python
"""Dashboard 统计查询用例。"""
|
|
|
|
from collections.abc import Callable
|
|
from typing import Any, Optional, Protocol
|
|
|
|
from app.schemas.dashboard import Statistic
|
|
|
|
|
|
class TransferHistoryQueryRepository(Protocol):
|
|
"""Dashboard 所需的整理历史统计端口。"""
|
|
|
|
def monthly_media_statistics(self) -> tuple[int, int, int, int]:
|
|
"""返回本月电影、剧集、单集和音乐数量。"""
|
|
...
|
|
|
|
async def async_statistic(self, days: int = 7) -> list[Any]:
|
|
"""返回最近若干天的整理趋势。"""
|
|
...
|
|
|
|
|
|
class DashboardQueryService:
|
|
"""汇总媒体服务数据与整理历史统计。"""
|
|
|
|
def __init__(
|
|
self,
|
|
*,
|
|
repository: TransferHistoryQueryRepository,
|
|
media_statistics: Callable[[Optional[str]], Optional[list[Statistic]]],
|
|
) -> None:
|
|
"""保存整理历史端口和媒体服务统计提供方。"""
|
|
self._repository = repository
|
|
self._media_statistics = media_statistics
|
|
|
|
def statistic(self, name: Optional[str] = None) -> Statistic:
|
|
"""返回媒体服务总量和本月新增量。"""
|
|
media_statistics = self._media_statistics(name)
|
|
if media_statistics:
|
|
result = Statistic()
|
|
has_episode_count = False
|
|
for item in media_statistics:
|
|
result.movie_count += item.movie_count or 0
|
|
result.tv_count += item.tv_count or 0
|
|
result.music_count += item.music_count or 0
|
|
result.user_count += item.user_count or 0
|
|
if item.episode_count is not None:
|
|
result.episode_count += item.episode_count or 0
|
|
has_episode_count = True
|
|
if not has_episode_count:
|
|
result.episode_count = None
|
|
else:
|
|
result = Statistic()
|
|
|
|
(
|
|
result.movie_count_month,
|
|
result.tv_count_month,
|
|
result.episode_count_month,
|
|
result.music_count_month,
|
|
) = self._repository.monthly_media_statistics()
|
|
return result
|
|
|
|
async def transfer(self, days: int = 7) -> list[int]:
|
|
"""返回最近若干天的整理数量序列。"""
|
|
rows = await self._repository.async_statistic(days)
|
|
return [row[1] for row in rows]
|