Python与SQL交互时的输入错误处理及查询报错问题咨询

1. SQL查询报错问题 #

• 核心原因是你直接用f-string拼接SQL语句时，没有给字符串类型的参数加单引号，执行时SQL会把输入的Plane识别为列名而非查询值，同时这种写法还存在SQL注入风险。
• 正确写法要用参数化查询，示例：

def elementcall(input_ele):
    # 用?作为参数占位符（不同数据库占位符可能有区别，比如PostgreSQL用%s，按需调整）
    cur.execute("SELECT Name FROM iElements WHERE Name=?", (input_ele,))
    cols = cur.fetchall()
    # 原来的while循环是无效的，fetchall调用一次就会拿完所有结果，不需要循环调用
    return cols

2. 原始循环校验失效问题 #

你直接判断elementinput in iElements['Name']时，iElements['Name']是Pandas的Series对象，in操作默认判断的是索引是否存在，不是判断值是否在列中，所以才会无论输入是否正确都触发错误。

你调整后的代码还不能满足需求，存在两个问题：

• 没有实现循环重试逻辑，用户输错一次就结束流程，不符合你「用循环实现更优输入体验」的要求
• 如果你这段代码是写在全局作用域而非函数内部，return elem会直接抛出语法错误，全局作用域不能直接用return

关于代码逻辑存放的说明 #

不需要把全部业务逻辑都放在循环内，校验通过的输入字符串可以提取到循环外使用，只要你在循环内部给变量赋值后通过break跳出循环即可。

# 提前把合法名称列表读出来，不用每次校验都读
valid_names = iElements['Name'].tolist()
valid_input = ""

# 循环校验输入
while True:
    elem = input('Enter a System Name:')
    if elem in valid_names:
        valid_input = elem
        break
    print("You typed something wrong! Please re-enter.")

# 循环外使用校验后的输入执行查询
result = elementcall(valid_input)
# 后续业务逻辑可以直接写在这

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