Python/ciphers/trafid_cipher.py

121 lines
3.2 KiB
Python
Raw Normal View History

2019-10-05 05:14:13 +00:00
# https://en.wikipedia.org/wiki/Trifid_cipher
2019-04-03 19:27:36 +00:00
def __encryptPart(messagePart, character2Number):
one, two, three = "", "", ""
tmp = []
2019-04-03 19:27:36 +00:00
for character in messagePart:
tmp.append(character2Number[character])
for each in tmp:
one += each[0]
two += each[1]
three += each[2]
2019-10-05 05:14:13 +00:00
return one + two + three
2019-04-03 19:27:36 +00:00
def __decryptPart(messagePart, character2Number):
tmp, thisPart = "", ""
result = []
for character in messagePart:
thisPart += character2Number[character]
for digit in thisPart:
tmp += digit
if len(tmp) == len(messagePart):
result.append(tmp)
tmp = ""
2019-04-03 19:27:36 +00:00
return result[0], result[1], result[2]
2019-10-05 05:14:13 +00:00
2019-04-03 19:27:36 +00:00
def __prepare(message, alphabet):
2019-10-05 05:14:13 +00:00
# Validate message and alphabet, set to upper and remove spaces
2019-04-03 19:27:36 +00:00
alphabet = alphabet.replace(" ", "").upper()
message = message.replace(" ", "").upper()
2019-10-05 05:14:13 +00:00
# Check length and characters
2019-04-03 19:27:36 +00:00
if len(alphabet) != 27:
raise KeyError("Length of alphabet has to be 27.")
for each in message:
if each not in alphabet:
raise ValueError("Each message character has to be included in alphabet!")
2019-10-05 05:14:13 +00:00
# Generate dictionares
numbers = (
"111",
"112",
"113",
"121",
"122",
"123",
"131",
"132",
"133",
"211",
"212",
"213",
"221",
"222",
"223",
"231",
"232",
"233",
"311",
"312",
"313",
"321",
"322",
"323",
"331",
"332",
"333",
)
2019-04-03 19:27:36 +00:00
character2Number = {}
number2Character = {}
for letter, number in zip(alphabet, numbers):
character2Number[letter] = number
number2Character[number] = letter
2019-04-03 19:27:36 +00:00
return message, alphabet, character2Number, number2Character
2019-10-05 05:14:13 +00:00
def encryptMessage(message, alphabet="ABCDEFGHIJKLMNOPQRSTUVWXYZ.", period=5):
2019-04-03 19:27:36 +00:00
message, alphabet, character2Number, number2Character = __prepare(message, alphabet)
encrypted, encrypted_numeric = "", ""
2019-10-05 05:14:13 +00:00
for i in range(0, len(message) + 1, period):
encrypted_numeric += __encryptPart(message[i : i + period], character2Number)
2019-04-03 19:27:36 +00:00
for i in range(0, len(encrypted_numeric), 3):
2019-10-05 05:14:13 +00:00
encrypted += number2Character[encrypted_numeric[i : i + 3]]
2019-04-03 19:27:36 +00:00
return encrypted
2019-10-05 05:14:13 +00:00
def decryptMessage(message, alphabet="ABCDEFGHIJKLMNOPQRSTUVWXYZ.", period=5):
2019-04-03 19:27:36 +00:00
message, alphabet, character2Number, number2Character = __prepare(message, alphabet)
decrypted_numeric = []
decrypted = ""
2019-10-05 05:14:13 +00:00
for i in range(0, len(message) + 1, period):
a, b, c = __decryptPart(message[i : i + period], character2Number)
2019-04-03 19:27:36 +00:00
for j in range(0, len(a)):
2019-10-05 05:14:13 +00:00
decrypted_numeric.append(a[j] + b[j] + c[j])
2019-04-03 19:27:36 +00:00
for each in decrypted_numeric:
decrypted += number2Character[each]
return decrypted
2019-10-05 05:14:13 +00:00
if __name__ == "__main__":
2019-04-03 19:27:36 +00:00
msg = "DEFEND THE EAST WALL OF THE CASTLE."
2019-10-05 05:14:13 +00:00
encrypted = encryptMessage(msg, "EPSDUCVWYM.ZLKXNBTFGORIJHAQ")
2019-04-03 19:27:36 +00:00
decrypted = decryptMessage(encrypted, "EPSDUCVWYM.ZLKXNBTFGORIJHAQ")
print(f"Encrypted: {encrypted}\nDecrypted: {decrypted}")