Python。自动创建 MySQL 数据库的备份。-CTO智库

此脚本自动创建 MySQL 数据库的备份、恢复它们以及管理目标 MySQL 服务器上的数据库和用户创建。

导入子流程
导入日期时间
导入系统
导入操作系统

def check_and_create_database(主机、端口、用户名、密码、数据库):
    # 检查数据库是否存在的命令
    check_database_command = f"mysql -sN --host={主机} --port={端口} --user={用户名} --password={密码} -e "SELECT EXISTS(SELECT 1 FROM INFORMATION_SCHEMA.SCHEMATA WHERE SCHEMA_NAME = '{数据库}')" 2>/dev/null"

    # 执行命令
    输出 = subprocess.check_output(check_database_command, shell=True)

    # 如果输出包含b'1'，则数据库存在
    如果输出中有 b'1'：
        subprocess.run(check_database_command, shell=True, check=True)
        print(f"数据库'{database}'已经存在。")
        系统退出(1)
    别的：
        # 如果命令失败，则数据库不存在
        print(f"数据库 '{database}' 不存在。正在创建...")

        # 创建数据库的命令
        create_database_command = f"mysql --host={主机} --port={端口} --user={用户名} --password={密码} -e '创建数据库 {数据库}' 2>/dev/null"
        subprocess.run(create_database_command, shell=True)

def check_and_create_user(主机、端口、用户名、密码、数据库、新用户名、新密码):
    # 检查用户是否存在的命令
    check_user_command = f"mysql -sN --host={主机} --port={端口} --user={用户名} --password={密码} -e "SELECT EXISTS(SELECT 1 FROM mysql.user WHERE user = '{new_username}')" 2>/dev/null"

    # 执行命令
    输出 = subprocess.check_output(check_user_command, shell=True)

    # 如果输出包含b'1'，则用户存在
    如果输出中有 b'1'：
        print(f"用户'{new_username}'已经存在。")
        系统退出(1)
    别的：
        # 用户不存在，创建
        print(f"用户 '{new_username}' 不存在。正在创建...")

        # 创建用户并授予权限的命令
        create_user_command = f"mysql --host={主机} --port={端口} --user={用户名} --password={密码} -e "创建用户'{new_username}'@'%' 识别者' {新密码}';将 {database}.* 上的所有权限授予 '{new_username}'@'%';刷新权限；“2>/dev/null”
        subprocess.run(create_user_command, shell=True)

def backup_mysql_database(主机、端口、用户名、密码、数据库、备份路径):

    # 检查备份目录是否存在
    如果不是 os.path.exists(backup_path):
        print(f"错误：备份目录'{backup_path}'不存在。")
        系统退出(1)

    # 使用当前日期和时间为备份创建文件名
    时间戳 = datetime.datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
    backup_file = f"{备份路径}/{数据库}_{时间戳}.sql"

    # 使用mysqldump创建数据库备份的命令
    dump_command = f"mysqldump --no-tablespaces --host={主机} --port={端口} --user={用户名} --password={密码} {数据库} > {备份文件} 2>/dev/无效的”

    # 执行mysqldump命令
    subprocess.run(dump_command, shell=True)

    返回备份文件

def Restore_mysql_database(主机、端口、用户名、密码、数据库、备份文件):
    # 使用 mysql 从备份恢复数据库的命令
    Restore_command = f"mysql --host={主机} --port={端口} --user={用户名} --password={密码} {数据库} /dev/null"

    # 执行mysql命令
    subprocess.run(restore_command, shell=True)

def main():
    # 源MySQL数据库的连接参数
    源主机=“127.0.0.1”
    源端口=“3309”
    源用户名=“我的用户”
    源密码=“我的密码”
    源数据库=“我的数据库”

    # 目标MySQL数据库的连接参数
    目标主机=“127.0.0.1”
    目标端口=“3309”
    新用户名 = "新用户名"
    新密码 = "新密码"
    target_database =“my_database_two”

    目标用户名=“root”
    目标密码=“根密码”

    # 本地保存备份的路径
    backup_path =“my_dbs_dumps”

    # 检查source_database与target_database是否不同
    如果源数据库==目标数据库：
        print("错误：源数据库应该与目标数据库不同。")
        系统退出(1)

    # 检查目标数据库，如果不存在则创建
    check_and_create_database(目标主机、目标端口、目标用户名、目标密码、目标数据库)

    # 检查目标用户，如果不存在则创建
    check_and_create_user(目标主机、目标端口、目标用户名、目标密码、目标数据库、新用户名、新密码)

    # 创建MySQL数据库的备份
    backup_file = backup_mysql_database(源主机、源端口、源用户名、源密码、源数据库、备份路径)
    print(f"已创建数据库备份：{backup_file}")

    # 从备份中恢复目标服务器上的数据库
    Restore_mysql_database(目标主机、目标端口、目标用户名、目标密码、目标数据库、备份文件)
    print("数据库备份已在目标服务器上恢复。")

如果 __name__ == "__main__":
    主要的（）

检查并创建_数据库：
此函数检查 MySQL 服务器上是否存在数据库。如果数据库不存在，则会创建它。它需要主机、端口、用户名、密码和数据库名称等参数来检查或创建。

检查并创建_用户：
与数据库函数一样，该函数检查 MySQL 服务器上是否存在用户。如果用户不存在，它将创建用户并授予特定数据库的权限。它还接受主机、端口、用户名、密码、数据库名称、新用户名和新密码等参数。

backup_mysql_database：
此函数使用 mysqldump 执行 MySQL 数据库的备份。它需要主机、端口、用户名、密码、数据库名称和保存备份文件的路径等参数。

restore_mysql_database：
此函数从备份文件恢复 MySQL 数据库。它接受主机、端口、用户名、密码、数据库名称和备份文件路径等参数。

主要：
这是该脚本的主要功能。它为源和目标 MySQL 数据库设置参数，包括连接详细信息、数据库名称和备份路径。然后，它执行检查以确保源数据库和目标数据库不同，如果目标数据库和用户不存在，则创建目标数据库和用户，创建源数据库的备份，最后将备份恢复到目标数据库。

此外，该脚本使用 subprocess 模块执行 MySQL 操作的 shell 命令（mysql、mysqldump），并执行错误处理和输出重定向（2>/dev/null）以抑制不必要的输出。

如果您正在使用 MySQL 数据库并想要创建自动化，此代码将为您提供帮助。

这段代码是一个很好的起始模板，用于创建管理 MySQL 数据库的自动化脚本。

dmi@dmi-laptop:~/my_python$ python3 mysql_backup_restore.py 
数据库“my_database_two”不存在。创造...
用户“new_username”不存在。创造...
创建数据库备份：my_dbs_dumps/my_database_2024-05-13_20-05-24.sql
数据库备份恢复到目标服务器上。
dmi@dmi-laptop:~/my_python$

ask_dima@yahoo.com