python编写改撒密码

import string
def kaisa_jiami(w,d):
    lower=string.ascii_lowercase
    upper=string.ascii_uppercase
    before=string.ascii_letters
    after=lower[d:]+lower[:d]+upper[d:]+upper[:d]
    table=''.maketrans(before,after)
    return w.translate(table)

def kaisa_jiemi(w,d):
    lower=string.ascii_lowercase
    upper=string.ascii_uppercase
    before=string.ascii_letters
    after=lower[d:]+lower[:d]+upper[d:]+upper[:d]
    table=''.maketrans(after,before)
    return w.translate(table)
def main():
    w=input("请输入要加密的字符串：")
    #将k转换为整数，默认输入为字符
    d=int(input("请输入一个整数密钥："))
    print('需要加密的字符串为：',w)
    l=kaisa_jiami(w,d)  
    print("加密后为：",l)  
    t= kaisa_jiemi(l,d)
    print("解密后为：",t)
if __name__=='__main__':
    main()