simon987/od-database
1
0
Fork 0
mirror of https://github.com/simon987/od-database.git synced 2025-12-13 23:09:01 +00:00
Code Issues Projects Releases Wiki Activity

hotfix attempt 2

Browse Source
This commit is contained in:
simon987
2019-02-02 10:58:09 -05:00
parent 8ced4859f3
commit 0ff6ea1682
2 changed files with 5 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
database.py
View File

@@ -373,15 +373,15 @@ class Database:
conn.commit()
def put_task(self, task: Task) -> None:
def put_task(self, task: Task, assigned_crawler=None) -> None:
with sqlite3.connect(self.db_path) as conn:
cursor = conn.cursor()
cursor.execute("INSERT INTO Queue (website_id, url, priority, callback_type, callback_args) "
"VALUES (?,?,?,?,?)",
cursor.execute("INSERT INTO Queue (website_id, url, priority, callback_type, callback_args, assigned_crawler) "
"VALUES (?,?,?,?,?,?)",
(task.website_id, task.url, task.priority,
task.callback_type, json.dumps(task.callback_args)))
task.callback_type, json.dumps(task.callback_args), assigned_crawler))
conn.commit()
def get_tasks(self) -> list: