Python 实现数据库(SQL)更新脚本的生成方法
时间:2019-03-30
本文章向大家介绍Python 实现数据库(SQL)更新脚本的生成方法,主要包括Python 实现数据库(SQL)更新脚本的生成方法使用实例、应用技巧、基本知识点总结和需要注意事项,具有一定的参考价值,需要的朋友可以参考一下。
我在工作的时候,在测试环境下使用的数据库跟生产环境的数据库不一致,当我们的测试环境下的数据库完成测试准备更新到生产环境上的数据库时候,需要准备更新脚本,真是一不小心没记下来就会忘了改了哪里,哪里添加了什么,这个真是非常让人头疼。因此我就试着用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 中。 ''' 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) os.system(sys_order) 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脚本的生成。
总结
以上所述是小编给大家介绍的Python 实现数据库(SQL)更新脚本的生成方法,希望对大家有所帮助,如果大家有任何疑问欢迎给我留言,小编会及时回复大家的!
- Facebook 发布 wav2letter 工具包,用于端到端自动语音识别
- Java企业面试——Java基础
- 从Java的类型转换看MySQL和Oracle中的隐式转换(二)(r6笔记第68天)
- R包—iGraph
- 深度学习中 GPU 和显存分析
- 数据库SQL优化大总结1之- 百万级数据库优化方案
- Golang语言社区--LollipopGO开源项目搭建商城路由分发
- Shapes and line types for R
- Python基本常用包整理(data analysis and machine learning),附查询包版本语句
- 一次 PyTorch 的踩坑经历,以及如何避免梯度成为NaN
- 递归与伪递归区别,Python 实现递归与尾递归
- N元分词算法
- 基于典型相关分析的词向量
- 分享一波关于做 Kaggle 比赛,Jdata,天池的经验,看完我这篇就够了。
- JavaScript 教程
- JavaScript 编辑工具
- JavaScript 与HTML
- JavaScript 与Java
- JavaScript 数据结构
- JavaScript 基本数据类型
- JavaScript 特殊数据类型
- JavaScript 运算符
- JavaScript typeof 运算符
- JavaScript 表达式
- JavaScript 类型转换
- JavaScript 基本语法
- JavaScript 注释
- Javascript 基本处理流程
- Javascript 选择结构
- Javascript if 语句
- Javascript if 语句的嵌套
- Javascript switch 语句
- Javascript 循环结构
- Javascript 循环结构实例
- Javascript 跳转语句
- Javascript 控制语句总结
- Javascript 函数介绍
- Javascript 函数的定义
- Javascript 函数调用
- Javascript 几种特殊的函数
- JavaScript 内置函数简介
- Javascript eval() 函数
- Javascript isFinite() 函数
- Javascript isNaN() 函数
- parseInt() 与 parseFloat()
- escape() 与 unescape()
- Javascript 字符串介绍
- Javascript length属性
- javascript 字符串函数
- Javascript 日期对象简介
- Javascript 日期对象用途
- Date 对象属性和方法
- Javascript 数组是什么
- Javascript 创建数组
- Javascript 数组赋值与取值
- Javascript 数组属性和方法
- 最长有效括号
- 统计全为1的子矩形
- Android Camera1中的人脸检测
- Executors功能如此强大,ThreadPoolExecutor功不可没(一)
- dotNET:怎样处理程序中的异常(实战篇)?
- 二维背包问题
- 小程序系列之禁用视频快进
- springBoot整合Mq报错:JmsMessagingTemplate that could not be found
- TCP:测试小工具TCPing
- 对象实例化内存布局与访问定位 Krains 2020-08-14
- 基于Docker Compose部署分布式MinIO集群
- Go语言 | 从并发模式看channel使用技巧
- 字节码文件 Krains 2020-08-04
- 为WordPress添加自定义设置上传头像功能
- 229. 求众数 II Krains 2020-08-04 16:03:03 数学