python - 有没有办法告诉用户输入错误？

Question

如何以一种告诉我的用户数据库中不可用的输入的方式进行编码？

这是我的代码：

def sql_getage(name):
    query = """\
        select age from store
        where name = '{}'
        """.format(name)
    results = (execute_read_query (conn, query))
    for i in results:
        theresult = str(i[0])
    return (theresult)
name = input("Please input your username")
age = sql_getage(name)
print("Your age is", age)

我的数据库有 2 列：

username    Age
John        20
Amy         21

我该如何编码，如果用户输入名称为 Jhn，并且在数据库表中找不到它，那么我会打印wrong username

目前如果我输入错误的名字，它仍然会继续下一个代码，即打印年龄。它会打印一个空的年龄。

标签： pythonsql-server

Answer 1

如果未找到名称，results将是一个空列表。您的函数可以检查这一点，并在这种情况下返回None：

results = (execute_read_query (conn, query))
if not results:
    return None

然后在调用代码中，您可以检查None返回值：

age = sql_getage(name)
if age is None:
    print("Name not found")
else:
    print("Your age is", age)

此外，请更改您的查询以使用参数而不是字符串格式设置名称值。您正在为 SQL 注入攻击敞开大门。