基于用户输入的SQL列选择Python实现方案咨询

核心思路 #

用字典直接映射用户输入（A/B/C）到对应的目标列列表，这种方式比嵌套if-else更直观，后续规则调整也只需要修改字典，维护成本低。

代码实现 #

# 定义输入与目标列的映射表
input_to_columns = {
    'A': ['Column B'],
    'B': ['Column A', 'Column B'],
    'C': ['Column A']
}

def build_select_sql(user_input, table='target_table'):
    # 校验输入合法性
    if user_input not in input_to_columns:
        raise ValueError("输入只能是A、B、C其中一个")
    # 拼接列名
    selected_columns = ', '.join(input_to_columns[user_input])
    # 生成最终SQL语句
    return f"SELECT {selected_columns} FROM {table};"

使用示例 #

• 当输入为A时：

print(build_select_sql('A'))
# 输出：SELECT Column B FROM target_table;

• 当输入为B时：

print(build_select_sql('B'))
# 输出：SELECT Column A, Column B FROM target_table;

• 当输入为C时：

print(build_select_sql('C'))
# 输出：SELECT Column A FROM target_table;

方案优势 #

• 简洁易维护：规则全部存在映射字典里，新增输入类型或调整列选择时，改字典就行，不用动逻辑代码
• 合法性校验：提前拦截非法输入，避免生成无效SQL
• 复用性强：生成SQL的逻辑封装成函数，随便哪个地方需要都能直接调用

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