为了做数据对象的版本控制，需要将MySQL数据库中的表结构导出成文件进行版本化管理，试写了一下，可以完整导出数据库中的表结构信息
# -*- coding: utf-8 -*-import osimport pymysqlclass DBTool:conn = Nonecursor = Nonedef __init__(self,conn_dict):self.conn = pymysql.connect(host=conn_dict['host'],port=conn_dict['port'],user=conn_dict['user'],passwd=conn_dict['password'],db=conn_dict['db'],charset=conn_dict['charset'])self.cursor = self.conn.cursor()def execute_query(self, sql_string):try:cursor=self.cursorcursor.execute(sql_string)list = cursor.fetchall()cursor.close()self.conn.close()return listexcept pymysql.Error as e:print("mysql execute error:", e)raisedef execute_noquery(self, sql_string):try:cursor = self.cursorcursor.execute(sql_string)mit()self.cursor.close()self.conn.close()except pymysql.Error as e:print("mysql execute error:", e)raisedef main():conn_dict = {'host': '127.0.0.1', 'port': 3306, 'user': '******', 'password': '******', 'db': 'test', 'charset': 'utf8'}conn = DBTool(conn_dict)sql_gettables = "select table_name from information_schema.`TABLES` WHERE TABLE_SCHEMA = 'databas_name';"list = conn.execute_query(sql_gettables)# 文件目标路径，如果不存在，新建一个mysql_file_path = 'D:\mysqlscript'if not os.path.exists(mysql_file_path):os.mkdir(mysql_file_path)mysqldump_commad_dict = {'dumpcommad': 'mysqldump --no-data ', 'server': '127.0.0.1', 'user': '******','password': '******', 'port': 3306, 'db': 'databse_name'}if list:for row in list:print(row[0])# 切换到新建的文件夹中 os.chdir(mysql_file_path)#表名dbtable = row[0]#文件名exportfile = row[0] + '.sql'# mysqldump 命令sqlfromat = "%s -h%s -u%s -p%s -P%s %s %s >%s"# 生成相应的sql语句sql = (sqlfromat % (mysqldump_commad_dict['dumpcommad'],mysqldump_commad_dict['server'],mysqldump_commad_dict['user'],mysqldump_commad_dict['password'],mysqldump_commad_dict['port'],mysqldump_commad_dict['db'],dbtable,exportfile))print(sql)result = os.system(sql)if result:print('export ok')else:print('export fail')if __name__ == '__main__':main()
建库测试
create database test_databasecharset utf8mb4 collate utf8mb4_bin;use test_database;create table table_a(id int auto_increment not null,name varchar(100) unique,create_date datetime,primary key pk_id(id),index idx_create_date(create_date));insert into table_a(name,create_date) values ('aaaaaa',now());insert into table_a(name,create_date) values ('bbbbbb',now());create table table_b(id int auto_increment not null,name varchar(100) unique,create_date datetime,primary key pk_id(id),index idx_create_date(create_date));insert into table_b(name,create_date) values ('aaaaaa',now());insert into table_b(name,create_date) values ('bbbbbb',now());
执行的时候会提示一个警告，但是不影响最终的结果
mysqldump: [Warning] Using a password on the command line interface can be insecure.
导出建表语句会根据表的数据情况编号自增列，这是mysqldump的问题而不是导出的问题，如果有必要可以需求做相应的修改
去掉mysqldump导出表结构中备注信息
import osfilepath = "D:\\mysqlscript"# 切换到新建的文件夹中os.chdir(filepath)pathDir = os.listdir(filepath)for file in pathDir:lines = open(file, "r")content = "use ***;"content = content + "\n"for line in lines:print(line)if not (str(line).startswith("--") or str(line).startswith("/*") ):if(line!="\n" and str(line).startswith(") ENGINE")):content = content +"\n"+ ")"else:content = content + line#将提炼后的内容重新写入文件print(content)fp = open(file, 'w')fp.write(content)fp.close()
