diff --git a/README.md b/README.md index 9703fda..e849f91 100644 --- a/README.md +++ b/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'` \ No newline at end of file +`uv run main.py 32 15 1 'Uijt!jt!b!uftu' 'd' 3` diff --git a/main.py b/main.py index d957e5e..6b7b74c 100644 --- a/main.py +++ b/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]) - - # first iteration - 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) + print("Seeds used:", seed1, seed2, seed3) + print("Result:", iteration_result) elif sys.argv[5] == "d": - firstIterationResult = decode(seed1, sys.argv[4]) - print(firstIterationResult) + for iteration in range(0, iterations): + iteration_result = decode(seed_values[iteration], sys.argv[4]) - secondIterationResult = decode(seed2, sys.argv[4]) - print(secondIterationResult) - - thirdIterationResult = decode(seed3, sys.argv[4]) - print(thirdIterationResult) + print("Seeds used:", seed1, seed2, seed3) + print("Result:", iteration_result) 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 = ""