From 900b47184ce8125daf47b61a20e079f966274bdc Mon Sep 17 00:00:00 2001
From: Andreas <andreasv@fsmpi.rwth-aachen.de>
Date: Wed, 18 Jul 2018 22:29:38 +0200
Subject: [PATCH] cleaned up schedule functions

---
 jobmanagement.py | 8 ++++----
 jobtypes.py      | 4 ++--
 2 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/jobmanagement.py b/jobmanagement.py
index 00c3587..0136bda 100644
--- a/jobmanagement.py
+++ b/jobmanagement.py
@@ -51,11 +51,11 @@ def schedule_job(jobtype, data=None, priority=0, queue="default"):
 	return modify('INSERT INTO jobs (type, priority, queue, data, time_created) VALUES (?, ?, ?, ?, ?)',
 			jobtype, priority, queue, json.dumps(data, default=date_json_handler), datetime.now())
 def cancel_job(job_id):
-	modify('UPDATE jobs SET state = "deleted" WHERE id = ? AND state = "ready"', job_id)
-	modify('UPDATE jobs SET canceled = 1 WHERE id = ?', job_id)
+	query('UPDATE jobs SET state = "deleted" WHERE id = ? AND state = "ready"', job_id)
+	query('UPDATE jobs SET canceled = 1 WHERE id = ?', job_id)
 
 def restart_job(job_id, canceled=False):
 	if canceled:
-		modify('UPDATE jobs SET state = "ready", canceled = 0 WHERE id = ? AND state = "failed"', job_id)
+		query('UPDATE jobs SET state = "ready", canceled = 0 WHERE id = ? AND state = "failed"', job_id)
 	else:
-		modify('UPDATE jobs SET state = "ready" WHERE id = ? AND state = "failed" AND NOT canceled', job_id)
+		query('UPDATE jobs SET state = "ready" WHERE id = ? AND state = "failed" AND NOT canceled', job_id)
diff --git a/jobtypes.py b/jobtypes.py
index e897608..8d11213 100644
--- a/jobtypes.py
+++ b/jobtypes.py
@@ -19,7 +19,7 @@ def schedule_thumbnail(lectureid):
 		JOIN formats ON (videos.video_format = formats.id)
 		WHERE videos.lecture_id = ?
 		ORDER BY formats.prio DESC''', lectureid)
-	schedule_job('thumbnail', {'lectureid': str(lectureid), 'path': videos[0]['path']})
+	return schedule_job('thumbnail', {'lectureid': str(lectureid), 'path': videos[0]['path']})
 
 
 def schedule_remux(lectureid, videoid=None):
@@ -73,5 +73,5 @@ def schedule_transcode(source, fmt_id=None, video=None):
 		data['lecture_id'] = lecture['id']
 	data['format_id'] = fmt['id']
 	data['source_id'] = source['id']
-	schedule_job('transcode', data, queue="background")
+	return schedule_job('transcode', data, queue="background")
 
-- 
GitLab