from cs50 import get_string import sys
def main(): if len(sys.argv) != 2: print("usage: python caesar.py key") exit(1) key = int(sys.argv[1]) translated = [] plaintext = get_string("Plaintext: ", end="")
for symbol in plaintext:
if symbol.isalpha():
translated.append(caesar(symbol, key))
else:
translated.append(symbol)
print("".join(translated))
exit(0)
def caesar(char, key):
if char.isupper():
return chr(((ord(char) - 65 + key) % 26) + 65)
else:
return(((ord(char) - 97 + key) % 26) + 97)
if name == "__ main__": main()