| import os |
| import asyncio |
| from sqlalchemy import create_engine, text |
| from sqlalchemy.engine import URL |
| import pandas as pd |
| from dotenv import load_dotenv |
|
|
| |
| load_dotenv() |
| load_dotenv(dotenv_path=os.path.join(os.path.dirname(__file__), '.env')) |
| load_dotenv(dotenv_path=os.path.join(os.path.dirname(__file__), '..', '.env')) |
|
|
| |
| LOCAL_DB = os.environ.get("DB_CONNECTION_URL") |
| |
| NEON_DB = os.environ.get("NEON_DB_URL") |
|
|
| async def migrate(): |
| if not NEON_DB: |
| print("❌ Error: NEON_DB_URL not found in environment variables.") |
| return |
|
|
| print("🚀 Starting Cloud Migration to Neon.tech...") |
| |
| |
| local_engine = create_engine(LOCAL_DB) |
| |
| |
| neon_uri = NEON_DB |
| if 'sslmode=' not in neon_uri: |
| sep = '&' if '?' in neon_uri else '?' |
| neon_uri += f"{sep}sslmode=require" |
| |
| neon_engine = create_engine(neon_uri) |
|
|
| tables = ["department", "employee", "dept_emp", "dept_manager", "salary", "title"] |
|
|
| for table in tables: |
| print(f"📦 Migrating table: {table}...") |
| try: |
| |
| df = pd.read_sql_table(table, local_engine) |
| print(f" - Found {len(df)} rows.") |
| |
| |
| df.to_sql(table, neon_engine, if_exists='replace', index=False) |
| print(f" ✅ {table} successfully uploaded to Neon.") |
| except Exception as e: |
| print(f" ❌ Failed to migrate {table}: {e}") |
|
|
| print("\n🎉 Migration Complete! Your cloud database is ready.") |
|
|
| if __name__ == "__main__": |
| asyncio.run(migrate()) |
|
|
