make number of files be a script parameter

1 files 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