From 2546b4e780f89349a7c6ee2053d872ddf27be1c4 Mon Sep 17 00:00:00 2001 From: Blondel MONDESIR Date: Sun, 21 Jul 2024 07:53:57 -0400 Subject: [PATCH] Ensure error column is created --- cps/tasks/metadata_extract.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/cps/tasks/metadata_extract.py b/cps/tasks/metadata_extract.py index a946790ba9..fa3fd183a8 100644 --- a/cps/tasks/metadata_extract.py +++ b/cps/tasks/metadata_extract.py @@ -63,9 +63,11 @@ def _fetch_requested_urls(self, conn): try: cursor = conn.execute("PRAGMA table_info(media)") self.columns = [column[1] for column in cursor.fetchall()] - query = ("SELECT path, duration, live_status FROM media WHERE error IS NULL AND path LIKE 'http%'" - if "error" in self.columns - else "SELECT path, duration, live_status FROM media WHERE path LIKE 'http%'") + query = "SELECT path, duration, live_status FROM media WHERE path LIKE 'http%'" + if "error" in self.columns: + query += " AND error IS NULL" + else: + conn.execute("ALTER TABLE media ADD COLUMN error TEXT") rows = conn.execute(query).fetchall() requested_urls = {} for path, duration, live_status in rows: