mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-08-19 05:03:57 +08:00
123 lines
5.4 KiB
Python
123 lines
5.4 KiB
Python
"""
|
||
订阅的写入路径。
|
||
|
||
这两个函数把 MediaInfo / MusicInfo 翻译成一行订阅，是订阅表的唯一写入口。翻译此前
|
||
长在 SubscribeOper.add 上，但取标题、选海报尺寸、判音乐实体、决定哪几个字段构成一条
|
||
订阅的身份，都是订阅业务的规则而非数据访问——Oper 只该收敛查询，领域对象不该出现在
|
||
它的入参里。搬上来之后 SubscribeOper 收到的是纯粹的持久化字典，与
|
||
app/application/history.py 里整理历史的写入路径同构。
|
||
|
||
留在 Oper 的是列类型强转与建库时间戳：那几步是为 PostgreSQL 的严格类型检查和订阅表
|
||
自己的列类型而存在的，跟着列走比跟着调用方走更不容易漂。
|
||
|
||
字段映射错了不会报错，只会让订阅静静地记错——而搜索、洗版、完成判定、去重全都读这
|
||
张表。同步与异步是两份逐字复制的实现，改一条漏一条就是真实缺陷，故翻译与身份构造由
|
||
下方 _translate 单点承担，两条链路只在「怎么查、怎么写」上分叉。
|
||
"""
|
||
from typing import Optional, Tuple
|
||
|
||
from app.db.oper.subscribe import SubscribeOper
|
||
from app.domain.context import MediaInfo, MusicInfo
|
||
from app.schemas.media import resolve_media_identity
|
||
from app.schemas.types import MUSIC_ENTITY_ALBUM, MediaType
|
||
|
||
# 身份不完整时的固定返回。身份不全的订阅写进去就是一条永远匹配不上资源的僵尸订阅，
|
||
# 而后续按身份去重也会失效，所以必须在查询与建模之前短路
|
||
INCOMPLETE_IDENTITY = (0, "媒体身份不完整")
|
||
|
||
|
||
def _music_entity(mediainfo: MediaInfo | MusicInfo) -> Optional[str]:
|
||
"""
|
||
取音乐实体类型；非音乐媒体一律为空。
|
||
|
||
影视订阅带着 music_type 会被音乐去重逻辑当成音乐实体，造成串号。查重身份与写入
|
||
字段都取这一个值，避免两处各算一遍后悄悄分叉。
|
||
:param mediainfo: 识别结果
|
||
:return: 音乐实体类型，非音乐媒体为 None
|
||
"""
|
||
if mediainfo.type != MediaType.MUSIC:
|
||
return None
|
||
return getattr(mediainfo, "music_type", None)
|
||
|
||
|
||
def _translate(mediainfo: MediaInfo | MusicInfo,
|
||
kwargs: dict) -> Optional[Tuple[dict, dict, Optional[str]]]:
|
||
"""
|
||
把识别结果翻译成查重身份与写入字段。
|
||
|
||
:param mediainfo: 识别结果
|
||
:param kwargs: 调用方传入的订阅设置，媒体相关的同名字段会被识别结果覆盖
|
||
:return: (查重身份, 写入字段, 限定用户)；媒体身份不完整时返回 None
|
||
"""
|
||
owner_scope = bool(kwargs.pop("owner_scope", False))
|
||
username = kwargs.get("username") if owner_scope else None
|
||
media_source, media_id = resolve_media_identity(
|
||
media=mediainfo,
|
||
media_source=kwargs.get("media_source"),
|
||
media_id=kwargs.get("media_id"),
|
||
)
|
||
if not media_source or not media_id:
|
||
return None
|
||
music_type = _music_entity(mediainfo)
|
||
identity = {
|
||
"media_source": str(media_source),
|
||
"media_id": media_id,
|
||
"music_type": music_type,
|
||
"season": kwargs.get("season"),
|
||
"episode_group": mediainfo.episode_group,
|
||
}
|
||
payload = dict(kwargs)
|
||
payload.update({
|
||
"name": mediainfo.title,
|
||
"year": mediainfo.year,
|
||
"type": mediainfo.type.value,
|
||
"media_source": str(media_source),
|
||
"media_id": media_id,
|
||
"episode_group": mediainfo.episode_group,
|
||
"poster": mediainfo.get_poster_image(),
|
||
"backdrop": mediainfo.get_backdrop_image(),
|
||
"vote": mediainfo.vote_average,
|
||
"description": mediainfo.overview,
|
||
"music_type": music_type,
|
||
# 整专完成判定拿 total_tracks 当分母，单曲带着专辑的曲目数会永远判不到完成
|
||
"total_tracks": getattr(mediainfo, "total_tracks", None)
|
||
if music_type == MUSIC_ENTITY_ALBUM else None,
|
||
})
|
||
return identity, payload, username
|
||
|
||
|
||
def add_subscribe(mediainfo: MediaInfo | MusicInfo,
|
||
subscribe_oper: Optional[SubscribeOper] = None,
|
||
**kwargs) -> Tuple[int, str]:
|
||
"""
|
||
新增订阅。
|
||
:param mediainfo: 识别结果
|
||
:param subscribe_oper: 复用的订阅操作对象，未传时新建
|
||
:param kwargs: 订阅设置；owner_scope 为真时按用户名限定查重范围
|
||
:return: (订阅 ID, 结果说明)；ID 为 0 表示未新增
|
||
"""
|
||
translated = _translate(mediainfo, kwargs)
|
||
if translated is None:
|
||
return INCOMPLETE_IDENTITY
|
||
identity, payload, username = translated
|
||
oper = subscribe_oper or SubscribeOper()
|
||
return oper.add(identity=identity, payload=payload, username=username)
|
||
|
||
|
||
async def async_add_subscribe(mediainfo: MediaInfo | MusicInfo,
|
||
subscribe_oper: Optional[SubscribeOper] = None,
|
||
**kwargs) -> Tuple[int, str]:
|
||
"""
|
||
异步新增订阅。
|
||
:param mediainfo: 识别结果
|
||
:param subscribe_oper: 复用的订阅操作对象，未传时新建
|
||
:param kwargs: 订阅设置；owner_scope 为真时按用户名限定查重范围
|
||
:return: (订阅 ID, 结果说明)；ID 为 0 表示未新增
|
||
"""
|
||
translated = _translate(mediainfo, kwargs)
|
||
if translated is None:
|
||
return INCOMPLETE_IDENTITY
|
||
identity, payload, username = translated
|
||
oper = subscribe_oper or SubscribeOper()
|
||
return await oper.async_add(identity=identity, payload=payload, username=username)
|