编写python程序，有会做的一起交流

编写程序，实现凯撒密码。程序接收用户输入待加密的明文，输出加密后的密文。设想在某些情况下给朋友传递字条信息，但又不希望传递中途被第三方看懂这些信息，因此需要对字条信息进行加密处理。凯撒密码采用替换算法：对信息中的每一个英文字符循环替换为该字符后第四个字符，对应关系如下： 

原文：A B C D E F G H I J K L M N O P Q R S T U V W X Y Z 

密文：E F G H I J K L M N O P Q R S T U V W X Y Z A B C D

其他字符保持不变。 

code = input()
new_code = ''
for i in code:
    if i<='V' and i >= 'A':
        new_code += chr(ord(i)+4)
    elif i =='W':
        new_code += 'A'
    elif i =='X':
        new_code += 'B'
    elif i =='Y':
        new_code += 'C'
    elif i =='Z':
        new_code += 'D'
    else:
        new_code += i
print(new_code)

加上小写字母。。。

a = [chr(i) for i in range(65,91)]
a1 = [chr(i) for i in range(97,123)]
b = a[4:] + a[:4] + a1[4:] + a1[:4]
a += a1
c = {a[i]:b[i] for i in range(52)}

code = input()
res = ''
for i in code:
    res += c.get(i) if c.get(i) else i

print(res)