From 652924eb8cf6d82aec90eb514628b3c51849ab98 Mon Sep 17 00:00:00 2001
From: Spoked <dreu.lavelle@gmail.com>
Date: Sat, 21 Sep 2024 00:43:14 -0400
Subject: [PATCH] fix: tweak db reset. fixed issue with mdblist.

---
 src/program/content/mdblist.py |  3 +-
 src/program/db/db_functions.py | 65 ++++++++++++++++++++++++----------
 2 files changed, 48 insertions(+), 20 deletions(-)

diff --git a/src/program/content/mdblist.py b/src/program/content/mdblist.py
index c3e0868d..96557ef1 100644
--- a/src/program/content/mdblist.py
+++ b/src/program/content/mdblist.py
@@ -54,8 +54,7 @@ def run(self) -> Generator[MediaItem, None, None]:
                     else:
                         items = list_items_by_url(list, self.settings.api_key)
                     for item in items:
-                        # Check if the item is already completed in the media container
-                        if item.imdb_id:
+                        if hasattr(item, "imdb_id") and item.imdb_id.startswith("tt"):
                             items_to_yield.append(MediaItem(
                                 {"imdb_id": item.imdb_id, "requested_by": self.key}
                             ))
diff --git a/src/program/db/db_functions.py b/src/program/db/db_functions.py
index 5c71fc0a..2598b8e3 100644
--- a/src/program/db/db_functions.py
+++ b/src/program/db/db_functions.py
@@ -386,34 +386,63 @@ def _run_thread_with_db_item(fn, service, program, input_item: "MediaItem" = Non
 
 def hard_reset_database():
     """Resets the database to a fresh state."""
-    logger.log("DATABASE", "Resetting Database")
+    logger.log("DATABASE", "Starting Hard Reset of Database")
 
-    # Drop all tables
-    db.Model.metadata.drop_all(db.engine)
-    logger.log("DATABASE","All MediaItem tables dropped")
-
-    # Drop the alembic_version table
+    # Disable foreign key checks temporarily
     with db.engine.connect() as connection:
-        connection.execute(text("DROP TABLE IF EXISTS alembic_version CASCADE"))
-    logger.log("DATABASE","Alembic table dropped")
+        if db.engine.name == 'sqlite':
+            connection.execute(text("PRAGMA foreign_keys = OFF"))
+        elif db.engine.name == 'postgresql':
+            connection.execute(text("SET CONSTRAINTS ALL DEFERRED"))
+
+        try:
+            for table in reversed(db.Model.metadata.sorted_tables):
+                try:
+                    table.drop(connection, checkfirst=True)
+                    logger.log("DATABASE", f"Dropped table: {table.name}")
+                except Exception as e:
+                    logger.log("DATABASE", f"Error dropping table {table.name}: {str(e)}")
+
+            try:
+                connection.execute(text("DROP TABLE IF EXISTS alembic_version"))
+                logger.log("DATABASE", "Alembic version table dropped")
+            except Exception as e:
+                logger.log("DATABASE", f"Error dropping alembic_version table: {str(e)}")
+
+            db.Model.metadata.create_all(connection)
+            logger.log("DATABASE", "All tables recreated")
+
+            # Re-enable foreign key checks
+            if db.engine.name == 'sqlite':
+                connection.execute(text("PRAGMA foreign_keys = ON"))
+            elif db.engine.name == 'postgresql':
+                connection.execute(text("SET CONSTRAINTS ALL IMMEDIATE"))
+
+            connection.commit()
+        except Exception as e:
+            connection.rollback()
+            logger.log("DATABASE", f"Error during database reset: {str(e)}")
+            raise
 
-    # Recreate all tables
-    db.Model.metadata.create_all(db.engine)
-    logger.log("DATABASE","All tables recreated")
+    try:
+        logger.log("DATABASE", "Removing Alembic Directory")
+        shutil.rmtree(alembic_dir, ignore_errors=True)
+        os.makedirs(alembic_dir, exist_ok=True)
+        alembic.init(alembic_dir)
+        logger.log("DATABASE", "Alembic reinitialized")
+    except Exception as e:
+        logger.log("DATABASE", f"Error reinitializing Alembic: {str(e)}")
 
-    # Reinitialize Alembic
-    logger.log("DATABASE","Removing Alembic Directory")
-    shutil.rmtree(alembic_dir, ignore_errors=True)
-    os.makedirs(alembic_dir, exist_ok=True)
-    alembic.init(alembic_dir)
-    logger.log("DATABASE","Alembic reinitialized")
+    logger.log("DATABASE", "Hard Reset Complete")
 
-    logger.log("DATABASE","Hard Reset Complete")
 
+# Hard Reset Database
 reset = os.getenv("HARD_RESET", None)
 if reset is not None and reset.lower() in ["true","1"]:
     hard_reset_database()
+    exit(0)
 
+# Repair Symlinks
 if os.getenv("REPAIR_SYMLINKS", None) is not None and os.getenv("REPAIR_SYMLINKS").lower() in ["true","1"]:
     fix_broken_symlinks(settings_manager.settings.symlink.library_path, settings_manager.settings.symlink.rclone_path)
     exit(0)
\ No newline at end of file