@@ -69,7 +69,7 @@ class TVProcessor:
|
|||||||
with self.db.get_connection() as conn:
|
with self.db.get_connection() as conn:
|
||||||
cursor = conn.cursor()
|
cursor = conn.cursor()
|
||||||
|
|
||||||
if self.db.db_type == "postgresql":
|
# PostgreSQL-only query
|
||||||
cursor.execute("""
|
cursor.execute("""
|
||||||
SELECT
|
SELECT
|
||||||
COUNT(*) as total_in_db,
|
COUNT(*) as total_in_db,
|
||||||
@@ -77,25 +77,14 @@ class TVProcessor:
|
|||||||
FROM episodes
|
FROM episodes
|
||||||
WHERE imdb_id = %s
|
WHERE imdb_id = %s
|
||||||
""", (imdb_id,))
|
""", (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,))
|
|
||||||
|
|
||||||
result = cursor.fetchone()
|
result = cursor.fetchone()
|
||||||
if not result:
|
if not result:
|
||||||
return False, "No database records found"
|
return False, "No database records found"
|
||||||
|
|
||||||
if self.db.db_type == "postgresql":
|
# PostgreSQL RealDictCursor returns dict-like objects
|
||||||
total_in_db = result['total_in_db']
|
total_in_db = result['total_in_db']
|
||||||
complete_episodes = result['complete_episodes']
|
complete_episodes = result['complete_episodes']
|
||||||
else:
|
|
||||||
total_in_db = result[0]
|
|
||||||
complete_episodes = result[1]
|
|
||||||
|
|
||||||
# Skip if:
|
# Skip if:
|
||||||
# 1. We have episodes in database
|
# 1. We have episodes in database
|
||||||
|
|||||||
Reference in New Issue
Block a user