ramforth/pytchat-fork
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update progress bar

Browse Source
This commit is contained in:
taizan-hokuto
2020-09-06 18:25:16 +09:00
parent 9a2e96d3a0
commit 64ace9dad6
3 changed files with 57 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
pytchat/cli/progressbar.py
View File

@@ -6,33 +6,46 @@ https://gist.github.com/vladignatyev/06860ec2040cb497f0f3
'''
import sys
ROT = ['\u25F4', '\u25F5', '\u25F6', '\u25F7']
class ProgressBar:
def __init__(self, duration):
self._duration = duration
self._count = 0
def __init__(self, total, status):
self._bar_len = 60
self._cancelled = False
self.reset(total=total, status=status)
self._blinker = 0
def reset(self, symbol_done="=", symbol_space=" ", total=100, status=''):
self._symbol_done = symbol_done
self._symbol_space = symbol_space
self._total = total
self._status = status
self._count = 0
def _disp(self, _, fetched):
self._progress(fetched / 1000, self._duration)
self._progress(fetched, self._total)
def _progress(self, fillin, total, status=''):
def _progress(self, fillin, total):
if total == 0 or self._cancelled:
return
self._count += fillin
filled_len = int(round(self._bar_len * self._count / float(total)))
percents = round(100.0 * self._count / float(total), 1)
if percents > 100:
percents = 100.0
if filled_len > self._bar_len:
filled_len = self._bar_len
percents = 100
bar = '=' * filled_len + ' ' * (self._bar_len - filled_len)
sys.stdout.write(' [%s] %s%s ...%s\r' % (bar, percents, '%', status))
bar = self._symbol_done * filled_len + \
self._symbol_space * (self._bar_len - filled_len)
sys.stdout.write(' [%s] %s%s ...%s %s \r' % (bar, percents, '%', self._status, ROT[self._blinker % 4]))
sys.stdout.flush()
self._blinker += 1
def close(self):
if not self._cancelled:
self._progress(self._duration, self._duration)
self._progress(self._total, self._total)
def cancel(self):
self._cancelled = True