ROT-code/rot.py

17 lines
461 B
Python
Executable File

#!/usr/bin/env python3
from string import ascii_lowercase as lc, ascii_uppercase as uc
from sys import argv
def rot(m: str, n: int):
return m.translate(m.maketrans(lc + uc, lc[n:] + lc[:n] + uc[n:] + uc[:n]))
if __name__ == "__main__":
try:
a=argv[1]
b=int(argv[2])
except IndexError:
a=input("Give text to encode/decode: ")
b=int(input("Type how many shifts do you want to do: "))
print(f"Text: {rot(a, b)}")