Skip to content

Commit

Permalink
WIP add progressbar debug
Browse files Browse the repository at this point in the history
  • Loading branch information
glensc committed Apr 13, 2024
1 parent 88f88a5 commit 54ec109
Showing 1 changed file with 7 additions and 0 deletions.
7 changes: 7 additions & 0 deletions plextraktsync/rich/RichProgressBar.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,20 +21,24 @@ def __init__(self, iterable, total=None, options=None, desc=""):
self.iterable_next = iter(iterable).__next__

def __iter__(self):
print("rich iter")
if self.iterable_awaitable:
raise RuntimeError("Iterable must be awaited")

return self

def __next__(self):
print("rich next")
res = self.iterable_next()
self.update()
return res

def __aiter__(self):
print("rich aiter")
return self

async def __anext__(self):
print("rich anext")
try:
if self.iterable_awaitable:
res = await self.iterable_next()
Expand All @@ -46,13 +50,16 @@ async def __anext__(self):
raise StopAsyncIteration

def __enter__(self):
print("rich enter")
self.progress.__enter__()
return self

def __exit__(self, *exc):
print("rich exit")
self.progress.__exit__(*exc)

def update(self):
print("rich update")
self.i += 1
self.progress.update(self.task_id, completed=self.i)

Expand Down

0 comments on commit 54ec109

Please sign in to comment.