Python3 备份 MySQL/MariaDB（本地+FTP）

hgjgh

# encoding: utf-8　　
# author: walker
　　
# date: 2018-08-02
　　
# summary: Python3 备份 MySQL/MariaDB（本地+FTP），一个库一个文件
　　

　　
import os
　　
import time
　　
import pymysql
　　
import pprint
　　
import psutil
　　
from ftplib import FTP
　　

　　
DBHost = r'127.0.0.1'
　　
DBPort = 3306
　　
DBUser = r'root'
　　
DBPwd = r'password'
　　

　　
# LocalBakRoot = r'/home/walker/sql_bak'    # 本地备份目录
　　
LocalBakRoot = r'D:\sql_bak'    # 本地备份目录
　　
LocalIP = ''
　　

　　
FtpHost = r'192.168.xx.xx'  # FTP 主机
　　
FtpBakRoot = r'sql_bak'     # FTP 目录
　　
FtpUser = r'ftpadmin'
　　
FtpPwd = r'password'
　　

　　
# 忽略掉系统库
　　
IgnoreDB = {'information_schema', 'mysql', 'performance_schema', 'sys'}
　　

　　
# 根据前缀获取IP
　　
def GetLocalIPByPrefix(prefix):
　　
    localIP = ''
　　
    dic = psutil.net_if_addrs()
　　
    for adapter in dic:
　　
        snicList = dic[adapter]
　　
        for snic in snicList:
　　
            if not snic.family.name.startswith('AF_INET'):
　　
                continue
　　
            ip = snic.address
　　
            if ip.startswith(prefix):
　　
                localIP = ip
　　

　　
    return localIP
　　

　　
# 处理一个数据库
　　
def ProcOneDB(dbName):
　　
    print('ProcOneDB %s ...' % dbName)
　　

　　
    filename = '%s_%s_%s.sql' % (time.strftime('%Y%m%d', time.localtime()), LocalIP, dbName)
　　
    localFile = os.path.join(LocalBakRoot, filename)
　　
    #备份数据库，如果没有 -B 参数不会导出数据库创建语句
　　
    cmd = 'mysqldump -u%s -p%s -B %s > %s' % (DBUser, DBPwd, dbName, localFile)
　　
    # print('cmd: %s' % cmd)
　　
    print('mysqldump ...')
　　
    rtn = os.system(cmd)
　　
    if 0 != rtn:
　　
        print('Error: 0 != rtn')
　　
        return False
　　

　　
    ftp = FTP()
　　
    ftp.encoding = 'gb18030'
　　
    ftp.set_pasv(False)
　　
    ftp.connect(FtpHost, port=21, timeout=10)
　　
    ftp.login(user=FtpUser, passwd=FtpPwd)
　　

　　
    print(ftp.getwelcome())
　　
    ftp.cwd(FtpBakRoot)
　　
    # 以 sql 服务器 ip 作为子目录名
　　
    if LocalIP not in ftp.nlst():
　　
        ftp.mkd(LocalIP)       # 创建子目录
　　
    ftp.cwd(LocalIP)
　　
    with open(localFile, mode='rb') as f:
　　
        ftp.storbinary('STOR ' + filename, f)
　　

　　
    return True
　　

　　
# 处理所有数据库
　　
def ProcAllDB():
　　
    connDB = pymysql.connect(host=DBHost,
　　
                                port=DBPort,
　　
                                user=DBUser,
　　
                                passwd=DBPwd,
　　
                                charset='utf8mb4')
　　

　　
    cur = connDB.cursor()
　　

　　
    sql = "show databases;"
　　
    print('input sql:' + sql)
　　
    cur.execute(sql)
　　
    rowList = cur.fetchall()
　　
    cur.close()
　　
    connDB.close()
　　

　　
    dbList = list()
　　
    for row in rowList:
　　
        dbList.append(row[0])
　　
    print('dbList(%d):\n%s\n' % (len(dbList), pprint.pformat(dbList, indent=4)))
　　

　　
    for dbName in dbList:
　　
        if dbName in IgnoreDB:
　　
            continue
　　
        print('current time: %s' % time.strftime('%Y-%m-%d %H:%M:%S',time.localtime()))
　　
        ProcOneDB(dbName)
　　
        print('current time: %s\n' % time.strftime('%Y-%m-%d %H:%M:%S',time.localtime()))
　　

　　
if __name__ == '__main__':
　　
    LocalIP = GetLocalIPByPrefix('192.168.xx.')
　　
    ProcAllDB()