新版红绿灯规则自动判断-python版

[python]代码库

RED = 0
OFF = 1
GREEN = 2


def left(l, m, r):
    # 左灯红 = 停
    if (l == RED):
        return False
    # 左灯绿 或 中间绿 = 走
    return l == GREEN or m == GREEN


def direct(l, m, r):
    # 中间绿 = 走
    return m == GREEN


def right(l, m, r):
    # 右灯关 或 右灯绿 = 走
    return r >= OFF


def check(l, m, r):
    return '左转', left(l, m, r), '直行', direct(l, m, r), '右转', right(l, m, r)


if __name__ == '__main__':
    print(check(RED, RED, RED))
    print(check(OFF, GREEN, OFF))
    print(check(RED, GREEN, OFF))
    print(check(OFF, GREEN, RED))
    print(check(GREEN, RED, RED))
    print(check(RED, RED, OFF))
    print(check(RED, GREEN, RED))
    print(check(OFF, RED, OFF))

[代码运行效果截图]