From 54f9b49e90db46a7725e9db52e8f6a7f07048e6c Mon Sep 17 00:00:00 2001 From: HelloZeroNet Date: Sat, 19 Mar 2016 18:09:20 +0100 Subject: [PATCH] Wait a bit for more tasks before consider site downloaded --- src/Worker/WorkerManager.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/src/Worker/WorkerManager.py b/src/Worker/WorkerManager.py index 461a5a3c..50c7f710 100644 --- a/src/Worker/WorkerManager.py +++ b/src/Worker/WorkerManager.py @@ -391,6 +391,14 @@ class WorkerManager: if not self.tasks: self.started_task_num = 0 + # Wait for other tasks + def checkComplete(self): + time.sleep(0.1) + if not self.tasks: + self.log.debug("Check compelte: No tasks") + self.started_task_num = 0 + self.site.onComplete() # No more task trigger site complete + # Mark a task done def doneTask(self, task): task["done"] = True @@ -398,5 +406,5 @@ class WorkerManager: self.site.onFileDone(task["inner_path"]) task["evt"].set(True) if not self.tasks: - self.started_task_num = 0 - self.site.onComplete() # No more task trigger site complete + self.log.debug("No tasks") + gevent.spawn(self.checkComplete)