From 43a07b3881a0f4cb5ab05bfef1b0256dba959bff Mon Sep 17 00:00:00 2001 From: Lars Wirzenius Date: Tue, 22 Sep 2020 09:26:01 +0300 Subject: make number of files be a script parameter --- create-files | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/create-files b/create-files index b343078..9aac59b 100755 --- a/create-files +++ b/create-files @@ -19,15 +19,15 @@ def dirnames(a, b): started = time.time() root = sys.argv[1] +numfiles = int(sys.argv[2]) if not os.path.exists(root): os.mkdir(root) -billion = 10 ** 9 per_level = 10 ** 3 a = b = c = 0 dir_a, dir_b = dirnames(a, b) prev_status = 0 -for i in range(billion): +for i in range(numfiles): filename = os.path.join(dir_b, f"file-{i}") open(filename, "w").close() @@ -40,7 +40,7 @@ for i in range(billion): a += 1 dir_a, dir_b = dirnames(a, b) - percent = 100.0 * float(i) / float(billion) + percent = 100.0 * float(i) / float(numfiles) now = time.time() duration = now - started if duration > 1: @@ -49,11 +49,11 @@ for i in range(billion): speed = 1 if speed < 1: print(speed, i, duration) - eta = (billion - i) / speed + eta = (numfiles - i) / speed if now - prev_status > 1: sys.stdout.write( - f"\rCreated {i} of {billion} files ({percent:.2f}%), at {speed:.0f}/s, {eta:.0f}s remaining" + f"\rCreated {i} of {numfiles} files ({percent:.2f}%), at {speed:.0f}/s, {eta:.0f}s remaining" ) prev_status = now -- cgit v1.2.1