Skip to content

Commit

Permalink
update rot13.py (TheAlgorithms#1790)
Browse files Browse the repository at this point in the history
* update rot13.py

* Update rot13.py

* Type hints, doctests, URL to Wikipedia

Co-authored-by: Christian Clauss <[email protected]>
  • Loading branch information
aakashdinkar and cclauss authored Mar 8, 2020
1 parent a9f73e3 commit 182e304
Showing 1 changed file with 18 additions and 6 deletions.
24 changes: 18 additions & 6 deletions ciphers/rot13.py
Original file line number Diff line number Diff line change
@@ -1,24 +1,36 @@
def dencrypt(s, n):
def dencrypt(s: str, n: int=13):
"""
https://en.wikipedia.org/wiki/ROT13
>>> msg = "My secret bank account number is 173-52946 so don't tell anyone!!"
>>> s = dencrypt(msg)
>>> s
"Zl frperg onax nppbhag ahzore vf 173-52946 fb qba'g gryy nalbar!!"
>>> dencrypt(s) == msg
True
"""
out = ""
for c in s:
if c >= "A" and c <= "Z":
if "A" <= c <= "Z":
out += chr(ord("A") + (ord(c) - ord("A") + n) % 26)
elif c >= "a" and c <= "z":
elif "a" <= c <= "z":
out += chr(ord("a") + (ord(c) - ord("a") + n) % 26)
else:
out += c
return out


def main():
s0 = "HELLO"
s0 = input("Enter message: ")

s1 = dencrypt(s0, 13)
print(s1) # URYYB
print("Encryption:", s1)

s2 = dencrypt(s1, 13)
print(s2) # HELLO
print("Decryption: ", s2)


if __name__ == "__main__":
import doctest
doctest.testmod()
main()

0 comments on commit 182e304

Please sign in to comment.