Python数据库问题：空格替换后IF语句未执行，直接进入ELSE分支

问题代码 #

game=input("enter student name to update report card: ")
nn=game.replace("  ", "_")
mycursor.execute("show tables")
klm = mycursor.fetchall()
if (nn,) in klm:
    b=int(input("enter sno: "))
    mycursor.execute("select * from {} where sno='{}'".format(nn,b))
    xer=mycursor.fetchall()
        
else:
    print("no student record found")

问题描述 #

运行上述代码操作数据库时，意图将输入字符串中的空格替换为下划线后判断对应表是否存在，但数据库中明明有名为some_one的表，程序却直接执行ELSE分支输出"no student record found"，IF语句未触发。

原因分析 #

核心问题出在空格替换逻辑：

• 代码里的replace(" ", "_")只会把两个连续空格替换成下划线，如果你输入的是单个空格（比如some one），这个替换操作完全不会生效，最终nn的值还是some one，和数据库里的some_one表名不匹配，自然触发不了IF分支。
• 少数情况可能是show tables返回的表名格式问题（比如部分驱动返回大写表名），但优先排查替换逻辑。

解决步骤 #

1. 修复空格替换逻辑
   如果要把所有单个空格替换成下划线，直接修改为：

   nn = game.replace(" ", "_")
   

   如果要把多个连续空格合并成一个下划线（比如输入some one变成some_one），用正则处理：

   import re
   nn = re.sub(r'\s+', '_', game)  # 匹配任意数量空白字符，替换为单个下划线
   

2. 添加调试输出验证
   在代码中加入打印语句，直观查看处理后的表名和数据库表列表，确认匹配情况：

   game=input("enter student name to update report card: ")
   nn=game.replace(" ", "_")
   print("处理后的表名：", nn)  # 新增调试打印
   mycursor.execute("show tables")
   klm = mycursor.fetchall()
   print("数据库中的表：", klm)  # 新增调试打印
   if (nn,) in klm:
       # 后续代码不变
   

3. 额外提醒：规避SQL注入风险
   原代码用format拼接SQL语句存在严重注入风险，建议改用参数化查询（不同数据库驱动占位符不同，例如MySQL用%s）：

   # MySQL示例
   mycursor.execute("select * from %s where sno=%s", (nn, b))
   

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