mirror of
https://github.com/EstrellaXD/Auto_Bangumi.git
synced 2026-04-02 10:10:32 +08:00
fix: collector torrent not added bug.
This commit is contained in:
@@ -14,9 +14,14 @@ class BangumiDatabase:
|
||||
self.session = session
|
||||
|
||||
def add(self, data: Bangumi):
|
||||
statement = select(Bangumi).where(Bangumi.title_raw == data.title_raw)
|
||||
bangumi = self.session.exec(statement).first()
|
||||
if bangumi:
|
||||
return False
|
||||
self.session.add(data)
|
||||
self.session.commit()
|
||||
logger.debug(f"[Database] Insert {data.official_title} into database.")
|
||||
return True
|
||||
|
||||
def add_all(self, datas: list[Bangumi]):
|
||||
self.session.add_all(datas)
|
||||
|
||||
@@ -22,6 +22,8 @@ class SeasonCollector(DownloadClient):
|
||||
logger.info(
|
||||
f"Collections of {bangumi.official_title} Season {bangumi.season} completed."
|
||||
)
|
||||
for torrent in torrents:
|
||||
torrent.downloaded = True
|
||||
bangumi.eps_collect = True
|
||||
if engine.bangumi.update(bangumi):
|
||||
engine.bangumi.add(bangumi)
|
||||
|
||||
Reference in New Issue
Block a user