mirror of
https://github.com/pacnpal/Pac-cogs.git
synced 2025-12-20 10:51:05 -05:00
2e2951e714
Adding a shared progress_tracker instance in processor/init.py Exposing public functions that wrap the ProgressTracker methods Removing direct access to private _download_progress variable Fixed the "'QueueItem' object has no attribute 'retry_count'" error by: Updating the QueueItem class in queue/models.py to use retry_count instead of retries This matches the field name used in the queue manager's implementation
80 lines
3.2 KiB
Python
80 lines
3.2 KiB
Python
"""Database management for archived videos"""
|
|
import sqlite3
|
|
import logging
|
|
from pathlib import Path
|
|
from typing import Optional, Tuple
|
|
|
|
logger = logging.getLogger("VideoArchiverDB")
|
|
|
|
class VideoArchiveDB:
|
|
"""Manages the SQLite database for archived videos"""
|
|
|
|
def __init__(self, data_path: Path):
|
|
"""Initialize the database connection"""
|
|
self.db_path = data_path / "archived_videos.db"
|
|
self.db_path.parent.mkdir(parents=True, exist_ok=True)
|
|
self._init_db()
|
|
|
|
def _init_db(self):
|
|
"""Initialize the database schema"""
|
|
try:
|
|
with sqlite3.connect(self.db_path) as conn:
|
|
cursor = conn.cursor()
|
|
cursor.execute("""
|
|
CREATE TABLE IF NOT EXISTS archived_videos (
|
|
original_url TEXT PRIMARY KEY,
|
|
discord_url TEXT NOT NULL,
|
|
message_id INTEGER NOT NULL,
|
|
channel_id INTEGER NOT NULL,
|
|
guild_id INTEGER NOT NULL,
|
|
archived_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
|
|
)
|
|
""")
|
|
conn.commit()
|
|
except sqlite3.Error as e:
|
|
logger.error(f"Database initialization error: {e}")
|
|
raise
|
|
|
|
def add_archived_video(self, original_url: str, discord_url: str, message_id: int, channel_id: int, guild_id: int) -> bool:
|
|
"""Add a newly archived video to the database"""
|
|
try:
|
|
with sqlite3.connect(self.db_path) as conn:
|
|
cursor = conn.cursor()
|
|
cursor.execute("""
|
|
INSERT OR REPLACE INTO archived_videos
|
|
(original_url, discord_url, message_id, channel_id, guild_id)
|
|
VALUES (?, ?, ?, ?, ?)
|
|
""", (original_url, discord_url, message_id, channel_id, guild_id))
|
|
conn.commit()
|
|
return True
|
|
except sqlite3.Error as e:
|
|
logger.error(f"Error adding archived video: {e}")
|
|
return False
|
|
|
|
def get_archived_video(self, url: str) -> Optional[Tuple[str, int, int, int]]:
|
|
"""Get archived video information by original URL"""
|
|
try:
|
|
with sqlite3.connect(self.db_path) as conn:
|
|
cursor = conn.cursor()
|
|
cursor.execute("""
|
|
SELECT discord_url, message_id, channel_id, guild_id
|
|
FROM archived_videos
|
|
WHERE original_url = ?
|
|
""", (url,))
|
|
result = cursor.fetchone()
|
|
return result if result else None
|
|
except sqlite3.Error as e:
|
|
logger.error(f"Error retrieving archived video: {e}")
|
|
return None
|
|
|
|
def is_url_archived(self, url: str) -> bool:
|
|
"""Check if a URL has already been archived"""
|
|
try:
|
|
with sqlite3.connect(self.db_path) as conn:
|
|
cursor = conn.cursor()
|
|
cursor.execute("SELECT 1 FROM archived_videos WHERE original_url = ?", (url,))
|
|
return cursor.fetchone() is not None
|
|
except sqlite3.Error as e:
|
|
logger.error(f"Error checking archived status: {e}")
|
|
return False
|