Python实现Caesar Cipher解密异常及大偏移量处理求助

问题描述 #

需要实现Caesar密码，字符集要求为小写字母在前、大写字母在后的52个字符（例如a偏移26位变为A）。目前编写的带命令行选项的Python代码存在两个问题：

1. 加密后部分字母无法正确解密；
2. 尝试支持大偏移量失败，需咨询实现方法。

现有代码 #

import argparse
import string

# 命令行参数解析
def parse_args():
    parser = argparse.ArgumentParser('Caesar Cypher')
    parser.add_argument('--msg', '-MSG', action='store', help='message to be encrypted/decrypted', type=str, default='Hello World!')
    parser.add_argument('--decrypt', action='store_true', help='turn on decryption mode', default=False)
    parser.add_argument('--shift', '-SHIFT', action='store', help='shift characters by SHIFT positions', type=int, default=4)
    return parser.parse_args()

# 加密函数
def shift(msg, shifted: int):
    as1 = string.ascii_lowercase
    as2 = string.ascii_uppercase
    al = list(as1 + as2)
    
    for char in msg:
        if char.isalpha():
            crypted_index = (al.index(char) + shifted) % len(al)
            crypted_char = al[crypted_index]
            msg = msg.replace(char, crypted_char)   
    return msg

# 解密函数
def decrypt(msg, shifted: int):
    as1 = string.ascii_lowercase
    as2 = string.ascii_uppercase
    al = list(as1 + as2)
    
    for char in msg:
        if char.isalpha():
            crypted_index = (al.index(char) - shifted) % len(al)
            crypted_char = al[crypted_index]
            msg = msg.replace(char, crypted_char)   
    return msg

# 主函数
if __name__ == '__main__':
    args = parse_args()
    
    # 解密时的偏移处理（此处存在错误）
    backshift = args.shift if not args.decrypt else args.decrypt
    
    crypted_cypher = shift(args.msg, backshift)
    print(crypted_cypher)

错误原因分析 #

1. 主函数逻辑错误：backshift赋值错误，args.decrypt是布尔值，解密时不应将布尔值作为偏移量传入，正确逻辑应该是调用decrypt函数，或者将偏移量设为负数后调用shift函数。
2. 字符串替换逻辑错误：使用msg.replace(char, crypted_char)会替换字符串中所有相同字符，例如Hello中的两个l，第一次替换后所有l变成目标字符，后续循环再次处理原字符时会出错，导致重复偏移。
3. 大偏移量未做模运算：未对偏移量取模52（字符集总长度），导致大偏移量计算时索引超出范围或重复计算，实际上偏移52位等于无偏移，大偏移量只需取模即可简化计算。

修复方案与大偏移量实现 #

核心修改点 #

• 替换msg.replace为逐个字符构建新字符串，避免批量替换导致的重复偏移；
• 主函数修正解密逻辑，通过正负偏移量区分加密/解密；
• 对偏移量做% 52处理，自动支持任意大小的偏移量（包括大偏移量）。

修复后的代码 #

import argparse
import string

def parse_args():
    parser = argparse.ArgumentParser('Caesar Cypher')
    parser.add_argument('--msg', '-MSG', action='store', help='需要加密/解密的消息', type=str, default='Hello World!')
    parser.add_argument('--decrypt', action='store_true', help='启用解密模式', default=False)
    parser.add_argument('--shift', '-SHIFT', action='store', help='字符偏移位数', type=int, default=4)
    return parser.parse_args()

def caesar_transform(msg, shift_amount: int):
    # 定义52个字符的顺序：小写在前，大写在后
    alphabet = string.ascii_lowercase + string.ascii_uppercase
    alpha_len = len(alphabet)
    # 处理大偏移量：对字符集长度取模，减少计算量
    shift_amount = shift_amount % alpha_len
    result = []
    
    for char in msg:
        if char in alphabet:
            original_idx = alphabet.index(char)
            new_idx = (original_idx + shift_amount) % alpha_len
            result.append(alphabet[new_idx])
        else:
            # 非字母字符直接保留
            result.append(char)
    return ''.join(result)

if __name__ == '__main__':
    args = parse_args()
    # 加密用正偏移，解密用负偏移
    shift_val = args.shift if not args.decrypt else -args.shift
    transformed_msg = caesar_transform(args.msg, shift_val)
    print(transformed_msg)

说明 #

1. 合并加密解密函数：通过传入正负偏移量，将加密和解密逻辑合并为一个caesar_transform函数，减少重复代码；
2. 逐个字符处理：用列表收集处理后的字符，最后拼接成字符串，避免批量替换的问题；
3. 大偏移量支持：shift_amount % alpha_len确保偏移量始终在0-51之间，即使传入100、200这样的大数值，也能正确计算；
4. 非字母字符保留：原代码中未处理非字母字符，修复后直接保留，保证消息中的标点、空格等不受影响。

内容的提问来源于stack exchange，提问作者patrick