File size: 4,717 Bytes
069c35a | 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 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 | import os
import sys
import psycopg2
from dotenv import load_dotenv
def run_migration():
print("=" * 60)
print("🏥 City General Hospital — Supabase Migration Tool")
print("=" * 60)
load_dotenv()
password = os.getenv("SUPABASE_DB_PASSWORD", "").strip()
host = os.getenv("SUPABASE_DB_HOST", "").strip()
if not password:
password = input("🔑 Enter your Supabase DB password: ").strip()
if not password:
print("❌ Password cannot be empty.")
sys.exit(1)
if not host:
host = input("🌐 Enter your Supabase DB host (e.g. db.xxxx.supabase.co): ").strip()
print(f"\n⚡ Connecting to {host}...")
try:
conn = psycopg2.connect(
host=host,
port="5432",
dbname="postgres",
user="postgres",
password=password
)
conn.autocommit = True
cursor = conn.cursor()
print("✅ Connected successfully!\n")
sql = """
-- Drop tables if exist (fresh start)
DROP TABLE IF EXISTS appointments;
DROP TABLE IF EXISTS patients;
-- Create patients table
CREATE TABLE patients (
id SERIAL PRIMARY KEY,
name VARCHAR(100) NOT NULL,
age INTEGER,
gender VARCHAR(10),
disease VARCHAR(100),
doctor VARCHAR(100),
admission_date DATE,
status VARCHAR(20) DEFAULT 'admitted'
);
-- Create appointments table
CREATE TABLE appointments (
id SERIAL PRIMARY KEY,
patient_name VARCHAR(100),
doctor VARCHAR(100),
date DATE,
time VARCHAR(20),
status VARCHAR(20) DEFAULT 'scheduled'
);
-- Insert patients
INSERT INTO patients (name, age, gender, disease, doctor, admission_date, status) VALUES
('Rahul Sharma', 34, 'Male', 'Diabetes', 'Dr. Mehta', '2025-05-01', 'admitted'),
('Priya Patel', 28, 'Female', 'Hypertension', 'Dr. Singh', '2025-05-03', 'admitted'),
('Amit Verma', 45, 'Male', 'Pneumonia', 'Dr. Mehta', '2025-05-05', 'admitted'),
('Sneha Joshi', 32, 'Female', 'Appendicitis', 'Dr. Rao', '2025-05-06', 'discharged'),
('Kiran Desai', 55, 'Male', 'Heart Disease', 'Dr. Singh', '2025-05-07', 'admitted'),
('Meera Nair', 41, 'Female', 'Diabetes', 'Dr. Mehta', '2025-05-08', 'admitted'),
('Rohan Gupta', 23, 'Male', 'Fracture', 'Dr. Rao', '2025-05-09', 'admitted'),
('Anita Shah', 60, 'Female', 'Arthritis', 'Dr. Singh', '2025-05-10', 'discharged'),
('Vijay Kumar', 38, 'Male', 'Typhoid', 'Dr. Mehta', '2025-05-11', 'admitted'),
('Pooja Iyer', 29, 'Female', 'Asthma', 'Dr. Rao', '2025-05-12', 'admitted'),
('Arjun Malhotra', 47, 'Male', 'Kidney Stone', 'Dr. Singh', '2025-05-13', 'admitted'),
('Divya Reddy', 35, 'Female', 'Migraine', 'Dr. Mehta', '2025-05-14', 'admitted'),
('Suresh Pillai', 52, 'Male', 'Diabetes', 'Dr. Rao', '2025-05-15', 'admitted'),
('Lakshmi Menon', 44, 'Female', 'Hypertension', 'Dr. Singh', '2025-05-16', 'admitted'),
('Nikhil Bansal', 31, 'Male', 'Dengue', 'Dr. Mehta', '2025-05-17', 'admitted');
-- Insert appointments
INSERT INTO appointments (patient_name, doctor, date, time, status) VALUES
('Rahul Sharma', 'Dr. Mehta', '2025-05-20', '10:00 AM', 'scheduled'),
('Priya Patel', 'Dr. Singh', '2025-05-20', '11:30 AM', 'scheduled'),
('Amit Verma', 'Dr. Mehta', '2025-05-21', '09:00 AM', 'scheduled'),
('Kiran Desai', 'Dr. Singh', '2025-05-21', '02:00 PM', 'scheduled'),
('Meera Nair', 'Dr. Mehta', '2025-05-22', '10:30 AM', 'scheduled'),
('Rohan Gupta', 'Dr. Rao', '2025-05-22', '03:00 PM', 'completed'),
('Vijay Kumar', 'Dr. Mehta', '2025-05-23', '08:30 AM', 'scheduled'),
('Pooja Iyer', 'Dr. Rao', '2025-05-23', '12:00 PM', 'cancelled'),
('Arjun Malhotra', 'Dr. Singh', '2025-05-24', '11:00 AM', 'scheduled'),
('Divya Reddy', 'Dr. Mehta', '2025-05-24', '04:00 PM', 'scheduled'),
('Suresh Pillai', 'Dr. Rao', '2025-05-25', '09:30 AM', 'scheduled'),
('Lakshmi Menon', 'Dr. Singh', '2025-05-25', '01:00 PM', 'scheduled'),
('Nikhil Bansal', 'Dr. Mehta', '2025-05-26', '10:00 AM', 'scheduled');
"""
print("⚙️ Running migration...")
cursor.execute(sql)
cursor.execute("SELECT COUNT(*) FROM patients;")
p_count = cursor.fetchone()[0]
cursor.execute("SELECT COUNT(*) FROM appointments;")
a_count = cursor.fetchone()[0]
print("=" * 60)
print(f"🎉 DONE! {p_count} patients | {a_count} appointments inserted.")
print("=" * 60)
cursor.close()
conn.close()
except Exception as e:
print(f"\n❌ MIGRATION FAILED: {str(e)}")
print("\n👉 Tips:")
print("1. Check your DB password in Supabase dashboard")
print("2. Make sure your IP is allowed in Supabase network settings")
sys.exit(1)
if __name__ == "__main__":
run_migration() |