From 6faab40fb2cd9cfb6e6499c3aef1a4fe6f828e81 Mon Sep 17 00:00:00 2001
From: Ethan Dalool <ethan@voussoir.net>
Date: Sun, 5 Jan 2020 21:31:19 -0800
Subject: [PATCH] Let chunk_size be an argument to override global default.

---
 voussoirkit/spinal.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/voussoirkit/spinal.py b/voussoirkit/spinal.py
index c271c77..99f165f 100644
--- a/voussoirkit/spinal.py
+++ b/voussoirkit/spinal.py
@@ -13,8 +13,8 @@ from voussoirkit import ratelimiter
 logging.basicConfig(level=logging.CRITICAL)
 log = logging.getLogger(__name__)
 
-CHUNK_SIZE = 2 * bytestring.MIBIBYTE
 # Number of bytes to read and write at a time
+CHUNK_SIZE = 2 * bytestring.MIBIBYTE
 
 HASH_CLASS = hashlib.md5
 
@@ -90,6 +90,7 @@ def copy_dir(
         callback_exclusion=None,
         callback_file=None,
         callback_permission_denied=None,
+        chunk_size=CHUNK_SIZE,
         destination_new_root=None,
         dry_run=False,
         exclude_directories=None,
@@ -255,6 +256,7 @@ def copy_dir(
             bytes_per_second=bytes_per_second,
             callback_progress=callback_file,
             callback_permission_denied=callback_permission_denied,
+            chunk_size=chunk_size,
             dry_run=dry_run,
             overwrite_old=overwrite_old,
             validate_hash=validate_hash,
@@ -282,6 +284,7 @@ def copy_file(
         callback_progress=None,
         callback_permission_denied=None,
         callback_validate_hash=None,
+        chunk_size=CHUNK_SIZE,
         dry_run=False,
         overwrite_old=True,
         validate_hash=False,
@@ -319,7 +322,7 @@ def copy_file(
 
     callback_progress:
         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,
         total number of bytes needed.
 
@@ -416,7 +419,7 @@ def copy_file(
     written_bytes = 0
     while True:
         try:
-            data_chunk = source_handle.read(CHUNK_SIZE)
+            data_chunk = source_handle.read(chunk_size)
         except PermissionError as e:
             print(source)
             raise