Used loops for iterations rather than hardcoded steps.

README.md

@@ -4,8 +4,15 @@ Recreates the functionality of the Enigma Machine from World War II.
 ## Usage
 `uv run main.py [0-58] [0-58] [0-58] '[message]' 'e|d' [int]`
 
+Where:
+- 0-58 [integer] is the first seed value
+- 0-58 [integer] is the second seed value
+- 0-58 [integer] is the third seed value
+- message is the message to be encrypted/decrypted
+- e|d is the direction of the message (encrypt or decrypt)
+- int is the number of iterations to be made (not yet implemented)
 
+## Examples
+`uv run main.py  32 15 1 'This is a test.' 'e' 3`
 
-`uv run main.py  32 15 1 'This is a test.' 'e'`
+`uv run main.py 32 15 1 'Uijt!jt!b!uftu' 'd' 3`
-
-`uv run main.py 32 15 1 'Uijt!jt!b!uftu' 'd'`

main.py

@@ -41,30 +41,24 @@ def main():
         iterations = int(sys.argv[6])
 
     ### Processing ###
+    seed_values = [seed1, seed2, seed3]
+    iteration_result = ""
+
+    current_timestamp = time.time()
+    print(current_timestamp)
+
     if sys.argv[5] == "e":
         for iteration in range(0, iterations):
-            print("Iteration (not yet in use): ", iteration)
+            iteration_result = encode(seed_values[iteration], sys.argv[4])
 
-
+        print("Seeds used:", seed1, seed2, seed3)
-        # first iteration
+        print("Result:", iteration_result)
-        firstIterationResult = encode(seed1, sys.argv[4])
-        print(firstIterationResult)
-        
-        secondIterationResult = encode(seed2, sys.argv[4])
-        print(secondIterationResult)
-        
-        thirdIterationResult = encode(seed3, sys.argv[4])
-        # print(thirdIterationResult)
-        print("Seeds: ", seed1, seed2, seed3, thirdIterationResult)
     elif sys.argv[5] == "d":
-        firstIterationResult = decode(seed1, sys.argv[4])
+        for iteration in range(0, iterations):
-        print(firstIterationResult)
+            iteration_result = decode(seed_values[iteration], sys.argv[4])
 
-        secondIterationResult = decode(seed2, sys.argv[4])
+        print("Seeds used:", seed1, seed2, seed3)
-        print(secondIterationResult)
+        print("Result:", iteration_result)
-
-        thirdIterationResult = decode(seed3, sys.argv[4])
-        print(thirdIterationResult)
     else:
         print("Invalid option!")
         sys.exit(1)
@@ -72,8 +66,6 @@ def main():
 
 def encode(offset, message):
     print("Encoding!")
-    current_timestamp = time.time()
-    print(current_timestamp)
 
     resultMessage = ""