mysql数据库差异比较

import pymysql

def compare_databases(host1, user1, password1, db1, host2, user2, password2, db2):
    conn1 = pymysql.connect(host=host1, user=user1, password=password1, db=db1)
    conn2 = pymysql.connect(host=host2, user=user2, password=password2, db=db2)
    
    cursor1 = conn1.cursor()
    cursor2 = conn2.cursor()
    
    # 比较表结构
    cursor1.execute("SHOW TABLES")
    tables1 = set(cursor1.fetchall())
    cursor2.execute("SHOW TABLES")
    tables2 = set(cursor2.fetchall())
    
    print("Tables only in DB1:", tables1 - tables2)
    print("Tables only in DB2:", tables2 - tables1)
    
    # 比较表数据
    for table in tables1 & tables2:
        cursor1.execute(f"SELECT * FROM {table}")
        data1 = set(cursor1.fetchall())
        cursor2.execute(f"SELECT * FROM {table}")
        data2 = set(cursor2.fetchall())
        
        print(f"Differences in table {table}:")
        print("Data only in DB1:", data1 - data2)
        print("Data only in DB2:", data2 - data1)
    
    conn1.close()
    conn2.close()

# 示例调用
compare_databases('localhost', 'user1', 'password1', 'db1', 'localhost', 'user2', 'password2', 'db2')