fix: collector torrent not added bug.

This commit is contained in:
EstrellaXD
2023-10-03 22:46:58 +08:00
parent f82ea902e6
commit 8e5d7d2e02
2 changed files with 7 additions and 0 deletions

View File

@@ -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)