fix(#78): use get_connection helper and add error handling to seed script

Author: William-Hill

operations/seed_guid_sis_map.py

@@ -5,68 +5,65 @@
 for POC/demo purposes. Maps real Student_GUIDs to fake SIS IDs.
 """
 
-import psycopg2
-from psycopg2.extras import RealDictCursor
-from .db_config import DB_CONFIG
+from .db_utils import get_connection
 
 
 def seed_guid_sis_map():
     """Create guid_sis_map table and seed with demo data."""
-    connection = psycopg2.connect(
-        host=DB_CONFIG['host'],
-        user=DB_CONFIG['user'],
-        password=DB_CONFIG['password'],
-        dbname=DB_CONFIG['database'],
-        port=DB_CONFIG['port'],
-        cursor_factory=RealDictCursor
-    )
+    connection = get_connection()
     cursor = connection.cursor()
 
-    # Create table
-    cursor.execute("""
-        CREATE TABLE IF NOT EXISTS guid_sis_map (
-            student_guid TEXT PRIMARY KEY,
-            sis_id       TEXT NOT NULL
-        );
-    """)
-    print("✓ guid_sis_map table created/verified")
+    try:
+        # Create table
+        cursor.execute("""
+            CREATE TABLE IF NOT EXISTS guid_sis_map (
+                student_guid TEXT PRIMARY KEY,
+                sis_id       TEXT NOT NULL
+            );
+        """)
+        print("✓ guid_sis_map table created/verified")
 
-    # Pick ~20 random GUIDs from student_level_with_predictions
-    cursor.execute("""
-        SELECT "Student_GUID"
-        FROM student_level_with_predictions
-        ORDER BY RANDOM()
-        LIMIT 20
-    """)
-    guids = [row['Student_GUID'] for row in cursor.fetchall()]
+        # Pick ~20 random GUIDs from student_level_with_predictions
+        cursor.execute("""
+            SELECT "Student_GUID"
+            FROM student_level_with_predictions
+            ORDER BY RANDOM()
+            LIMIT 20
+        """)
+        guids = [row['Student_GUID'] for row in cursor.fetchall()]
 
-    if not guids:
-        print("✗ No students found in student_level_with_predictions")
-        cursor.close()
-        connection.close()
-        return False
+        if not guids:
+            print("✗ No students found in student_level_with_predictions")
+            return False
+
+        # Clear existing demo data and insert fresh mappings
+        cursor.execute("DELETE FROM guid_sis_map")
 
-    # Clear existing demo data and insert fresh mappings
-    cursor.execute("DELETE FROM guid_sis_map")
+        for i, guid in enumerate(guids, start=100001):
+            sis_id = f"BSC-{i}"
+            cursor.execute(
+                "INSERT INTO guid_sis_map (student_guid, sis_id) VALUES (%s, %s)",
+                (guid, sis_id)
+            )
 
-    for i, guid in enumerate(guids, start=100001):
-        sis_id = f"BSC-{i}"
-        cursor.execute(
-            "INSERT INTO guid_sis_map (student_guid, sis_id) VALUES (%s, %s)",
-            (guid, sis_id)
-        )
+        connection.commit()
+        print(f"✓ Seeded {len(guids)} GUID → SIS ID mappings (BSC-100001 .. BSC-{100000 + len(guids)})")
 
-    connection.commit()
-    print(f"✓ Seeded {len(guids)} GUID → SIS ID mappings (BSC-100001 .. BSC-{100000 + len(guids)})")
+        # Verify
+        cursor.execute("SELECT COUNT(*) AS count FROM guid_sis_map")
+        count = cursor.fetchone()['count']
+        print(f"✓ Verified: {count} records in guid_sis_map")
 
-    # Verify
-    cursor.execute("SELECT COUNT(*) AS count FROM guid_sis_map")
-    count = cursor.fetchone()['count']
-    print(f"✓ Verified: {count} records in guid_sis_map")
+        return True
 
-    cursor.close()
-    connection.close()
-    return True
+    except Exception as e:
+        connection.rollback()
+        print(f"✗ Failed to seed guid_sis_map: {e}")
+        return False
+
+    finally:
+        cursor.close()
+        connection.close()
 
 
 if __name__ == "__main__":