python 凯撒密码

import string


def caesar_cipher(text):
    """接收一个字符串为参数，采用字母表和数字中后面第3个字符代替当前字符的方法
    对字符串中的字母和数字进行替换，实现加密效果，返回值为加密的字符串。
    例如：2019 abc 替换为5342 def """
    lowers = string.ascii_lowercase  # lowers是全部的小写英文字母
    uppers = string.ascii_uppercase  # uppers是全部的大写英文字母
    digits = string.digits
    before = string.ascii_letters + digits
    after = lowers[3:] + lowers[:3] + uppers[3:] + uppers[:3] + digits[3:] + digits[:3]
    table = ''.maketrans(before, after)
    place = text.translate(table)
    return place


if __name__ == '__main__':
    plaintext = input()
    print(caesar_cipher(plaintext))