python脚本--mysql数据库升级、备份

在公司经常要做测试环境的升级、备份、维护；升级后台的应用，不可避免要进行数据库的升级与备份，花了一个上午琢磨了一个脚本分享给大家。

ToB的业务，在做环境维护的时候，有初始化环境和增量升级的环境，在测试阶段，几乎每轮都会动数据库脚本，那么测试环境每轮都需要进行数据库的升级维护

对应业务，基本上会出现以下三种场景：

1、初始化环境，在业务上对应新的B端客户需要安装系统

　　需要将数据库删除以后重建（必须要这么做，否则可能会出现上一轮的初始化脚本中用到的表，这一轮脚本中删除了，如果不删库重建，可能会存在冗余的表）

2、升级环境，业务上对应现有客户，需要进行系统升级，需执行增量脚本

　　不论是测试环境还是线上环境，在升级执行之前都需要将数据库进行备份，防止升级失败需要回滚

3、升级环境（非首次），业务上仅会在测试环境上出现，如第一轮测试以后，升级脚本有更新了，在第二轮测试前，需要将数据库恢复到升级之前，再进行升级

设计思路：

执行脚本，需要传入4个参数，

第一个参数是sql脚本名称，因为每轮升级时，名称都可能不一样

第二个参数是要做升级或者初始化的数据库的名称

第三个参数是升级标识，

第四个参数是数据库备份的文件版本标识，例如升级前的版本是1.0.1，升级后的版本是1.0.2，那么将1.0.1传入，这样可以标识每次备份的文件版本

# coding=utf-8
import MySQLdb
import sys,os
from  subprocess import  Popen,PIPE

db_file=sys.argv[1]
db_name = sys.argv[2]
update_flag = sys.argv[3]
#upadte_flag是升级表示，为1：表示是初始化环境，只需要删除数据库实例，并重建 2：首次进行升级，需要备份数据库    3：非首次升级，需要将数据库先进行回滚再升级
db_back_file_version = sys.argv[4]
db_back_path = '/home/DB'
db_host = 'localhost'
db_port = 3306
db_user = 'root'
db_passwd = 'password'
file_name =  db_back_path+"/"+db_name+'_'+db_back_file_version+'.sql'

if not os.path.exists(db_file):
    print('升级脚本不存在，请检查')

#如果备份文件存在，重命名
if os.path.exists(file_name):
    os.rename(file_name,file_name+'bak')

#如果是首次执行数据库升级,先备份
if update_flag == "2":
    print('开始备份数据库…………………………')
    dumpcmd = "mysqldump -u" + db_user + " -p" +db_passwd+" -h"+db_host+ " -R " + db_name + " > " + file_name
    os.system(dumpcmd)

conn = MySQLdb.connect(host=db_host,port=db_port,user=db_user,passwd=db_passwd,db=db_name)
c = conn.cursor()
process = Popen('mysql -u %s -h%s -p%s %s' %(db_user,db_host,db_passwd,db_name),stdout=PIPE, stdin=PIPE, shell=True)
if update_flag == "1":
    c.execute('Drop database %s' %db_name)
    c.execute('create database %s' %db_name)
    c.execute('use %s' %db_name)
    output = process.communicate('source '+db_file)
elif update_flag == "2":
    c.execute('use %s' %db_name)
    output = process.communicate('source '+db_file)
elif update_flag == "3":
    c.execute('Drop database %s' %db_name)
    c.execute('create database %s' %db_name)
    c.execute('use %s' %db_name)
    print('开始恢复数据库并升级数据库')
    cmd = 'source '+file_name +';'+'source '+db_back_path+'/'+db_file+';'
    output = process.communicate(cmd)

c.close()
print('run success')

有问题欢迎大家拍砖，还有很大优化的空间