Encryption and Decryption within the alphabet - Python GCSE

限于喜欢 提交于 2019-12-02 13:36:06
def getOffset():
    offset = -1
    while not 0 < offset < 25:
        try: offset = int(input("Please enter an integer offset between 1 and 25: "))
        except ValueError: continue
    return offset


def encrypt(msg, offset):
    base = ord('a')
    chars = (ord(char)-base for char in msg)
    cipher = (base + ((c+offset)%26) for c in chars)
    return ''.join([chr(c) for c in cipher])


decrypt = lambda msg,offset: encrypt(msg, -offset)


def main():
    msg = """Please choose 
[1] to encrypt
[2] to decrypt
[3] to exit
"""

    funcs = {1: encrypt, 2: decrypt}

    while True:
        try: choice = int(input(msg))
        except ValueError: continue

        if choice == 3: return
        message = input("Please enter the message that you would like to {}: ".format(funcs[choice].__name__))
        offset = int(input("Please enter the offset: "))
        print(funcs[choice](message, offset))
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!