File tree Expand file tree Collapse file tree 1 file changed +14
-2
lines changed
Expand file tree Collapse file tree 1 file changed +14
-2
lines changed Original file line number Diff line number Diff 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" )
You can’t perform that action at this time.
0 commit comments