Python 实现数据库更新脚本的生成-演道网
主要操作如下:
1.在原先 basedao.py 中添加如下方法,这样旧能很方便的获取数据库的数据,为测试数据库和生产数据库做对比打下了基础。
def select_database_struts(self):
”’
查找当前连接配置中的数据库结构以字典集合
”’
sql = ”’SELECT COLUMN_NAME, IS_NULLABLE, COLUMN_TYPE, COLUMN_KEY, COLUMN_COMMENT
FROM information_schema.`COLUMNS`
WHERE TABLE_SCHEMA=”%s” AND TABLE_NAME=”{0}” ”’%(self.__database)
struts = {}
for k in self.__primaryKey_dict.keys():
self.__cursor.execute(sql.format(k))
results = self.__cursor.fetchall()
struts[k] = {}
for result in results:
struts[k][result[0]] = {}
struts[k][result[0]][“COLUMN_NAME”] = result[0]
struts[k][result[0]][“IS_NULLABLE”] = result[1]
struts[k][result[0]][“COLUMN_TYPE”] = result[2]
struts[k][result[0]][“COLUMN_KEY”] = result[3]
struts[k][result[0]][“COLUMN_COMMENT”] = result[4]
return self.__config, struts
2.编写对比的Python脚本
”’
数据库迁移脚本, 目前支持一下几种功能:
1.生成旧数据库中没有的数据库表执行 SQL 脚本(支持是否带表数据),生成的 SQL 脚本在 temp 目录下(表名.sql)。
2.生成添加列 SQL 脚本,生成的 SQL 脚本统一放在 temp 目录下的 depoyed.sql 中。
3.生成修改列属性 SQL 脚本,生成的 SQL 脚本统一放在 temp 目录下的 depoyed.sql 中。
4.生成删除列 SQL 脚本,生成的 SQL 脚本统一放在 temp 目录下的 depoyed.sql 中。
注意事项:
1.系统必须配置 mysql 的环境变量,确保 mysqldump 指令能执行。
2.数据库表必须带有主键。
”’
import json, os, sys
from basedao import BaseDao
temp_path = sys.path[0] + “/temp”
if not os.path.exists(temp_path):
os.mkdir(temp_path)
def main(old, new, has_data=False):
”’
@old 旧数据库(目标数据库)
@new 最新的数据库(源数据库)
@has_data 是否生成结构+数据的sql脚本
”’
clear_temp() # 先清理 temp 目录
old_config, old_struts = old
new_config, new_struts = new
for new_table, new_fields in new_struts.items():
if old_struts.get(new_table) is None:
gc_sql(new_config[“user”], new_config[“password”], new_config[“database”], new_table, has_data)
else:
cmp_table(old_struts[new_table], new_struts[new_table], new_table)
def cmp_table(old, new, table):
”’
对比表结构生成 sql
”’
old_fields = old
new_fields = new
sql_add_column = “ALTER TABLE `{TABLE}` ADD COLUMN `{COLUMN_NAME}` {COLUMN_TYPE} COMMENT ‘{COLUMN_COMMENT}’;\n”
sql_change_column = “ALTER TABLE `{TABLE}` CHANGE `{COLUMN_NAME}` `{COLUMN_NAME}` {COLUMN_TYPE} COMMENT ‘{COLUMN_COMMENT}’;\n”
sql_del_column = “ALTER TABLE `{TABLE}` DROP {COLUMN_NAME};”
if old_fields != new_fields:
f = open(sys.path[0] + “/temp/deploy.sql”, “a”, encoding=”utf8″)
content = “”
for new_field, new_field_dict in new_fields.items():
old_filed_dict = old_fields.get(new_field)
if old_filed_dict is None:
# 生成添加列 sql
content += sql_add_column.format(TABLE=table, **new_field_dict)
else:
# 生成修改列 sql
if old_filed_dict != new_field_dict:
content += sql_change_column.format(TABLE=table, **new_field_dict)
pass
# 生成删除列 sql
for old_field, old_field_dict in old_fields.items():
if new_fields.get(old_field) is None:
content += sql_del_column.format(TABLE=table, COLUMN_NAME=old_field)
f.write(content)
f.close()
def gc_sql(user, pwd, db, table, has_data):
”’
生成 sql 文件
”’
if has_data:
sys_order = “mysqldump -u%s -p%s %s %s > %s/%s.sql”%(user, pwd, db, table, temp_path, table)
else:
sys_order = “mysqldump -u%s -p%s -d %s %s > %s/%s.sql”%(user, pwd, db, table, temp_path, table)
try:
res = os.system(sys_order)
if res == 0:
print(“生成%s表sql文件。”%table)
except Exception as e:
print(e)
def clear_temp():
”’
每次执行的时候调用这个,先清理下temp目录下面的旧文件
”’
if os.path.exists(temp_path):
files = os.listdir(temp_path)
for file in files:
f = os.path.join(temp_path, file)
if os.path.isfile(f):
os.remove(f)
print(“临时文件目录清理完成”)
if __name__ == “__main__”:
test1_config = {
“user” : “root”,
“password” : “root”,
“database” : “test1”,
}
test2_config = {
“user” : “root”,
“password” : “root”,
“database” : “test2”,
}
test1_dao = BaseDao(**test1_config)
test1_struts = test1_dao.select_database_struts()
test2_dao = BaseDao(**test2_config)
test2_struts = test2_dao.select_database_struts()
main(test2_struts, test1_struts)
目前只支持了4种SQL脚本的生成。
转载自演道,想查看更及时的互联网产品技术热点文章请点击http://go2live.cn