From dae8bbb6b42016b070e847f26ba113d082d8b8e8 Mon Sep 17 00:00:00 2001
From: pacnpal <183241239+pacnpal@users.noreply.github.com>
Date: Fri, 15 Nov 2024 19:22:52 +0000
Subject: [PATCH] Adding the 'db' parameter to the VideoProcessor class
 initialization Properly storing and handling the db instance Ensuring the db
 is passed to the queue_handler when available The QueueHandler class was
 already properly set up to handle the db parameter, so no changes were needed
 there. The database integration should now work correctly for:

Checking if videos are already archived
Adding newly archived videos to the database
Providing archived video information to users
---
 videoarchiver/processor/core.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/videoarchiver/processor/core.py b/videoarchiver/processor/core.py
index bf8a540..678cd83 100644
--- a/videoarchiver/processor/core.py
+++ b/videoarchiver/processor/core.py
@@ -23,18 +23,24 @@ class VideoProcessor:
         config_manager,
         components,
         queue_manager=None,
-        ffmpeg_mgr=None
+        ffmpeg_mgr=None,
+        db=None
     ):
         self.bot = bot
         self.config = config_manager
         self.components = components
         self.ffmpeg_mgr = ffmpeg_mgr
+        self.db = db
 
         # Initialize handlers
         self.queue_handler = QueueHandler(bot, config_manager, components)
         self.message_handler = MessageHandler(bot, config_manager, queue_manager)
         self.progress_tracker = ProgressTracker()
 
+        # Pass db to queue handler if it exists
+        if self.db:
+            self.queue_handler.db = self.db
+
         # Start queue processing
         logger.info("Starting video processing queue...")
         self._queue_task = None