使用Python检查PostgreSQL表中是否存在一列

Question

我搜索了很长时间以寻找答案，并做了以下工作：

（1）

query = "SELECT COUNT(*) FROM %s WHERE user_name = %s" % (table_name, username)
result = conn.query(query).result()[0][0]
if result == 0:
    (do something)

（2）

query = "SELECT 1 FROM %s WHERE user_name = %s" %(table_name, username)
result = conn.query(query).result()[0][0]
if result == ' ':
    (do something)

（3）

query = "SELECT EXISTS (SELECT 1 FROM %s WHERE user_name = %s)" %(table_name, username)
result = conn.query(query).result()[0][0]
if result == 't':
    (do something)

但是所有方法都不起作用...错误始终是：

column "Tom" does not exist

由于它确实不存在，我只想检查它是否存在。 任何帮助表示赞赏。

Answer 1

sql = """SELECT count(*)
         FROM information_schema.columns
         WHERE table_name = '%s'
         AND column_name = '%s'
      """ % (thetable,thecolumn)

Answer 2

您没有引用字符串。

一旦查询到PostgreSQL，您将查询如下：

SELECT COUNT(*) FROM Table WHERE user_name = Tom

user_name列与不存在的Tom列进行比较。

您想要的是这样的查询：

SELECT COUNT(*) FROM Table WHERE user_name = 'Tom'

为此，您应该使用参数化语句，以避免SQL注入的任何可能性。 使用DBAPI模块，它很简单：

cursor = conn.cursor()
cursor.execute('SELECT COUNT(*) FROM Table WHERE user_name = %s', user_name)

如果您还需要表名是动态的，则必须像这样构造它：

cursor = conn.cursor()
query = 'SELECT COUNT(*) FROM %s WHERE user_name = %%s' % table_name
cursor.execute(query, user_name)

但是您需要绝对确定表名是有效的。 任何可以控制该变量的人都可以对您的数据库做任何事情。

Answer 3

永远记住： EAFP

try:
   result = conn.query(query).result()
except ProgrammingError:  # Edumacated guess based on the documentation
   # however untested so psycopg2 raises some other instead
   pass # or do_something_when_it_does_not_exist()
else:
   (do something)