File size: 1,815 Bytes
de36225 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 | import os
import asyncio
from sqlalchemy import create_engine, text
from sqlalchemy.engine import URL
import pandas as pd
from dotenv import load_dotenv
# Try to find .env in current dir and Agent dir
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 = "postgresql://postgres:postgres@localhost:5432/employee"
LOCAL_DB = os.environ.get("DB_CONNECTION_URL")
# NEON_DB = Your new Neon connection string
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...")
# 1. Connect to Local
local_engine = create_engine(LOCAL_DB)
# 2. Connect to Neon (Ensure SSL)
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:
# Read data from local
df = pd.read_sql_table(table, local_engine)
print(f" - Found {len(df)} rows.")
# Write to Neon (replace if exists)
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())
|