From 60fd075506093d94107515788511f4a56f876817 Mon Sep 17 00:00:00 2001 From: Guilhem Moulin Date: Mon, 3 Jun 2024 16:06:25 +0200 Subject: webmap-download: Move format_bytes() and format_time() to common.py. --- webmap-download | 23 +++-------------------- 1 file changed, 3 insertions(+), 20 deletions(-) (limited to 'webmap-download') diff --git a/webmap-download b/webmap-download index 6040356..fb52d7d 100755 --- a/webmap-download +++ b/webmap-download @@ -10,7 +10,6 @@ import itertools from pathlib import Path from email.utils import parsedate_to_datetime, formatdate from hashlib import sha1 -from math import modf import requests import common @@ -112,24 +111,8 @@ def download(url, dest, dir_fd=None, headers={}, session=requests, progress=None raise e elapsed = time_monotonic() - start - logging.info("%s: Downloaded %s in %s (%s/s)", dest, format_bytes(size), - format_time(elapsed), format_bytes(int(size/elapsed))) - -def format_bytes(n): - if n < 768: - return f'{n}B' - elif n < 786432: - return f'{n/1024:.2f}kiB' - elif n < 805306368: - return f'{n/1048576:.2f}MiB' - else: - return f'{n/1073741824:.2f}GiB' - -def format_time(s): - fs, s = modf(s) - m, s = divmod(int(s), 60) - h, m = divmod(m, 60) - return f'{h:02d}:{m:02d}:{s + fs:06.3f}' + logging.info("%s: Downloaded %s in %s (%s/s)", dest, common.format_bytes(size), + common.format_time(elapsed), common.format_bytes(int(size/elapsed))) if __name__ == '__main__': logging.basicConfig(format='%(levelname)s: %(message)s', level=logging.INFO) @@ -229,7 +212,7 @@ if __name__ == '__main__': s = max_age + max(st.st_ctime, st.st_mtime) - time() if s > 0: logging.info('%s: Too young, try again in %s', - dest, format_time(s)) + dest, common.format_time(s)) continue headers['If-Modified-Since'] = formatdate(timeval=st.st_mtime, localtime=False, usegmt=True) fetch(dl, dest, dir_fd=destdir_fd, -- cgit v1.2.3