tubearchivist/tubearchivist-frontend
1
0
Fork 0
mirror of https://github.com/tubearchivist/tubearchivist-frontend.git synced 2024-11-22 20:00:15 +00:00
Code Issues Projects Releases Wiki Activity

fix killing task with missing task id

Browse Source
This commit is contained in:
simon 2021-10-11 15:26:31 +07:00
parent e202265a5f
commit 567e9d473b
2 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tubearchivist/home/tasks.py
View File

@ -138,7 +138,9 @@ def run_restore_backup():
def kill_dl(task_id): def kill_dl(task_id):
"""kill download worker task by ID""" """kill download worker task by ID"""
if task_id:
app.control.revoke(task_id, terminate=True) app.control.revoke(task_id, terminate=True)
_ = RedisArchivist().del_message("dl_queue_id") _ = RedisArchivist().del_message("dl_queue_id")
RedisQueue("dl_queue").clear() RedisQueue("dl_queue").clear()

3
tubearchivist/home/views.py
View File

@ -588,6 +588,9 @@ class PostData:
RedisQueue("dl_queue").clear() RedisQueue("dl_queue").clear()
elif to_execute == "kill": elif to_execute == "kill":
task_id = RedisArchivist().get_message("dl_queue_id") task_id = RedisArchivist().get_message("dl_queue_id")
if not task_id["status"]:
task_id = False
else:
print("brutally killing " + task_id) print("brutally killing " + task_id)
kill_dl(task_id) kill_dl(task_id)