misc/progress/progress.py

62 lines
1.8 KiB
Python

import datetime
import re
import subprocess
DISK_SIZE_GB = 5.44
def sizeof_fmt(num: int, suffix: str = 'B') -> str:
for unit in ['', 'Ki', 'Mi', 'Gi', 'Ti', 'Pi', 'Ei', 'Zi']:
if abs(num) < 1024.0:
return "%3.1f %s%s" % (num, unit, suffix)
num /= 1024.0
return "%.1f%s%s" % (num, 'Yi', suffix)
def time_fmt(_time):
hours = int(_time / 3600)
minutes = int((_time % 3600) / 60)
seconds = _time % 60
fmt = f"{hours}h {minutes}min {seconds}s"
return fmt
def main():
subprocess.run(["sudo", "killall", "-SIGINFO", "dd"])
print("Input last dd progress line")
last_line = input("> ")
regex = re.compile(
r"^(?P<transferred>\d+) \w+ transferred in (?P<elapsed>\d+).\d+ secs \((?P<speed>\d+) \w+/sec\)$")
results = regex.match(last_line).groupdict()
disk_size = DISK_SIZE_GB * 1024 * 1024 * 1024
transferred = int(results["transferred"])
transferred_fmt = sizeof_fmt(transferred)
speed = int(results["speed"])
speed_fmt = sizeof_fmt(speed)
remaining = disk_size - transferred
remaining_fmt = sizeof_fmt(remaining)
eta = int(remaining / speed)
eta_fmt = time_fmt(eta)
elapsed = int(results["elapsed"])
elapsed_fmt = time_fmt(elapsed)
total = eta + elapsed
total_fmt = time_fmt(total)
eta_absolute = datetime.datetime.now() + datetime.timedelta(seconds=eta)
percentage = transferred / disk_size * 100
print()
print(f"{transferred_fmt} transferred over {DISK_SIZE_GB}GiB, or {percentage:.1f}%.")
print(f"Transferring at {speed_fmt} per second.")
print(f"Remaining : {remaining_fmt}")
print(f"ETA : {eta_fmt} ({eta_absolute.ctime()}).")
print(f"Elapsed : {elapsed_fmt}.")
print(f"Estimated total time : {total_fmt}.")
if __name__ == '__main__':
main()