mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-06-02 22:21:03 +08:00
402 lines
12 KiB
Python
402 lines
12 KiB
Python
import time
|
|
from typing import List, Optional
|
|
|
|
from sqlalchemy import Column, Integer, String, JSON, select
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
from sqlalchemy.orm import Session
|
|
|
|
from app.db import db_query, db_update, get_id_column, Base, async_db_query
|
|
|
|
|
|
class DownloadHistory(Base):
|
|
"""
|
|
下载历史记录
|
|
"""
|
|
|
|
id = get_id_column()
|
|
# 保存路径
|
|
path = Column(String, nullable=False, index=True)
|
|
# 类型 电影/电视剧
|
|
type = Column(String, nullable=False)
|
|
# 标题
|
|
title = Column(String, nullable=False)
|
|
# 年份
|
|
year = Column(String)
|
|
tmdbid = Column(Integer, index=True)
|
|
imdbid = Column(String)
|
|
tvdbid = Column(Integer)
|
|
doubanid = Column(String)
|
|
# Sxx
|
|
seasons = Column(String)
|
|
# Exx
|
|
episodes = Column(String)
|
|
# 海报
|
|
image = Column(String)
|
|
# 下载器
|
|
downloader = Column(String)
|
|
# 下载任务Hash
|
|
download_hash = Column(String, index=True)
|
|
# 种子名称
|
|
torrent_name = Column(String)
|
|
# 种子描述
|
|
torrent_description = Column(String)
|
|
# 种子站点
|
|
torrent_site = Column(String)
|
|
# 下载用户
|
|
userid = Column(String)
|
|
# 下载用户名/插件名
|
|
username = Column(String)
|
|
# 下载渠道
|
|
channel = Column(String)
|
|
# 创建时间
|
|
date = Column(String)
|
|
# 附加信息
|
|
note = Column(JSON)
|
|
# 自定义媒体类别
|
|
media_category = Column(String)
|
|
# 剧集组
|
|
episode_group = Column(String)
|
|
# 自定义识别词（用于整理时应用）
|
|
custom_words = Column(String)
|
|
|
|
@classmethod
|
|
@db_query
|
|
def get_by_hash(cls, db: Session, download_hash: str):
|
|
return (
|
|
db.query(DownloadHistory)
|
|
.filter(DownloadHistory.download_hash == download_hash)
|
|
.order_by(DownloadHistory.date.desc())
|
|
.first()
|
|
)
|
|
|
|
@classmethod
|
|
@db_query
|
|
def get_by_hashes(cls, db: Session, download_hashes: List[str]):
|
|
"""
|
|
批量查询多个下载任务的最新历史记录，避免在上层形成 N+1 查询。
|
|
"""
|
|
normalized_hashes = []
|
|
seen_hashes = set()
|
|
for download_hash in download_hashes or []:
|
|
if not download_hash or download_hash in seen_hashes:
|
|
continue
|
|
seen_hashes.add(download_hash)
|
|
normalized_hashes.append(download_hash)
|
|
|
|
if not normalized_hashes:
|
|
return []
|
|
|
|
histories = (
|
|
db.query(DownloadHistory)
|
|
.filter(DownloadHistory.download_hash.in_(normalized_hashes))
|
|
.order_by(DownloadHistory.download_hash, DownloadHistory.date.desc())
|
|
.all()
|
|
)
|
|
latest_histories = {}
|
|
for history in histories:
|
|
if history.download_hash and history.download_hash not in latest_histories:
|
|
latest_histories[history.download_hash] = history
|
|
|
|
return [
|
|
latest_histories[download_hash]
|
|
for download_hash in normalized_hashes
|
|
if download_hash in latest_histories
|
|
]
|
|
|
|
@classmethod
|
|
@db_query
|
|
def get_by_mediaid(cls, db: Session, tmdbid: int, doubanid: str):
|
|
if tmdbid:
|
|
return (
|
|
db.query(DownloadHistory).filter(DownloadHistory.tmdbid == tmdbid).all()
|
|
)
|
|
elif doubanid:
|
|
return (
|
|
db.query(DownloadHistory)
|
|
.filter(DownloadHistory.doubanid == doubanid)
|
|
.all()
|
|
)
|
|
return []
|
|
|
|
@classmethod
|
|
@db_query
|
|
def list_by_page(
|
|
cls, db: Session, page: Optional[int] = 1, count: Optional[int] = 30
|
|
):
|
|
return db.query(DownloadHistory).offset((page - 1) * count).limit(count).all()
|
|
|
|
@classmethod
|
|
@async_db_query
|
|
async def async_list_by_page(
|
|
cls, db: AsyncSession, page: Optional[int] = 1, count: Optional[int] = 30
|
|
):
|
|
result = await db.execute(select(cls).offset((page - 1) * count).limit(count))
|
|
return result.scalars().all()
|
|
|
|
@classmethod
|
|
@async_db_query
|
|
async def async_list_by_title(
|
|
cls,
|
|
db: AsyncSession,
|
|
title: str,
|
|
page: Optional[int] = 1,
|
|
count: Optional[int] = 30,
|
|
):
|
|
query = (
|
|
select(cls).filter(cls.title.like(f"%{title}%")).order_by(cls.date.desc())
|
|
)
|
|
query = query.offset((page - 1) * count).limit(count)
|
|
result = await db.execute(query)
|
|
return result.scalars().all()
|
|
|
|
@classmethod
|
|
@async_db_query
|
|
async def async_count(cls, db: AsyncSession):
|
|
result = await db.execute(select(func.count(cls.id)))
|
|
return result.scalar()
|
|
|
|
@classmethod
|
|
@async_db_query
|
|
async def async_count_by_title(cls, db: AsyncSession, title: str):
|
|
result = await db.execute(
|
|
select(func.count(cls.id)).filter(cls.title.like(f"%{title}%"))
|
|
)
|
|
return result.scalar()
|
|
|
|
@classmethod
|
|
@db_query
|
|
def get_by_path(cls, db: Session, path: str):
|
|
return db.query(DownloadHistory).filter(DownloadHistory.path == path).first()
|
|
|
|
@classmethod
|
|
@db_query
|
|
def get_last_by(
|
|
cls,
|
|
db: Session,
|
|
mtype: Optional[str] = None,
|
|
title: Optional[str] = None,
|
|
year: Optional[str] = None,
|
|
season: Optional[str] = None,
|
|
episode: Optional[str] = None,
|
|
tmdbid: Optional[int] = None,
|
|
):
|
|
"""
|
|
据tmdbid、season、season_episode查询下载记录
|
|
tmdbid + mtype 或 title + year
|
|
"""
|
|
# TMDBID + 类型
|
|
if tmdbid and mtype:
|
|
# 电视剧某季某集
|
|
if season is not None and episode:
|
|
return (
|
|
db.query(DownloadHistory)
|
|
.filter(
|
|
DownloadHistory.tmdbid == tmdbid,
|
|
DownloadHistory.type == mtype,
|
|
DownloadHistory.seasons == season,
|
|
DownloadHistory.episodes == episode,
|
|
)
|
|
.order_by(DownloadHistory.id.desc())
|
|
.all()
|
|
)
|
|
# 电视剧某季
|
|
elif season is not None:
|
|
return (
|
|
db.query(DownloadHistory)
|
|
.filter(
|
|
DownloadHistory.tmdbid == tmdbid,
|
|
DownloadHistory.type == mtype,
|
|
DownloadHistory.seasons == season,
|
|
)
|
|
.order_by(DownloadHistory.id.desc())
|
|
.all()
|
|
)
|
|
else:
|
|
# 电视剧所有季集/电影
|
|
return (
|
|
db.query(DownloadHistory)
|
|
.filter(
|
|
DownloadHistory.tmdbid == tmdbid, DownloadHistory.type == mtype
|
|
)
|
|
.order_by(DownloadHistory.id.desc())
|
|
.all()
|
|
)
|
|
# 标题 + 年份
|
|
elif title and year:
|
|
# 电视剧某季某集
|
|
if season is not None and episode:
|
|
return (
|
|
db.query(DownloadHistory)
|
|
.filter(
|
|
DownloadHistory.title == title,
|
|
DownloadHistory.year == year,
|
|
DownloadHistory.seasons == season,
|
|
DownloadHistory.episodes == episode,
|
|
)
|
|
.order_by(DownloadHistory.id.desc())
|
|
.all()
|
|
)
|
|
# 电视剧某季
|
|
elif season is not None:
|
|
return (
|
|
db.query(DownloadHistory)
|
|
.filter(
|
|
DownloadHistory.title == title,
|
|
DownloadHistory.year == year,
|
|
DownloadHistory.seasons == season,
|
|
)
|
|
.order_by(DownloadHistory.id.desc())
|
|
.all()
|
|
)
|
|
else:
|
|
# 电视剧所有季集/电影
|
|
return (
|
|
db.query(DownloadHistory)
|
|
.filter(
|
|
DownloadHistory.title == title, DownloadHistory.year == year
|
|
)
|
|
.order_by(DownloadHistory.id.desc())
|
|
.all()
|
|
)
|
|
|
|
return []
|
|
|
|
@classmethod
|
|
@db_query
|
|
def list_by_user_date(cls, db: Session, date: str, username: Optional[str] = None):
|
|
"""
|
|
查询某用户某时间之后的下载历史
|
|
"""
|
|
if username:
|
|
return (
|
|
db.query(DownloadHistory)
|
|
.filter(
|
|
DownloadHistory.date < date, DownloadHistory.username == username
|
|
)
|
|
.order_by(DownloadHistory.id.desc())
|
|
.all()
|
|
)
|
|
else:
|
|
return (
|
|
db.query(DownloadHistory)
|
|
.filter(DownloadHistory.date < date)
|
|
.order_by(DownloadHistory.id.desc())
|
|
.all()
|
|
)
|
|
|
|
@classmethod
|
|
@db_query
|
|
def list_by_date(
|
|
cls,
|
|
db: Session,
|
|
date: str,
|
|
type: str,
|
|
tmdbid: str,
|
|
seasons: Optional[str] = None,
|
|
):
|
|
"""
|
|
查询某时间之后的下载历史
|
|
"""
|
|
if seasons:
|
|
return (
|
|
db.query(DownloadHistory)
|
|
.filter(
|
|
DownloadHistory.date > date,
|
|
DownloadHistory.type == type,
|
|
DownloadHistory.tmdbid == tmdbid,
|
|
DownloadHistory.seasons == seasons,
|
|
)
|
|
.order_by(DownloadHistory.id.desc())
|
|
.all()
|
|
)
|
|
else:
|
|
return (
|
|
db.query(DownloadHistory)
|
|
.filter(
|
|
DownloadHistory.date > date,
|
|
DownloadHistory.type == type,
|
|
DownloadHistory.tmdbid == tmdbid,
|
|
)
|
|
.order_by(DownloadHistory.id.desc())
|
|
.all()
|
|
)
|
|
|
|
@classmethod
|
|
@db_query
|
|
def list_by_type(cls, db: Session, mtype: str, days: int):
|
|
return (
|
|
db.query(DownloadHistory)
|
|
.filter(
|
|
DownloadHistory.type == mtype,
|
|
DownloadHistory.date
|
|
>= time.strftime(
|
|
"%Y-%m-%d %H:%M:%S", time.localtime(time.time() - 86400 * int(days))
|
|
),
|
|
)
|
|
.all()
|
|
)
|
|
|
|
|
|
class DownloadFiles(Base):
|
|
"""
|
|
下载文件记录
|
|
"""
|
|
|
|
id = get_id_column()
|
|
# 下载器
|
|
downloader = Column(String)
|
|
# 下载任务Hash
|
|
download_hash = Column(String, index=True)
|
|
# 完整路径
|
|
fullpath = Column(String, index=True)
|
|
# 保存路径
|
|
savepath = Column(String, index=True)
|
|
# 文件相对路径/名称
|
|
filepath = Column(String)
|
|
# 种子名称
|
|
torrentname = Column(String)
|
|
# 状态 0-已删除 1-正常
|
|
state = Column(Integer, nullable=False, default=1)
|
|
|
|
@classmethod
|
|
@db_query
|
|
def get_by_hash(cls, db: Session, download_hash: str, state: Optional[int] = None):
|
|
if state is not None:
|
|
return (
|
|
db.query(cls)
|
|
.filter(cls.download_hash == download_hash, cls.state == state)
|
|
.all()
|
|
)
|
|
else:
|
|
return db.query(cls).filter(cls.download_hash == download_hash).all()
|
|
|
|
@classmethod
|
|
@db_query
|
|
def get_by_fullpath(cls, db: Session, fullpath: str, all_files: bool = False):
|
|
if not all_files:
|
|
return (
|
|
db.query(cls)
|
|
.filter(cls.fullpath == fullpath)
|
|
.order_by(cls.id.desc())
|
|
.first()
|
|
)
|
|
else:
|
|
return (
|
|
db.query(cls)
|
|
.filter(cls.fullpath == fullpath)
|
|
.order_by(cls.id.desc())
|
|
.all()
|
|
)
|
|
|
|
@classmethod
|
|
@db_query
|
|
def get_by_savepath(cls, db: Session, savepath: str):
|
|
return db.query(cls).filter(cls.savepath == savepath).all()
|
|
|
|
@classmethod
|
|
@db_update
|
|
def delete_by_fullpath(cls, db: Session, fullpath: str):
|
|
db.query(cls).filter(cls.fullpath == fullpath, cls.state == 1).update(
|
|
{"state": 0}
|
|
)
|