diff --git a/VERSION b/VERSION index fe16b34..0cadbc1 100644 --- a/VERSION +++ b/VERSION @@ -1 +1 @@ -2.5.4 +2.5.5 diff --git a/processors/tv_processor.py b/processors/tv_processor.py index c35c728..b4d5c24 100644 --- a/processors/tv_processor.py +++ b/processors/tv_processor.py @@ -69,33 +69,22 @@ class TVProcessor: with self.db.get_connection() as conn: cursor = conn.cursor() - if self.db.db_type == "postgresql": - cursor.execute(""" - SELECT - COUNT(*) as total_in_db, - COUNT(CASE WHEN dateadded IS NOT NULL AND source IS NOT NULL AND source != 'unknown' AND source != 'no_valid_date_source' THEN 1 END) as complete_episodes - FROM episodes - WHERE imdb_id = %s - """, (imdb_id,)) - else: - cursor.execute(""" - SELECT - COUNT(*) as total_in_db, - COUNT(CASE WHEN dateadded IS NOT NULL AND dateadded != '' AND source IS NOT NULL AND source != 'unknown' AND source != 'no_valid_date_source' THEN 1 END) as complete_episodes - FROM episodes - WHERE imdb_id = ? - """, (imdb_id,)) + # PostgreSQL-only query + cursor.execute(""" + SELECT + COUNT(*) as total_in_db, + COUNT(CASE WHEN dateadded IS NOT NULL AND source IS NOT NULL AND source != 'unknown' AND source != 'no_valid_date_source' THEN 1 END) as complete_episodes + FROM episodes + WHERE imdb_id = %s + """, (imdb_id,)) result = cursor.fetchone() if not result: return False, "No database records found" - if self.db.db_type == "postgresql": - total_in_db = result['total_in_db'] - complete_episodes = result['complete_episodes'] - else: - total_in_db = result[0] - complete_episodes = result[1] + # PostgreSQL RealDictCursor returns dict-like objects + total_in_db = result['total_in_db'] + complete_episodes = result['complete_episodes'] # Skip if: # 1. We have episodes in database