如何精简时间输入解析的多分支if/elif/else代码结构？

优化方向 #

针对连续多段if/elif重复判断的问题，最适合入门开发者的精简方案是用统一规则遍历替代硬编码分支，把每个时间单位的匹配逻辑从判断语句中抽离出来，和核心处理流程解耦。

先修正原代码的隐藏逻辑漏洞：原hr分支中提取数值时写的是amount[:-1]，但匹配的是长度为2的hr后缀，正确应该通过amount[:-2]提取数字部分，否则会把末尾的r纳入数值校验范围，根本无法匹配6hr这类合法输入。

可直接落地的代码实现 #

首先把所有支持的时间单位的匹配规则整理成独立的配置项，每条规则包含匹配后缀、需要从输入末尾切掉的字符长度、换算为秒的倍率、对应的单位名称，后续只需要遍历规则做统一判断即可，完全替代原来的连续分支：

# 时间单位规则配置：(匹配后缀, 末尾切除长度, 秒级换算倍率, 单位标识)
TIME_MATCH_RULES = [
    ("s", 1, 1, "seconds"),
    ("m", 1, 60, "minutes"),
    ("hr", 2, 60 * 60, "hours"),
    ("d", 1, 60 * 60 * 24, "days"),
]

parsed_seconds = None
matched_unit = None

# 统一规则匹配流程，替代原有的if/elif链
for suffix, cut_length, scale, unit_label in TIME_MATCH_RULES:
    # 输入长度不足时直接跳过，避免字符串切片出错
    if len(amount) <= cut_length:
        continue
    # 统一的校验逻辑，所有单位共用
    if amount.endswith(suffix) and amount[:-cut_length].isnumeric():
        parsed_seconds = int(amount[:-cut_length]) * scale
        matched_unit = unit_label
        break

# 保留原有的reason兜底逻辑
if parsed_seconds is None:
    # 兼容原逻辑，避免拼接出多余空格
    reason = f"{amount} {reason}".strip() if reason is not None else amount
else:
    # 按原代码的方式存储解析后的结果
    amount = parsed_seconds
    unit = matched_unit

方案的实际收益 #

• 消除重复代码：原来每个分支都要重复写「判断后缀、校验剩余字符是否为数字、转整数乘倍率」的逻辑，现在这部分通用逻辑只写一次，所有时间单位共用
• 维护成本极低：后续如果要新增支持的时间单位（比如w代表周、y代表年），只要在TIME_MATCH_RULES列表里加一行配置就行，完全不用改动核心匹配逻辑
• 降低出错概率：统一的切片、校验逻辑不会出现原代码里hr分支切错长度这类疏漏，所有单位的处理逻辑保持一致
• 代码可读性强：没有跳来跳去的分支跳转，顺着读代码就能看懂整个解析流程，不会随着支持的单位变多，越写越缠成难以维护的“意大利面条代码”

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