Queue can be emptied more easily

This commit is contained in:
Simon 2018-08-09 17:12:43 -04:00
parent ffeed4192e
commit 5a084cb857

View File

@ -407,12 +407,12 @@ class Database:
cursor = conn.cursor() cursor = conn.cursor()
cursor.execute("SELECT id, website_id, url, priority, callback_type, callback_args FROM " cursor.execute("SELECT id, website_id, url, priority, callback_type, callback_args FROM "
"Queue WHERE website_id=? AND assigned_crawler=?", (website_id, name)) "Queue WHERE website_id=?", (website_id, name))
task = cursor.fetchone() task = cursor.fetchone()
if task: if task:
cursor.execute("DELETE FROM Queue WHERE website_id=? AND assigned_crawler=?", (website_id, name)) cursor.execute("DELETE FROM Queue WHERE website_id=?", (website_id, name))
conn.commit() conn.commit()
return Task(task[1], task[2], task[3], task[4], task[5]) return Task(task[1], task[2], task[3], task[4], task[5])
else: else: