python求密码翻译

Question

密码翻译
在情报传递过程中，为了防止情报被截获，往往需要对情报用一定的方式加密，简单的加密算法虽然不足以完全避免情报被破译，但仍然能防止情报被轻易的识别。我们给出一种最简的的加密方法，对给定的一个字符串，把其中从a-y，A-Y的字母用其后继字母替代，把z和Z用a和A替代，其他非字母字符不变，则可得到一个简单的加密字符串。
输入1
Hello! How are you!
输出1
Ifmmp! Ipx bsf zpv!
输入2
ZLX,xnt zqd udqx fnnc!
输出2
AMY,you are very good!

def encrypt(s):
    result = ''
    for char in s:
        if char.isalpha():
            if char == 'y':
                result += 'z'
            elif char == 'z':
                result += 'b'
            elif char == 'Y':
                result += 'A'
            elif char == 'Z':
                result += 'B'
            else:
                result += chr(ord(char) + 1)
        else:
            result += char
    return result
s =input()
encrypted_s = encrypt(s)
print(encrypted_s)

Answer 1

def encrypt(s):
    result = ''
    for char in s:
        if char.isalpha():
            if char == 'z':
                result += 'a'
            elif char == 'Z':
                result += 'A'
            else:
                 result += chr(ord(char) + 1)
        else:
            result += char
    return result

s =input()
encrypted_s = encrypt(s)
print(encrypted_s)