def displayResult(self):
name = input("> ")
self.cursor.execute("SELECT username FROM customer Where username = '%s' " % name)
if name in self.cursor.fetchone():
print("flag")
elif name not in self.cursor.fetchone():
print("Not in here.")
elif name == "" or name == " ":
print("Input something.")
我尝试将函数用于:当用户输入任何用户名时,如果用户名在数据库中,它将打印' flag',如果不在数据库中或用户没有输入任何东西,它将&# 34;打印不在这里", 因此,当我运行此代码并尝试输入一些用户名不在数据库中并尝试将其留空时,我发现此错误
TypeError: argument of type 'NoneType' is not iterable
如何关闭此错误警告
由于
答案 0 :(得分:0)
在尝试迭代之前尝试检查fetchone
的返回值,例如:
one = self.cursor.fetchone()
if one is not None:
if name in one:
print("flag")
# etc