ROT-code/rot.py

#!/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)}")