mirrors/legendary
1
0
Fork 0
mirror of https://github.com/derrod/legendary.git synced 2024-12-22 17:55:27 +00:00
Code Issues Projects Releases Wiki Activity

[utils] Return hashes from validate_files

Browse source
This commit is contained in:
derrod 2020-05-20 12:03:44 +02:00
parent 5b2ebada78
commit 9cc5d071d0
1 changed files with 5 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
legendary/utils/lfs.py
View file

@ -47,7 +47,7 @@ def validate_files(base_path: str, filelist: List[tuple], hash_type='sha1') -> I
# logger.debug(f'Checking "{file_path}"...') # logger.debug(f'Checking "{file_path}"...')
if not os.path.exists(full_path): if not os.path.exists(full_path):
yield VerifyResult.FILE_MISSING, file_path yield VerifyResult.FILE_MISSING, file_path, ''
continue continue
with open(full_path, 'rb') as f: with open(full_path, 'rb') as f:
@ -55,10 +55,11 @@ def validate_files(base_path: str, filelist: List[tuple], hash_type='sha1') -> I
while chunk := f.read(8192): while chunk := f.read(8192):
real_file_hash.update(chunk) real_file_hash.update(chunk)
if file_hash != real_file_hash.hexdigest(): result_hash = real_file_hash.hexdigest()
yield VerifyResult.HASH_MISMATCH, file_path if file_hash != result_hash:
yield VerifyResult.HASH_MISMATCH, file_path, result_hash
else: else:
yield VerifyResult.HASH_MATCH, file_path yield VerifyResult.HASH_MATCH, file_path, result_hash
def clean_filename(filename): def clean_filename(filename):