tubearchivist/tubearchivist
1
0
Fork 0
mirror of https://github.com/tubearchivist/tubearchivist.git synced 2024-12-23 18:30:12 +00:00
Code Issues Projects Releases Wiki Activity

implement api-stop for download_pending

Browse Source
This commit is contained in:
simon 2023-03-23 10:48:43 +07:00
parent 860213c427
commit 33b9a012d4
No known key found for this signature in database
GPG Key ID: 2C15AA5E89985DD4
3 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tubearchivist/home/src/download/yt_dlp_handler.py
View File

@ -174,7 +174,7 @@ class VideoDownloader:
while True: while True:
youtube_data = queue.get_next() youtube_data = queue.get_next()
if not youtube_data: if self.task.is_stopped() or not youtube_data:
break break
youtube_data = json.loads(youtube_data) youtube_data = json.loads(youtube_data)

1
tubearchivist/home/src/ta/task_manager.py
View File

@ -43,7 +43,6 @@ class TaskManager:
def is_stopped(self, task_id): def is_stopped(self, task_id):
"""check if task_id has received STOP command""" """check if task_id has received STOP command"""
task = self.get_task(task_id) task = self.get_task(task_id)
print(task)
return task.get("command") == "STOP" return task.get("command") == "STOP"

1
tubearchivist/home/tasks.py
View File

@ -59,6 +59,7 @@ class BaseTask(Task):
"title": "Downloading", "title": "Downloading",
"group": "download:run", "group": "download:run",
"api-start": True, "api-start": True,
"api-stop": True,
}, },
"extract_download": { "extract_download": {
"title": "Add to download queue", "title": "Add to download queue",