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] Show speed while verifying large files

Browse source
This commit is contained in:
derrod 2022-01-04 12:11:24 +01:00
parent cea5f42425
commit ecb04324d5
1 changed files with 10 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
legendary/utils/lfs.py
View file

@ -7,6 +7,7 @@ import logging
from pathlib import Path from pathlib import Path
from sys import stdout from sys import stdout
from time import time
from typing import List, Iterator from typing import List, Iterator
from legendary.models.game import VerifyResult from legendary.models.game import VerifyResult
@ -103,6 +104,9 @@ def validate_files(base_path: str, filelist: List[tuple], hash_type='sha1',
continue continue
show_progress = False show_progress = False
interval = 0
speed = 0.0
start_time = 0.0
try: try:
_size = os.path.getsize(full_path) _size = os.path.getsize(full_path)
@ -111,8 +115,7 @@ def validate_files(base_path: str, filelist: List[tuple], hash_type='sha1',
stdout.write('\n') stdout.write('\n')
show_progress = True show_progress = True
interval = (_size / (1024 * 1024)) // 100 interval = (_size / (1024 * 1024)) // 100
else: start_time = time()
interval = 0
with open(full_path, 'rb') as f: with open(full_path, 'rb') as f:
real_file_hash = hashlib.new(hash_type) real_file_hash = hashlib.new(hash_type)
@ -122,14 +125,17 @@ def validate_files(base_path: str, filelist: List[tuple], hash_type='sha1',
if show_progress and i % interval == 0: if show_progress and i % interval == 0:
pos = f.tell() pos = f.tell()
perc = (pos / _size) * 100 perc = (pos / _size) * 100
speed = pos / 1024 / 1024 / (time() - start_time)
stdout.write(f'\r=> Verifying large file "{file_path}": {perc:.0f}% ' stdout.write(f'\r=> Verifying large file "{file_path}": {perc:.0f}% '
f'({pos / 1024 / 1024:.1f}/{_size / 1024 / 1024:.1f} MiB)') f'({pos / 1024 / 1024:.1f}/{_size / 1024 / 1024:.1f} MiB) '
f'[{speed:.1f} MiB/s]\t')
stdout.flush() stdout.flush()
i += 1 i += 1
if show_progress: if show_progress:
stdout.write(f'\r=> Verifying large file "{file_path}": 100% ' stdout.write(f'\r=> Verifying large file "{file_path}": 100% '
f'({_size / 1024 / 1024:.1f}/{_size / 1024 / 1024:.1f} MiB)\n') f'({_size / 1024 / 1024:.1f}/{_size / 1024 / 1024:.1f} MiB) '
f'[{speed:.1f} MiB/s]\t\n')
result_hash = real_file_hash.hexdigest() result_hash = real_file_hash.hexdigest()
if file_hash != result_hash: if file_hash != result_hash: