From f8b5b6b577f84cdc3057d2225e731e121a9f4b4e Mon Sep 17 00:00:00 2001
From: Kenn Kitchen <me@kenneth.kitchen>
Date: Sun, 26 Oct 2025 13:03:03 -0400
Subject: [PATCH] Can now handle user-specified iterations. Uses a file for the
 seeds.

---
 .gitignore |  2 ++
 main.py    | 72 +++++++++++++++++++++++++++---------------------------
 2 files changed, 38 insertions(+), 36 deletions(-)

diff --git a/.gitignore b/.gitignore
index 142fa8a..95545a2 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,3 +1,5 @@
+seeds.txt
+
 ### VirtualEnv template
 # Virtualenv
 # http://iamzed.com/2009/05/07/a-primer-on-virtualenv/
diff --git a/main.py b/main.py
index f3c202f..e542233 100644
--- a/main.py
+++ b/main.py
@@ -2,7 +2,7 @@ import random
 import sys
 import time
 
-alphaBase = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz,. !="
+alpha_base = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz,. !="
 
 def main():
     ### Argument validation ###
@@ -10,7 +10,7 @@ def main():
         print("Usage: uv run main.py [0-58] [0-58] [0-58] '[message]' 'e|d' [int]")
         sys.exit(1)
 
-    base_max = len(alphaBase)
+    base_max = len(alpha_base)
 
     if not is_number(sys.argv[1]) or not is_number(sys.argv[2]) or not is_number(sys.argv[3]):
         print("Invalid seed value.")
@@ -41,72 +41,72 @@ def main():
         iterations = int(sys.argv[6])
 
     ### Processing ###
-    # current_timestamp = time.time()
-    # print(current_timestamp)
-
     seed_values = []
 
     if sys.argv[5] == "e":
-        for iteration in range(0, iterations):
-            random.seed(time.time())
-            seed_values.append(random.randint(1, base_max))
+        with open("seeds.txt", "w") as f:
+            for iteration in range(0, iterations):
+                random.seed(time.time())
+                seed_values.append(random.randint(1, base_max))
+                f.write(str(seed_values[iteration]) + "\n")
+        f.close()
 
     if sys.argv[5] == "d":
-        seed_values.append(seed1)
-        seed_values.append(seed2)
-        seed_values.append(seed3)
+        with open("seeds.txt", "r") as f:
+            for iteration in range(0, iterations):
+                seed = f.readline()
+                seed_values.append(int(seed))
+        f.close()
 
     iteration_result = ""
 
     if sys.argv[5] == "e":
+        print("Encoding!")
         for iteration in range(0, iterations):
             iteration_result = encode(seed_values[iteration], sys.argv[4])
-            print("Seed:", iteration, "is", seed_values[iteration])
+            # print("Seed:", iteration, "is", seed_values[iteration])
 
-        print("Result:", iteration_result)
+        print("Result:", "\"" + iteration_result + "\"")
     elif sys.argv[5] == "d":
+        print("Decoding!")
         for iteration in range(0, iterations):
             iteration_result = decode(seed_values[iteration], sys.argv[4])
 
-        print("Seeds used:", seed1, seed2, seed3)
-        print("Result:", iteration_result)
+        # print("Seeds used:", seed1, seed2, seed3)
+        print("Result:", "\"" + iteration_result + "\"")
     else:
         print("Invalid option!")
         sys.exit(1)
 
 
 def encode(offset, message):
-    print("Encoding!")
+    result_message = ""
 
-    resultMessage = ""
-
-    for indexA in range(len(message)):
-        for indexB in range(len(alphaBase)):
-            if message[indexA] == alphaBase[indexB]:
-                location = indexB + offset
-                if location > (len(alphaBase) - 1):
-                    location = location - (len(alphaBase) - 1)
-                resultMessage = resultMessage + alphaBase[location]
+    for index_a in range(len(message)):
+        for index_b in range(len(alpha_base)):
+            if message[index_a] == alpha_base[index_b]:
+                location = index_b + offset
+                if location > (len(alpha_base) - 1):
+                    location = location - (len(alpha_base) - 1)
+                result_message = result_message + alpha_base[location]
                 break
 
-    return resultMessage
+    return result_message
 
 
 def decode(offset, message):
-    print("Decoding!")
+    result_message = ""
 
-    resultMessage = ""
-
-    for indexA in range(len(message)):
-        for indexB in range(len(alphaBase)):
-            if message[indexA] == alphaBase[indexB]:
-                location = indexB - offset
+    for index_a in range(len(message)):
+        for index_b in range(len(alpha_base)):
+            if message[index_a] == alpha_base[index_b]:
+                location = index_b - offset
                 if location < 0:
-                    location = location + (len(alphaBase) - 1)
-                resultMessage = resultMessage + alphaBase[location]
+                    location = location + (len(alpha_base) - 1)
+                result_message = result_message + alpha_base[location]
                 break
 
-    return resultMessage
+    return result_message
 
 
 def is_number(i):