Pass consistent task progress during syncing.

learningequality · Aug 22, 2023 · 95ba857 · 95ba857
1 parent 761ebf6
commit 95ba857
Showing 1 changed file with 4 additions and 9 deletions.
diff --git a/kolibri/core/auth/management/utils.py b/kolibri/core/auth/management/utils.py
@@ -4,7 +4,6 @@
 import getpass
 import json
 import logging
-import math
 import sys
 from contextlib import contextmanager
 from functools import wraps
@@ -659,23 +658,19 @@ def stats(transfer_session):
 
         def started(transfer_session):
             stats(transfer_session)
-            self.start_progress(total=100)
+            self.start_progress(total=transfer_session.records_total or 100)
 
         def handler(transfer_session):
             """
             :type transfer_session: morango.models.core.TransferSession
             """
             if transfer_session.records_total > 0:
-                progress = (
-                    100
-                    * transfer_session.records_transferred
-                    / float(transfer_session.records_total)
-                )
+                progress = transfer_session.records_transferred
             else:
                 progress = 100
 
             self.update_progress(
-                increment=math.ceil(progress - self.progresstracker.progress),
+                current_progress=progress,
                 message=stats_msg(transfer_session),
                 extra_data=dict(
                     bytes_sent=transfer_session.bytes_sent,
@@ -703,7 +698,7 @@ def _queueing_tracker_adapter(
         """
 
         def started(transfer_session):
-            self.start_progress(total=2)
+            self.start_progress(total=1)
             dataset_cache.clear()
             if noninteractive or self.progresstracker.progressbar is None:
                 if (