You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
53 lines
1.2 KiB
53 lines
1.2 KiB
3 years ago
|
import io
|
||
|
|
|
||
|
|
|
||
|
|
class CancelledError(Exception):
|
||
|
|
def __init__(self, msg):
|
||
|
|
self.msg = msg
|
||
|
|
Exception.__init__(self, msg)
|
||
|
|
|
||
|
|
def __str__(self):
|
||
|
|
return self.msg
|
||
|
|
|
||
|
|
__repr__ = __str__
|
||
|
|
|
||
|
|
|
||
|
|
class BufferReader(io.BytesIO):
|
||
|
|
def __init__(self, buf=b"", desc=None):
|
||
|
|
self._len = len(buf)
|
||
|
|
io.BytesIO.__init__(self, buf)
|
||
|
|
self._progress = 0
|
||
|
|
self._callback = progress(len(buf), desc=desc)
|
||
|
|
|
||
|
|
def __len__(self):
|
||
|
|
return self._len
|
||
|
|
|
||
|
|
def read(self, n=-1):
|
||
|
|
chunk = io.BytesIO.read(self, n)
|
||
|
|
self._progress += len(chunk)
|
||
|
|
if self._callback:
|
||
|
|
try:
|
||
|
|
self._callback(self._progress)
|
||
|
|
except Exception as e: # catches exception from the callback
|
||
|
|
raise CancelledError("The upload was cancelled: {}".format(e))
|
||
|
|
return chunk
|
||
|
|
|
||
|
|
|
||
|
|
def progress(total, desc):
|
||
|
|
import tqdm # type: ignore
|
||
|
|
|
||
|
|
meter = tqdm.tqdm(total=total, unit_scale=True, desc=desc)
|
||
|
|
|
||
|
|
def incr(progress):
|
||
|
|
meter.n = progress
|
||
|
|
if progress == total:
|
||
|
|
meter.close()
|
||
|
|
else:
|
||
|
|
meter.refresh()
|
||
|
|
|
||
|
|
return incr
|
||
|
|
|
||
|
|
|
||
|
|
def MB(i):
|
||
|
|
return int(i // 1024**2)
|