spacepaste

Paste details

reply | raw | toggle line wrapping

  1.  
  2. #!/usr/bin/python3
  3. from io import StringIO, BytesIO
  6. # Old
  7. def copyfileobj(fsrc, fdst, length=16*1024):
  8. """copy data from file-like object fsrc to file-like object fdst"""
  9. while 1:
  10. buf = fsrc.read(length)
  11. if not buf:
  12. break
  13. fdst.write(buf)
  16. # New
  17. def _copyfileobj(fsrc, fdst, length=16*1024):
  18. """copy data from file-like object fsrc to file-like object fdst"""
  19. write = fdst.write
  20. if hasattr(fsrc, 'readinto') and length > 0:
  21. buf = memoryview(bytearray(length))
  22. readinto = fsrc.readinto
  23. while 1:
  24. recv_len = readinto(buf)
  25. if recv_len < length:
  26. # write remaining content if any.
  27. write(buf[:recv_len])
  28. break
  29. write(buf)
  30. else:
  31. read = fsrc.read
  32. while 1:
  33. buf = read(length)
  34. if not buf:
  35. break
  36. write(buf)
  39. value = bytearray([0]*107374182) # 100mb
  40. # value = bytearray([0]*536870912) # 500mb
  41. with BytesIO(value) as fsrc:
  42. with BytesIO() as fdst:
  43. copyfileobj(fsrc, fdst)
  44.  
Powered by spacepaste. Paste removal.