migrate.py

from mysql_connector import connect_mysql, fetch_data
from mongo_connector import connect_mongo, insert_data
from schema_transform import transform_data

def migrate_data(mysql_host, mysql_user, mysql_password, mysql_db, mongo_host, mongo_port, mongo_db):
    mysql_conn = connect_mysql(mysql_host, mysql_user, mysql_password, mysql_db)
    mongo_conn = connect_mongo(mongo_host, mongo_port, mongo_db)

    result = {}
    with mysql_conn:
        with mysql_conn.cursor() as cursor:
            cursor.execute("SHOW TABLES")
            tables = [row[0] for row in cursor.fetchall()]

        for table in tables:
            columns, rows = fetch_data(mysql_conn, table)
            transformed_data = transform_data(columns, rows)
            mongo_conn[table].insert_many(transformed_data)
            result[table] = len(rows)

    return result