update

database.py

@@ -16,7 +16,7 @@ class Database:
         if not file_exists:
             self.create_database()
 
-    def __del__(self):
+    def __exit__(self):
         # Destructors are unpredictable in Python; explicitly close when done instead
         try:
             self.close_database()
@@ -102,11 +102,20 @@ class Database:
         self.cursor.execute(f"SELECT {self.columns} FROM twitch_videos WHERE downloaded = 0")
         return self.cursor.fetchall()
 
+    def get_download(self) -> list[Any]:
+        """Retrieves all rows that were downloaded"""
+        self.cursor.execute(f"SELECT {self.columns} FROM twitch_videos WHERE downloaded = 1")
+        return self.cursor.fetchall()
+
     def get_clips(self) -> list[Any]:
         """Retrieves all rows that are clip_is"""
         self.cursor.execute(f"SELECT {self.columns} FROM twitch_videos WHERE clip_is = 1")
         return self.cursor.fetchall()
 
+    def get_vods(self) -> list[Any]:
+        self.cursor.execute(f"SELECT {self.columns} FROM twitch_videos WHERE clip_is = 0")
+        return self.cursor.fetchall()
+
     def insert_video_record(
         self, id: str, title: str, created_at: str, view_count: int, duration: str, 
         url: str, thumbnail_url: str, game_id: int, game_name: str, stream_id: str, 
@@ -120,10 +129,23 @@ class Database:
 
         # Explicitly defining columns removes the security risk and column-count bug
         query = """
-            INSERT OR IGNORE INTO twitch_videos (
+            INSERT INTO twitch_videos (
                 id, title, created_at, view_count, duration, url, thumbnail_url, 
                 game_id, game_name, stream_id, creator_name, clip_is
             ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+            ON CONFLICT(id) DO UPDATE SET
+                title = excluded.title,
+                created_at = excluded.created_at,
+                view_count = excluded.view_count,
+                duration = excluded.duration,
+                url = excluded.url,
+                thumbnail_url = excluded.thumbnail_url,
+                game_id = excluded.game_id,
+                game_name = excluded.game_name,
+                stream_id = excluded.stream_id,
+                creator_name = excluded.creator_name,
+                clip_is = excluded.clip_is
+            WHERE excluded.duration != twitch_videos.duration
         """
 
         values = (