diff --git a/05-caesar-cipher/caesar.py b/05-caesar-cipher/caesar.py index 80eb5e3..185fa80 100644 --- a/05-caesar-cipher/caesar.py +++ b/05-caesar-cipher/caesar.py @@ -1,5 +1,5 @@ def char_shift(char, shift): - start = 65 if char.isupper() else 97 + start = ord('a') if char.isupper() else ord('A') char = ord(char) - start shift = (char + shift) % 26 return chr(shift+start) @@ -11,6 +11,9 @@ def caesar(message, shift): letter = char_shift(letter, shift) out += letter + # or in one line + # out += char_shift(letter, shift) if letter.isalpha() else letter + return out def decrypt_caesar(message, shift):