Python/ciphers/playfair_cipher.py

105 lines
3.0 KiB
Python
Raw Normal View History

2017-08-19 02:41:46 +00:00
import itertools
import string
from collections.abc import Generator, Iterable
2017-08-19 02:41:46 +00:00
2019-10-05 05:14:13 +00:00
def chunker(seq: Iterable[str], size: int) -> Generator[tuple[str, ...], None, None]:
2017-08-19 02:41:46 +00:00
it = iter(seq)
while True:
2019-10-05 05:14:13 +00:00
chunk = tuple(itertools.islice(it, size))
if not chunk:
return
yield chunk
2017-08-19 02:41:46 +00:00
def prepare_input(dirty: str) -> str:
2017-08-19 02:41:46 +00:00
"""
Prepare the plaintext by up-casing it
and separating repeated letters with X's
2017-08-19 02:41:46 +00:00
"""
2019-10-05 05:14:13 +00:00
dirty = "".join([c.upper() for c in dirty if c in string.ascii_letters])
2017-08-19 02:41:46 +00:00
clean = ""
2019-10-05 05:14:13 +00:00
2017-08-19 02:41:46 +00:00
if len(dirty) < 2:
return dirty
2019-10-05 05:14:13 +00:00
for i in range(len(dirty) - 1):
2017-08-19 02:41:46 +00:00
clean += dirty[i]
2019-10-05 05:14:13 +00:00
if dirty[i] == dirty[i + 1]:
clean += "X"
2017-08-19 02:41:46 +00:00
clean += dirty[-1]
if len(clean) & 1:
2019-10-05 05:14:13 +00:00
clean += "X"
2017-08-19 02:41:46 +00:00
return clean
2019-10-05 05:14:13 +00:00
def generate_table(key: str) -> list[str]:
# I and J are used interchangeably to allow
2017-08-19 02:41:46 +00:00
# us to use a 5x5 table (25 letters)
alphabet = "ABCDEFGHIKLMNOPQRSTUVWXYZ"
2019-10-05 05:14:13 +00:00
# we're using a list instead of a '2d' array because it makes the math
2017-08-19 02:41:46 +00:00
# for setting up the table and doing the actual encoding/decoding simpler
table = []
# copy key chars into the table if they are in `alphabet` ignoring duplicates
for char in key.upper():
if char not in table and char in alphabet:
table.append(char)
# fill the rest of the table in with the remaining alphabet chars
for char in alphabet:
if char not in table:
table.append(char)
return table
2019-10-05 05:14:13 +00:00
def encode(plaintext: str, key: str) -> str:
2017-08-19 02:41:46 +00:00
table = generate_table(key)
plaintext = prepare_input(plaintext)
ciphertext = ""
# https://en.wikipedia.org/wiki/Playfair_cipher#Description
for char1, char2 in chunker(plaintext, 2):
row1, col1 = divmod(table.index(char1), 5)
row2, col2 = divmod(table.index(char2), 5)
if row1 == row2:
2019-10-05 05:14:13 +00:00
ciphertext += table[row1 * 5 + (col1 + 1) % 5]
ciphertext += table[row2 * 5 + (col2 + 1) % 5]
2017-08-19 02:41:46 +00:00
elif col1 == col2:
2019-10-05 05:14:13 +00:00
ciphertext += table[((row1 + 1) % 5) * 5 + col1]
ciphertext += table[((row2 + 1) % 5) * 5 + col2]
else: # rectangle
ciphertext += table[row1 * 5 + col2]
ciphertext += table[row2 * 5 + col1]
2017-08-19 02:41:46 +00:00
return ciphertext
def decode(ciphertext: str, key: str) -> str:
2017-08-19 02:41:46 +00:00
table = generate_table(key)
plaintext = ""
# https://en.wikipedia.org/wiki/Playfair_cipher#Description
for char1, char2 in chunker(ciphertext, 2):
2017-08-19 02:41:46 +00:00
row1, col1 = divmod(table.index(char1), 5)
row2, col2 = divmod(table.index(char2), 5)
if row1 == row2:
2019-10-05 05:14:13 +00:00
plaintext += table[row1 * 5 + (col1 - 1) % 5]
plaintext += table[row2 * 5 + (col2 - 1) % 5]
2017-08-19 02:41:46 +00:00
elif col1 == col2:
2019-10-05 05:14:13 +00:00
plaintext += table[((row1 - 1) % 5) * 5 + col1]
plaintext += table[((row2 - 1) % 5) * 5 + col2]
else: # rectangle
plaintext += table[row1 * 5 + col2]
plaintext += table[row2 * 5 + col1]
2017-08-19 02:41:46 +00:00
return plaintext