mirror of
https://github.com/TheAlgorithms/Python.git
synced 2024-11-23 21:11:08 +00:00
added enigma machine algorithm (#932)
This commit is contained in:
parent
4ff2a9dd4e
commit
9532492728
61
hashes/enigma_machine.py
Normal file
61
hashes/enigma_machine.py
Normal file
|
@ -0,0 +1,61 @@
|
|||
from __future__ import print_function
|
||||
|
||||
alphabets = [chr(i) for i in range(32, 126)]
|
||||
gear_one = [i for i in range(len(alphabets))]
|
||||
gear_two = [i for i in range(len(alphabets))]
|
||||
gear_three = [i for i in range(len(alphabets))]
|
||||
reflector = [i for i in reversed(range(len(alphabets)))]
|
||||
code = []
|
||||
gear_one_pos = gear_two_pos = gear_three_pos = 0
|
||||
|
||||
|
||||
def rotator():
|
||||
global gear_one_pos
|
||||
global gear_two_pos
|
||||
global gear_three_pos
|
||||
i = gear_one[0]
|
||||
gear_one.append(i)
|
||||
del gear_one[0]
|
||||
gear_one_pos += 1
|
||||
if gear_one_pos % int(len(alphabets)) == 0:
|
||||
i = gear_two[0]
|
||||
gear_two.append(i)
|
||||
del gear_two[0]
|
||||
gear_two_pos += 1
|
||||
if gear_two_pos % int(len(alphabets)) == 0:
|
||||
i = gear_three[0]
|
||||
gear_three.append(i)
|
||||
del gear_three[0]
|
||||
gear_three_pos += 1
|
||||
|
||||
|
||||
def engine(input_character):
|
||||
target = alphabets.index(input_character)
|
||||
target = gear_one[target]
|
||||
target = gear_two[target]
|
||||
target = gear_three[target]
|
||||
target = reflector[target]
|
||||
target = gear_three.index(target)
|
||||
target = gear_two.index(target)
|
||||
target = gear_one.index(target)
|
||||
code.append(alphabets[target])
|
||||
rotator()
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
decode = input("Type your message:\n")
|
||||
decode = list(decode)
|
||||
while True:
|
||||
try:
|
||||
token = int(input("Please set token:(must be only digits)\n"))
|
||||
break
|
||||
except Exception as error:
|
||||
print(error)
|
||||
for i in range(token):
|
||||
rotator()
|
||||
for i in decode:
|
||||
engine(i)
|
||||
print("\n" + "".join(code))
|
||||
print(
|
||||
f"\nYour Token is {token} please write it down.\nIf you want to decode "
|
||||
f"this message again you should input same digits as token!")
|
Loading…
Reference in New Issue
Block a user