atbash.py

""" https://en.wikipedia.org/wiki/Atbash """
import string


def atbash_slow(sequence: str) -> str:
 """
 >>> atbash_slow("ABCDEFG")
 'ZYXWVUT'

 >>> atbash_slow("aW;;123BX")
 'zD;;123YC'
 """
 output = ""
 for i in sequence:
 extract = ord(i)
 if 65 <= extract <= 90:
 output += chr(155 - extract)
 elif 97 <= extract <= 122:
 output += chr(219 - extract)
 else:
 output += i
 return output


def atbash(sequence: str) -> str:
 """
 >>> atbash("ABCDEFG")
 'ZYXWVUT'

 >>> atbash("aW;;123BX")
 'zD;;123YC'
 """
 letters = string.ascii_letters
 letters_reversed = string.ascii_lowercase[::-1] + string.ascii_uppercase[::-1]
 return "".join(
 letters_reversed[letters.index(c)] if c in letters else c for c in sequence
 )


def benchmark() -> None:
 """Let's benchmark them side-by-side..."""
 from timeit import timeit

 print("Running performance benchmarks...")
 print(
 "> atbash_slow()",
 timeit(
 "atbash_slow(printable)",
 setup="from string import printable ; from __main__ import atbash_slow",
 ),
 "seconds",
 )
 print(
 "> atbash()",
 timeit(
 "atbash(printable)",
 setup="from string import printable ; from __main__ import atbash",
 ),
 "seconds",
 )


if __name__ == "__main__":
 for example in ("ABCDEFGH", "123GGjj", "testStringtest", "with space"):
 print(f"{example} encrypted in atbash: {atbash(example)}")
 benchmark()