一个python脚本,用于检查数据库表列是否存在

时间:2014-02-11 17:10:20

标签: python postgresql

我希望编写一个python脚本来检查数据库表中的特定列,如果它不存在则添加它,这可能吗?如果有,是否有任何文件指出我正确的方向? 提前谢谢!

1 个答案:

答案 0 :(得分:1)

使用MySQLdb连接到计算机的数据库并创建游标后,您可以先执行以下查询。

import MySQLdb

conn = MySQLdb.connect(host='localhost', db='test', user='root')
cursor = conn.cursor()

cursor.execute("select * from information_schema.COLUMNS where table_name = 'table-name' and column_name = 'column-name' ")

_filed = cursor.fetchall()

if len(_filed) == 0, u can execute another alter query to add the column

cursor.execute('alter table table-name add column <column-name> int(11) default 0')