voussoir
/
voussoirkit
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Let chunk_size be an argument to override global default.

master
Ethan Dalool 2020-01-05 21:31:19 -08:00
parent c33d16bff0
commit 6faab40fb2
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
voussoirkit/spinal.py
View File

@ -13,8 +13,8 @@ from voussoirkit import ratelimiter
logging.basicConfig(level=logging.CRITICAL) logging.basicConfig(level=logging.CRITICAL)
log = logging.getLogger(__name__) log = logging.getLogger(__name__)
CHUNK_SIZE = 2 * bytestring.MIBIBYTE
# Number of bytes to read and write at a time # Number of bytes to read and write at a time
CHUNK_SIZE = 2 * bytestring.MIBIBYTE
HASH_CLASS = hashlib.md5 HASH_CLASS = hashlib.md5
@ -90,6 +90,7 @@ def copy_dir(
callback_exclusion=None, callback_exclusion=None,
callback_file=None, callback_file=None,
callback_permission_denied=None, callback_permission_denied=None,
chunk_size=CHUNK_SIZE,
destination_new_root=None, destination_new_root=None,
dry_run=False, dry_run=False,
exclude_directories=None, exclude_directories=None,
@ -255,6 +256,7 @@ def copy_dir(
bytes_per_second=bytes_per_second, bytes_per_second=bytes_per_second,
callback_progress=callback_file, callback_progress=callback_file,
callback_permission_denied=callback_permission_denied, callback_permission_denied=callback_permission_denied,
chunk_size=chunk_size,
dry_run=dry_run, dry_run=dry_run,
overwrite_old=overwrite_old, overwrite_old=overwrite_old,
validate_hash=validate_hash, validate_hash=validate_hash,
@ -282,6 +284,7 @@ def copy_file(
callback_progress=None, callback_progress=None,
callback_permission_denied=None, callback_permission_denied=None,
callback_validate_hash=None, callback_validate_hash=None,
chunk_size=CHUNK_SIZE,
dry_run=False, dry_run=False,
overwrite_old=True, overwrite_old=True,
validate_hash=False, validate_hash=False,
@ -319,7 +322,7 @@ def copy_file(
callback_progress: callback_progress:
If provided, this function will be called after writing If provided, this function will be called after writing
each CHUNK_SIZE bytes to destination with three parameters: each chunk_size bytes to destination with three parameters:
the Path object being copied, number of bytes written so far, the Path object being copied, number of bytes written so far,
total number of bytes needed. total number of bytes needed.
@ -416,7 +419,7 @@ def copy_file(
written_bytes = 0 written_bytes = 0
while True: while True:
try: try:
data_chunk = source_handle.read(CHUNK_SIZE) data_chunk = source_handle.read(chunk_size)
except PermissionError as e: except PermissionError as e:
print(source) print(source)
raise raise