如何消除elif语句编写整洁代码？Python清洁代码优化建议

原代码存在的问题 #

• 堆叠了7个elif分支，后续新增MIME类型需要持续追加分支，维护性差
• 子串匹配逻辑有缺陷：使用'.gif' in filename这类判断会误判，比如文件名为test.gif.exe时会被错误识别为gif类型，实际后缀应为exe
• 存在冗余分支：.jpg和.jpeg两个分支执行完全相同的输出逻辑，代码重复
• 变量命名不规范：nameoffile为连写形式，不符合Python PEP8命名规范
• 逻辑和输出耦合：每个分支直接执行print操作，后续如果要调整输出逻辑需要修改所有分支

优化方案：用字典映射替代elif分支 #

对于这种固定键值匹配的多分支场景，最适合用字典做映射表替代长串elif，把匹配关系和判断逻辑解耦。
优化后代码如下：

import os

# 读取输入并做标准化处理
file_name = input('File name: ').strip().lower()
# 精确提取文件后缀，避免子串误判
_, file_suffix = os.path.splitext(file_name)

# MIME类型映射表，集中维护所有匹配规则
MIME_TYPE_MAP = {
    '.gif': 'image/gif',
    '.jpg': 'image/jpeg',
    '.jpeg': 'image/jpeg',
    '.png': 'image/png',
    '.pdf': 'application/pdf',
    '.txt': 'text/plain',
    '.zip': 'application/zip'
}

# 匹配不到时返回默认MIME类型
mime_type = MIME_TYPE_MAP.get(file_suffix, 'application/octet-stream')
print(mime_type)

如果不想引入os模块，也可以通过字符串切割自行提取后缀，注意处理无后缀的边界场景：

# 替代os.path.splitext的纯Python实现
if '.' in file_name:
    file_suffix = '.' + file_name.split('.')[-1]
else:
    file_suffix = ''

额外的清洁代码建议 #

• 遵循PEP8命名规范：变量使用蛇形命名（如file_name），固定配置的常量使用全大写下划线分隔命名（如MIME_TYPE_MAP），提升代码可读性
• 优先使用标准库处理路径/文件名相关逻辑：os.path.splitext可以覆盖各种边界场景（比如无后缀文件、点开头的隐藏文件、多后缀文件），比自己写字符串切割逻辑更稳妥
• 配置与逻辑分离：所有MIME匹配规则集中在映射表中维护，后续新增/修改类型只需要调整字典内容，不需要改动核心判断逻辑
• 逻辑与输出解耦：先计算得到最终的mime_type值，再统一执行输出，后续如果需要把结果返回给接口、写入日志，只需要修改最后一行的输出逻辑即可
• 输入预处理：对输入的文件名先做strip()去除首尾空白字符，避免用户误输入空格导致匹配失败

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