FilesExpand file tree

Write-for-current-Python

/

atbash.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

66 lines (55 loc) · 1.56 KB

Write-for-current-Python

/

atbash.py

Copy path

Top

File metadata and controls

• Code
• Blame

66 lines (55 loc) · 1.56 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

""" 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()