From 4c94a51d57727be9967ef9a037dc4ae7067766ac Mon Sep 17 00:00:00 2001
From: EstrellaXD <estrellaxd05@gmail.com>
Date: Thu, 15 Jun 2023 08:55:22 +0800
Subject: [PATCH] fix: lost poster problem

---
 src/module/database/bangumi.py          | 28 ++++++-------------------
 src/module/notification/notification.py |  3 ++-
 2 files changed, 8 insertions(+), 23 deletions(-)

diff --git a/src/module/database/bangumi.py b/src/module/database/bangumi.py
index f149bd52..1c49938b 100644
--- a/src/module/database/bangumi.py
+++ b/src/module/database/bangumi.py
@@ -131,14 +131,14 @@ class BangumiDatabase(DataConnector):
     def match_poster(self, bangumi_name: str) -> str:
         data = self._cursor.execute(
             """
-            SELECT official_title, poster_link FROM bangumi
-            WHERE INSTR(:title_raw, title_raw) > 0
+            SELECT poster_link FROM bangumi
+            WHERE INSTR(official_title, :official_title) > 0
             """,
-            {"title_raw": bangumi_name},
+            {"official_title": bangumi_name},
         ).fetchone()
         if not data:
             return ""
-        official_title, poster_link = data
+        poster_link = data[0]
         if not poster_link:
             return ""
         return poster_link
@@ -219,21 +219,5 @@ class BangumiDatabase(DataConnector):
 
 if __name__ == '__main__':
     with BangumiDatabase() as db:
-        db.update_table()
-        data = BangumiData(
-            id=1,
-            title_raw="",
-            official_title="",
-            poster_link="",
-            rss_link=[],
-            eps_collect=0,
-            eps_total=0,
-            eps_update=0,
-            eps_complete=False,
-            added=False,
-            rule_name=None,
-            save_path=None,
-        )
-        db.insert(data)
-        print(db.not_added())
-        print(db.not_complete())
\ No newline at end of file
+        name = "久保"
+        print(db.match_poster(name))
\ No newline at end of file
diff --git a/src/module/notification/notification.py b/src/module/notification/notification.py
index 1e13ff67..a59230b7 100644
--- a/src/module/notification/notification.py
+++ b/src/module/notification/notification.py
@@ -63,9 +63,10 @@ class PostNotification:
     def __exit__(self, exc_type, exc_val, exc_tb):
         self.notifier.__exit__(exc_type, exc_val, exc_tb)
 
+
 if __name__ == "__main__":
     info = Notification(
-        official_title="魔法纪录 魔法少女小圆外传",
+        official_title="久保同学不放过我",
         season=2,
         episode=1,
     )