mirror of
https://github.com/EstrellaXD/Auto_Bangumi.git
synced 2026-03-20 03:46:40 +08:00
6cbe7090fe
When a bangumi was deleted, its associated Torrent records remained in the database. This prevented re-downloading the same torrents if the user re-added the anime, because check_new() deduplicates by URL and would filter out the orphaned records. Now delete_rule() removes Torrent records before deleting the Bangumi, so re-adding the same anime correctly treats those torrents as new. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
104 lines
3.7 KiB
Python
104 lines
3.7 KiB
Python
import logging
|
|
|
|
from sqlmodel import Session, select
|
|
|
|
from module.models import Torrent
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class TorrentDatabase:
|
|
def __init__(self, session: Session):
|
|
self.session = session
|
|
|
|
def add(self, data: Torrent):
|
|
self.session.add(data)
|
|
self.session.commit()
|
|
logger.debug("Insert %s in database.", data.name)
|
|
|
|
def add_all(self, datas: list[Torrent]):
|
|
self.session.add_all(datas)
|
|
self.session.commit()
|
|
logger.debug("Insert %s torrents in database.", len(datas))
|
|
|
|
def update(self, data: Torrent):
|
|
self.session.add(data)
|
|
self.session.commit()
|
|
logger.debug("Update %s in database.", data.name)
|
|
|
|
def update_all(self, datas: list[Torrent]):
|
|
self.session.add_all(datas)
|
|
self.session.commit()
|
|
|
|
def update_one_user(self, data: Torrent):
|
|
self.session.add(data)
|
|
self.session.commit()
|
|
logger.debug("Update %s in database.", data.name)
|
|
|
|
def search(self, _id: int) -> Torrent | None:
|
|
result = self.session.execute(select(Torrent).where(Torrent.id == _id))
|
|
return result.scalar_one_or_none()
|
|
|
|
def search_all(self) -> list[Torrent]:
|
|
result = self.session.execute(select(Torrent))
|
|
return list(result.scalars().all())
|
|
|
|
def search_rss(self, rss_id: int) -> list[Torrent]:
|
|
result = self.session.execute(select(Torrent).where(Torrent.rss_id == rss_id))
|
|
return list(result.scalars().all())
|
|
|
|
def check_new(self, torrents_list: list[Torrent]) -> list[Torrent]:
|
|
if not torrents_list:
|
|
return []
|
|
urls = [t.url for t in torrents_list]
|
|
statement = select(Torrent.url).where(Torrent.url.in_(urls))
|
|
result = self.session.execute(statement)
|
|
existing_urls = set(result.scalars().all())
|
|
return [t for t in torrents_list if t.url not in existing_urls]
|
|
|
|
def search_by_qb_hash(self, qb_hash: str) -> Torrent | None:
|
|
"""Find torrent by qBittorrent hash."""
|
|
result = self.session.execute(select(Torrent).where(Torrent.qb_hash == qb_hash))
|
|
return result.scalar_one_or_none()
|
|
|
|
def search_by_qb_hashes(self, qb_hashes: list[str]) -> list[Torrent]:
|
|
"""Find torrents by multiple qBittorrent hashes (batch query)."""
|
|
if not qb_hashes:
|
|
return []
|
|
result = self.session.execute(
|
|
select(Torrent).where(Torrent.qb_hash.in_(qb_hashes))
|
|
)
|
|
return list(result.scalars().all())
|
|
|
|
def delete_by_bangumi_id(self, bangumi_id: int) -> int:
|
|
"""Delete all torrent records associated with a bangumi.
|
|
|
|
Returns the number of deleted records.
|
|
"""
|
|
statement = select(Torrent).where(Torrent.bangumi_id == bangumi_id)
|
|
result = self.session.execute(statement)
|
|
torrents = list(result.scalars().all())
|
|
count = len(torrents)
|
|
for t in torrents:
|
|
self.session.delete(t)
|
|
if count > 0:
|
|
self.session.commit()
|
|
logger.debug("Deleted %s torrent records for bangumi_id %s.", count, bangumi_id)
|
|
return count
|
|
|
|
def search_by_url(self, url: str) -> Torrent | None:
|
|
"""Find torrent by URL."""
|
|
result = self.session.execute(select(Torrent).where(Torrent.url == url))
|
|
return result.scalar_one_or_none()
|
|
|
|
def update_qb_hash(self, torrent_id: int, qb_hash: str) -> bool:
|
|
"""Update the qb_hash for a torrent."""
|
|
torrent = self.search(torrent_id)
|
|
if torrent:
|
|
torrent.qb_hash = qb_hash
|
|
self.session.add(torrent)
|
|
self.session.commit()
|
|
logger.debug("Updated qb_hash for torrent %s: %s", torrent_id, qb_hash)
|
|
return True
|
|
return False
|