From 42fba5391abdc0abb3efd95822911a3a2550528c Mon Sep 17 00:00:00 2001 From: t0is Date: Sat, 5 Apr 2025 15:25:04 +0200 Subject: [PATCH] download edit --- download_only.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/download_only.py b/download_only.py index 8a1353f..3b9408e 100644 --- a/download_only.py +++ b/download_only.py @@ -74,7 +74,10 @@ def db_set_col(db, video_id, column, value=True): Also updates the updated_at timestamp. """ cursor = db.cursor() - query = f"UPDATE videos SET {column} = %s WHERE id = %s" + if column == "data_downloaded": + query = f"UPDATE videos SET {column} = %s, download_end=NOW() WHERE id = %s" + else: + query = f"UPDATE videos SET {column} = %s WHERE id = %s" cursor.execute(query, (value, video_id)) db.commit() cursor.close() @@ -88,7 +91,7 @@ def try_lock_video(db, video_id): cursor = db.cursor() query = """ UPDATE videos - SET data_downloading = 1, updated_at = NOW() + SET data_downloading = 1, updated_at = NOW(), download_start = NOW() WHERE id = %s AND data_downloading = 0 """ cursor.execute(query, (video_id,))