mysql 容错处理

import mysql.connector
from mysql.connector import Error
import datetime

def backup_database():
    try:
        connection = mysql.connector.connect(host='localhost',
                                             database='testdb',
                                             user='root',
                                             password='password')

        if connection.is_connected():
            cursor = connection.cursor()
            backup_time = datetime.datetime.now().strftime('%Y%m%d%H%M%S')
            backup_file = f"backup_{backup_time}.sql"

            with open(backup_file, 'w') as file:
                cursor.execute("SHOW TABLES")
                tables = cursor.fetchall()
                for table in tables:
                    table = table[0]
                    cursor.execute(f"SELECT * INTO OUTFILE '{backup_file}' FROM {table}")
                    file.write(f"DROP TABLE IF EXISTS {table};\n")
                    file.write(f"CREATE TABLE {table} (\n")
                    cursor.execute(f"SHOW CREATE TABLE {table}")
                    create_table = cursor.fetchone()[1]
                    file.write(create_table + ";\n\n")

            print(f"Database backup completed successfully: {backup_file}")

    except Error as e:
        print(f"Error while connecting to MySQL: {e}")
    finally:
        if connection.is_connected():
            cursor.close()
            connection.close()
            print("MySQL connection is closed")

if __name__ == "__main__":
    backup_database()