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

Add source to spinal returned dotdict.

Browse source 
Just in case you store the results and process them later, out of
order, etc.
This commit is contained in:
voussoir 2021-05-08 09:02:22 -07:00
parent 14c9cb5212
commit 450f42bca6
No known key found for this signature in database
GPG key ID: 5F7554F8C26DACCB
1 changed files with 7 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
voussoirkit/spinal.py
View file

@ -184,8 +184,8 @@ def copy_dir(
validate_hash: validate_hash:
Passed directly into each `copy_file`. Passed directly into each `copy_file`.
Returns a dotdict containing at least `destination` and `written_bytes`. Returns a dotdict containing at least `source`, `destination`,
(Written bytes is 0 if all files already existed.) and `written_bytes`. (Written bytes is 0 if all files already existed.)
''' '''
# Prepare parameters # Prepare parameters
if not is_xor(destination, destination_new_root): if not is_xor(destination, destination_new_root):
@ -302,6 +302,7 @@ def copy_dir(
files_per_second.limit(1) files_per_second.limit(1)
results = dotdict.DotDict({ results = dotdict.DotDict({
'source': source,
'destination': destination, 'destination': destination,
'written_bytes': written_bytes, 'written_bytes': written_bytes,
}) })
@ -372,7 +373,7 @@ def copy_file(
hash_class: hash_class:
If provided, should be a hashlib class or a callable that returns an If provided, should be a hashlib class or a callable that returns an
instance of one. The hash will be computed while the file is being instance of one. The hash will be computed while the file is being
copied, and returned in the dotdict as `hash`. copied. The instance will be returned in the dotdict as `hash`.
Note that if the function returns early due to dry_run or file not Note that if the function returns early due to dry_run or file not
needing overwrite, this won't be set, so be prepared to handle None. needing overwrite, this won't be set, so be prepared to handle None.
If None, the hash will not be calculated. If None, the hash will not be calculated.
@ -387,8 +388,8 @@ def copy_file(
and its hash will be compared against the hash of the source file. and its hash will be compared against the hash of the source file.
If hash_class is None, then the global HASH_CLASS is used. If hash_class is None, then the global HASH_CLASS is used.
Returns a dotdict containing at least `destination` and `written_bytes`. Returns a dotdict containing at least `source`, `destination`,
(Written bytes is 0 if the file already existed.) and `written_bytes`. (Written bytes is 0 if the file already existed.)
''' '''
# Prepare parameters # Prepare parameters
if not is_xor(destination, destination_new_root): if not is_xor(destination, destination_new_root):
@ -415,6 +416,7 @@ def copy_file(
bytes_per_second = limiter_or_none(bytes_per_second) bytes_per_second = limiter_or_none(bytes_per_second)
results = dotdict.DotDict({ results = dotdict.DotDict({
'source': source,
'destination': destination, 'destination': destination,
'written_bytes': 0, 'written_bytes': 0,
}, default=None) }, default=None)