python 对比数据库,生成sql

python 复制代码
import pymysql


def get_table_structure(connection, cursor, database, table):
    # 获取表结构
    query = f"SHOW COLUMNS FROM {database}.{table}"
    cursor.execute(query)
    return cursor.fetchall()


def generate_create_table(cursor, table_structure, database, table):
    # 生成创建表的SQL语句,包含字段约束、默认值和主键信息
    columns = []
    for column in table_structure:
        column_name = column[0]
        column_type = column[1]
        is_nullable = "NOT NULL" if column[2] == "NO" else ""
        default_value = f"DEFAULT {column[4]}" if column[4] is not None else ""
        columns.append(f"`{column_name}` {column_type} {is_nullable} {default_value}")

    # 查找主键信息
    cursor.execute(f"SHOW KEYS FROM {table} WHERE Key_name = 'PRIMARY'")
    primary_key = cursor.fetchone()
    if primary_key:
        columns.append(f"PRIMARY KEY (`{primary_key[4]}`)")

    return f"CREATE TABLE `{database}`.`{table}` ({', '.join(columns)});"


def generate_add_or_modify_column(connection, cursor, database, table, column_structure):
    # 生成添加字段或修改字段的SQL语句
    alter_statements = []
    for column_info in column_structure:
        # column_name, column_type, is_nullable, default_value = column_info
        # default_clause = f"DEFAULT {default_value}" if default_value is not None else ""
        column_name = column_info[0]
        column_type = column_info[1]
        is_nullable = "NOT NULL" if column_info[2] == "NO" else ""
        default_value = f"DEFAULT  {column_info[4]}" if (column_info[4] is not None and column_info[4]) else "DEFAULT NULL"


        # 检查字段是否已经存在
        if is_column_exists(cursor, database, table, column_name):
            # 字段存在,生成修改字段的SQL语句
            alter_statements.append(
                f"ALTER TABLE `{database}`.`{table}` MODIFY COLUMN `{column_name}` {column_type} {is_nullable} {default_value} ;")
        else:
            # 字段不存在,生成添加字段的SQL语句
            alter_statements.append(
                f"ALTER TABLE `{database}`.`{table}` ADD COLUMN `{column_name}` {column_type} {is_nullable} {default_value} ;")

    return alter_statements


def is_column_exists(cursor, database, table, column_name):
    # 检查字段是否存在于目标表中
    query = f"SHOW COLUMNS FROM `{database}`.`{table}` LIKE '{column_name}'"
    cursor.execute(query)
    return cursor.fetchone() is not None


if __name__ == "__main__":

    dba ="test";
    dbb = "test1";
    # 连接数据库A
    db_a = pymysql.connect(host="127.0.0.1", port=3307, user="root", password="123456", database=dba)
    cursor_a = db_a.cursor()

    # 连接数据库B
    db_b = pymysql.connect(host="127.0.0.1", port=3307, user="root", password="123456", database=dbb)
    cursor_b = db_b.cursor()

    # 获取数据库A中的表
    cursor_a.execute("SHOW TABLES")
    database_a_tables = [table[0] for table in cursor_a.fetchall()]

    # 获取数据库B中的表
    cursor_b.execute("SHOW TABLES")
    database_b_tables = [table[0] for table in cursor_b.fetchall()]

    # 比较两个数据库的表结构并生成SQL语句
    for table in database_a_tables:
        if table not in database_b_tables:
            # 表在库B中缺失,生成创建表语句
            table_structure = get_table_structure(db_a, cursor_a,dba, table)
            create_table_sql = generate_create_table(cursor_a, table_structure, dbb, table)
            print(create_table_sql)
        else:
            # 表在库B中存在,比较字段
            columns_a = get_table_structure(db_a, cursor_a, dba, table)
            columns_b = get_table_structure(db_b, cursor_b,dbb, table)

            # 字段差异
            column_diff = [col for col in columns_a if col not in columns_b]

            # 生成添加字段或修改字段的SQL语句
            alter_column_sqls = generate_add_or_modify_column(db_b, cursor_b, dbb, table, column_diff)

            for alter_column_sql in alter_column_sqls:
                print(alter_column_sql)

    # 关闭数据库连接
    db_a.close()
    db_b.close()
相关推荐
像风一样!3 小时前
MySQL Galera Cluster部署如何实现负载均衡和高可用
数据库·mysql
last_zhiyin4 小时前
Oracle sql tuning guide 翻译 Part 6-4 --- Hint使用准则和Hint使用报告
数据库·sql·oracle·sql tunning
chenchihwen4 小时前
AI代码开发宝库系列:FAISS向量数据库
数据库·人工智能·python·faiss·1024程序员节
小光学长5 小时前
基于Vue的课程达成度分析系统t84pzgwk(程序+源码+数据库+调试部署+开发环境)带论文文档1万字以上,文末可获取,系统界面在最后面。
前端·数据库·vue.js
摇滚侠5 小时前
全面掌握PostgreSQL关系型数据库,备份和恢复,笔记46和笔记47
java·数据库·笔记·postgresql·1024程序员节
周杰伦fans6 小时前
Navicat - 连接 mysql 、 sqlserver 数据库 步骤与问题解决
数据库·mysql·sqlserver
csdn_aspnet6 小时前
如何在 Ubuntu 24.04/22.04/20.04 上安装 MySQL 8.0
linux·mysql·ubuntu
码以致用6 小时前
StarRocks笔记
数据库·starrocks·olap·1024程序员节
auspicious航7 小时前
PostgreSQL数据库关于pg_rewind的认识
数据库·postgresql·oracle
最好结果7 小时前
MyBatis 精确查询逗号分隔字符串
mysql·mybatis·1024程序员节