Python条件分支返回None原因及车牌校验代码优化问题

校验规则回顾 #

需要实现的车牌合法性判定规则如下：

• 前两位必须为字母
• 总字符长度在2-6位之间
• 仅允许包含字母和数字
• 若包含数字：
  • 首位出现的数字不能为0
  • 数字必须全部连续位于车牌末尾，数字后不允许再出现字母

问题复现 #

原有代码运行时，valid_order()函数中len(l) == 4分支执行后无有效返回值，其余长度分支可正常运行。例如输入CSAA50时，预期返回Valid，实际返回Invalid。
原有代码如下：

def main():
    plate = input("Plate: ")
    if is_valid(plate):
        print("Valid")
    else:
        print("Invalid")

def is_valid(s):
    if s[0:2].isalpha() and 6 >= len(s) >= 2 and s.isalnum() and valid_order(s):
        return True
    else:
        return False


def valid_order(c):
    n = []
    l = list(c[2:len(c)])

    for i in c:
        if i.isdigit():
            n += i
    if n and n[0] == "0":
        return False

    if len(l) == 2:
        if l[0].isdigit() and l[1].isalpha():
            return False
    if len(l) == 3:
        if l[0].isdigit():
            if l[1].isalpha() or l[2].isalpha():
                return False
        else:
            if l[1].isdigit() and l[2].isalpha():
                return False

    if len(l) == 4:
        if l[0].isdigit():
            if l[1].isalpha() or l[2].isalpha() or l[3].isalpha():
                return False
        else:
            if l[1].isdigit():
                if l[2].isalpha() or l[3].isalpha():
                    return False
            else:
                if l[2].isdigit() and l[3].isalpha():
                    return False
    else:
        return True

main()

Bug原因 #

问题出在valid_order的分支结构设计：

1. len(l) == 2、len(l) == 3的判断都是独立if语句，没有配套返回True的逻辑，只有最后len(l) == 4的判断写了else: return True。
2. 当l长度为4、且所有非法判定条件都不触发时，函数走完所有分支后没有显式return True语句，Python中函数无显式返回值时默认返回None，None在布尔判断中会被当作False处理，直接导致is_valid校验不通过。

以输入CSAA50为例：总长度为6，l是前两位之后的子串，即['A','A','5','0']，长度正好为4。判断过程中：

• l[0]为'A'（非数字），进入else分支
• l[1]为'A'（非数字），进入内层else分支
• l[2]为'5'（数字），l[3]为'0'（数字），不满足l[2].isdigit() and l[3].isalpha()的返回False条件
• 整个len(l)==4块执行完没有触发任何return，函数直接结束返回None，最终判定为Invalid。

更简洁的实现方案 #

原有实现按后缀长度硬编码枚举所有非法场景，可维护性极差，长度规则调整时需要新增大量分支。实际上不需要拆分字符串、枚举长度，单次遍历字符串即可完成数字顺序校验：

1. 遍历过程中标记是否已经遇到过数字
2. 第一次遇到数字时，判断该数字是否为0
3. 遇到数字之后如果再出现字母，直接判定为非法

优化后的完整代码：

def main():
    plate = input("Plate: ")
    print("Valid" if is_valid(plate) else "Invalid")

def is_valid(s):
    # 基础规则校验
    if not (2 <= len(s) <= 6 and s[:2].isalpha() and s.isalnum()):
        return False
    # 数字顺序校验
    digit_started = False
    for char in s:
        if char.isdigit():
            if not digit_started:
                # 首位数字不能为0
                if char == "0":
                    return False
                digit_started = True
        # 出现过数字后不能再出现字母
        elif digit_started:
            return False
    return True

main()

该实现无长度分支兼容所有合法长度的车牌，逻辑清晰，不存在隐式返回None的问题。

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