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

Add callback_post_file.

Browse source
This commit is contained in:
voussoir 2021-05-08 09:39:48 -07:00
parent 133225f77c
commit 313070754f
No known key found for this signature in database
GPG key ID: 5F7554F8C26DACCB
1 changed files with 10 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
voussoirkit/spinal.py
View file

@ -90,6 +90,7 @@ def copy_dir(
callback_permission_denied=None, callback_permission_denied=None,
callback_pre_directory=None, callback_pre_directory=None,
callback_pre_file=None, callback_pre_file=None,
callback_post_file=None,
chunk_size=CHUNK_SIZE, chunk_size=CHUNK_SIZE,
destination_new_root=None, destination_new_root=None,
dry_run=False, dry_run=False,
@ -142,6 +143,12 @@ def copy_dir(
callback_pre_file: callback_pre_file:
Passed into each `copy_file` as `callback_pre_copy`. Passed into each `copy_file` as `callback_pre_copy`.
callback_post_file:
This function will be called after the file finishes copying with one
argument: the dotdict returned by copy_file.
If you think copy_dir should be rewritten as a generator instead,
I agree!
destination_new_root: destination_new_root:
Determine the destination path by calling Determine the destination path by calling
`new_root(source, destination_new_root)`. `new_root(source, destination_new_root)`.
@ -219,6 +226,7 @@ def copy_dir(
callback_directory_progress = callback_directory_progress or do_nothing callback_directory_progress = callback_directory_progress or do_nothing
callback_pre_directory = callback_pre_directory or do_nothing callback_pre_directory = callback_pre_directory or do_nothing
callback_pre_file = callback_pre_file or do_nothing callback_pre_file = callback_pre_file or do_nothing
callback_post_file = callback_post_file or do_nothing
bytes_per_second = limiter_or_none(bytes_per_second) bytes_per_second = limiter_or_none(bytes_per_second)
files_per_second = limiter_or_none(files_per_second) files_per_second = limiter_or_none(files_per_second)
@ -298,6 +306,8 @@ def copy_dir(
else: else:
callback_directory_progress(copied.destination, written_bytes, total_bytes) callback_directory_progress(copied.destination, written_bytes, total_bytes)
callback_post_file(copied)
if files_per_second is not None: if files_per_second is not None:
files_per_second.limit(1) files_per_second.limit(1)