sbcrumb/nfoguard
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

db: fixes
Local Docker Build (Dev) / build-dev (push) Successful in 4s
Details

Browse Source
This commit is contained in:
sbcrumb
2025-10-19 10:28:32 -04:00
parent d1d1e1a3d1
commit 3c4cac6177
2 changed files with 12 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files
processors/tv_processor.py
+11 -22
View File
@@ -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