Skip to content

Commit 749fccd

Browse files
authored
fix progress bar initialized twice (#765)
1 parent 3da0ef2 commit 749fccd

File tree

1 file changed

+14
-2
lines changed

1 file changed

+14
-2
lines changed

mapillary_tools/upload.py

Lines changed: 14 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -236,7 +236,6 @@ def _setup_tdqm(emitter: uploader.EventEmitter) -> None:
236236
upload_pbar: tqdm | None = None
237237

238238
@emitter.on("upload_start")
239-
@emitter.on("upload_fetch_offset")
240239
def upload_start(payload: uploader.Progress) -> None:
241240
nonlocal upload_pbar
242241

@@ -263,9 +262,22 @@ def upload_start(payload: uploader.Progress) -> None:
263262
disable=LOG.getEffectiveLevel() <= logging.DEBUG,
264263
)
265264

265+
@emitter.on("upload_fetch_offset")
266+
def upload_fetch_offset(payload: uploader.Progress) -> None:
267+
assert upload_pbar is not None, (
268+
"progress_bar must be initialized in upload_start"
269+
)
270+
offset = payload.get("offset", 0)
271+
if offset > 0:
272+
upload_pbar.reset()
273+
upload_pbar.update(offset)
274+
266275
@emitter.on("upload_progress")
267276
def upload_progress(payload: uploader.Progress) -> None:
268-
assert upload_pbar is not None, "progress_bar must be initialized"
277+
assert upload_pbar is not None, (
278+
"progress_bar must be initialized in upload_start"
279+
)
280+
upload_pbar.colour = "GREEN"
269281
upload_pbar.update(payload["chunk_size"])
270282

271283
@emitter.on("upload_end")

0 commit comments

Comments
 (0)