如何用Pythonic方式实现JS中return内嵌套多条件判断的等效逻辑

你可以通过拆分逻辑+惯用的Python语法实现完全等效的功能，完全不需要强行把所有逻辑塞在return语句中，以下是两种符合PEP8规范的实现方式：

1. 直观改写版（易理解，适合逻辑固定的场景） #

假设你已经实现了对应辅助函数get_cols、get_query、format_sql（避免和Python内置format重名）、value：

def build_query(query: dict, table: str) -> str:
    condition_parts = []
    for op, val in query.items():
        if op == "$or":
            sub_conditions = get_query(val, table)
            condition_parts.append(f"({' OR '.join(sub_conditions)})")
        elif op == "$null":
            condition_parts.append(format_sql("%s IS NULL", val))
        elif op == "$notnull":
            condition_parts.append(format_sql("%s IS NOT NULL", val))
        elif op == "$not":
            sub_condition = get_query(val, table)
            condition_parts.append(f"NOT ({sub_condition})")
        elif op == "$arrayany":
            sub_parts = [
                format_sql("%s=ANY(%L)", get_cols(sub_k, table), value(sub_v))
                for sub_k, sub_v in val.items()
            ]
            condition_parts.append(" AND ".join(sub_parts))
    # 所有条件拼接后整体包在括号中返回
    return f"({' '.join(condition_parts)})"

2. 调度字典优化版（易扩展，适合后续要加新操作符的场景） #

用字典映射替代长if-elif链，是Python中处理多分支映射的常用方案，后续新增操作符只需要加对应的处理函数和映射项即可：

# 各操作符对应的处理逻辑
def handle_or(val: str, table: str) -> str:
    return f"({' OR '.join(get_query(val, table))})"

def handle_null(val: str, table: str) -> str:
    return format_sql("%s IS NULL", val)

def handle_notnull(val: str, table: str) -> str:
    return format_sql("%s IS NOT NULL", val)

def handle_not(val: str, table: str) -> str:
    return f"NOT ({get_query(val, table)})"

def handle_arrayany(val: dict, table: str) -> str:
    sub_parts = [
        format_sql("%s=ANY(%L)", get_cols(sub_k, table), value(sub_v))
        for sub_k, sub_v in val.items()
    ]
    return " AND ".join(sub_parts)

# 操作符-处理函数映射表
OP_HANDLER_MAP = {
    "$or": handle_or,
    "$null": handle_null,
    "$notnull": handle_notnull,
    "$not": handle_not,
    "$arrayany": handle_arrayany,
}

def build_query(query: dict, table: str) -> str:
    condition_parts = [
        OP_HANDLER_MAP[op](val, table)
        for op, val in query.items()
    ]
    return f"({' '.join(condition_parts)})"

注意事项 #

• Python社区优先保证可读性，不需要强行把所有逻辑压缩到return语句内，拆分后的逻辑性能和压缩写法没有差异
• 列表推导比map()更符合Python惯用写法，可读性更高
• 字符串拼接优先使用f-string，比+拼接更简洁，也不容易出现括号匹配错误
• 原JS代码示例存在未闭合的括号，实际实现时请注意核对拼接后的SQL括号匹配逻辑

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