Attempt to catch MemoryError if possible
This commit is contained in:
parent
16054cc3bc
commit
3cb44f5630
16
speedtest.py
16
speedtest.py
|
@ -743,11 +743,17 @@ class HTTPUploaderData(object):
|
|||
chars = '0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ'
|
||||
multiplier = int(round(int(self.length) / 36.0))
|
||||
IO = BytesIO or StringIO
|
||||
self._data = IO(
|
||||
('content1=%s' %
|
||||
(chars * multiplier)[0:int(self.length) - 9]
|
||||
).encode()
|
||||
)
|
||||
try:
|
||||
self._data = IO(
|
||||
('content1=%s' %
|
||||
(chars * multiplier)[0:int(self.length) - 9]
|
||||
).encode()
|
||||
)
|
||||
except MemoryError:
|
||||
raise SpeedtestCLIError(
|
||||
'Insufficient memory to pre-allocate upload data. Please '
|
||||
'use --no-pre-allocate'
|
||||
)
|
||||
|
||||
@property
|
||||
def data(self):
|
||||
|
|
Loading…
Reference in New Issue