咨询Python字符串填充字符移除函数的最佳实现方案

嘿，我来帮你搞定这个移除填充的函数！先给你提个小细节：你现有的填充函数里有两个小问题——一是变量名大小写写错了（Message应该是小写的message），二是函数没有返回处理后的结果，这样你根本拿不到填充后的消息。先把填充函数修正得能用：

import random
import string

def add_padding():
    message = input('Enter your message: ')
    padding_char = random.choice(string.ascii_letters)
    extra = len(message) % 16
    if extra > 0:
        padded_message = message + (padding_char * (16 - extra))
        # 返回填充后的消息和所用的填充字符，方便后续去填充
        return padded_message, padding_char
    else:
        # 如果原消息长度已经是16的倍数，不需要填充
        return message, None

接下来是移除填充的核心逻辑：因为你的填充规则是用同一个随机字母补全到16的倍数，所以去填充的关键是要知道当时用了哪个填充字符，然后从消息末尾移除连续的该字符。

实现去填充函数 #

def remove_padding(padded_message, padding_char):
    # 如果没有填充字符，说明原消息本来就是16的倍数，直接返回
    if padding_char is None:
        return padded_message
    
    # 从末尾往前数，统计连续的填充字符数量
    padding_length = 0
    for char in reversed(padded_message):
        if char == padding_char:
            padding_length += 1
        else:
            break
    
    # 可选：加个校验，确保填充是有效的（避免恶意输入破坏原消息）
    original_length = len(padded_message) - padding_length
    if original_length % 16 != (len(padded_message) - padding_length) % 16:
        raise ValueError("无效的填充格式，请检查输入")
    
    # 移除末尾的填充字符，返回原消息
    return padded_message[:-padding_length]

另一种方案：把填充字符嵌入消息中 #

如果你不想返回元组，也可以把填充字符放在消息的开头，这样去填充时直接提取第一个字符即可：

改进的填充函数

def add_padding():
    message = input('Enter your message: ')
    extra = len(message) % 16
    if extra > 0:
        padding_char = random.choice(string.ascii_letters)
        # 把填充字符放在消息最前面
        padded_message = padding_char + message + (padding_char * (16 - extra))
        return padded_message
    else:
        return message

对应的去填充函数

def remove_padding(padded_message):
    # 如果消息长度不是16的倍数，说明没有填充，直接返回
    if len(padded_message) % 16 != 0:
        return padded_message
    
    # 提取开头的填充字符
    padding_char = padded_message[0]
    # 剩下的部分是原消息加填充
    content = padded_message[1:]
    
    # 统计末尾的填充字符数量
    padding_length = 0
    for char in reversed(content):
        if char == padding_char:
            padding_length += 1
        else:
            break
    
    return content[:-padding_length]

注意事项 #

• 一定要确保填充时的规则和去填充时完全对应，比如填充用的是同一个字符，去填充就只移除末尾连续的该字符
• 加入校验逻辑可以避免因为错误输入导致原消息被误截断，让函数更健壮

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