Update app.py

app.py

@@ -1,9 +1,15 @@
 import streamlit as st
 from sqlalchemy import create_engine, text
+import pyodbc
 
-# Fixed SQL Server details
+# Fixed SQL Server details (same as VBA)
 SERVER = "indsr-foundry.database.windows.net"
 DATABASE = "foundry_db_monarch"
+DRIVER = "ODBC Driver 18 for SQL Server"
+
+# VBA uses a service account initially
+SERVICE_USERNAME = "bulk_import_user"
+SERVICE_PASSWORD = "673704CB-053D-4B66-8AFA-3A18147EB2B3"
 
 st.set_page_config(page_title="SQL Server Connection Test", layout="centered")
 
@@ -13,10 +19,35 @@ st.title("SQL Server Connection Test")
 username = st.text_input("Database Username", placeholder="Enter your username")
 password = st.text_input("Database Password", type="password", placeholder="Enter your password")
 
-def test_connection(username, password):
-    """Attempts to connect to the SQL Server using SQLAlchemy + pymssql (no ODBC)."""
+
+def get_user_key():
+    """Step 1: Connect using service account and retrieve user key"""
     try:
-        conn_str = f"mssql+pymssql://{username}:{password}@{SERVER}:1433/{DATABASE}"
+        service_conn_str = (
+            f"mssql+pyodbc://{SERVICE_USERNAME}:{SERVICE_PASSWORD}@{SERVER}/{DATABASE}?"
+            f"driver={DRIVER}&Encrypt=yes&TrustServerCertificate=no"
+        )
+        engine = create_engine(service_conn_str, pool_pre_ping=True)
+
+        with engine.connect() as connection:
+            result = connection.execute(text("EXEC [entity].[PR_user_find] @people_id=:user_id"), {"user_id": username})
+            user_key = result.scalar()  # Extract the user key
+
+        return user_key
+    except Exception as e:
+        return f"❌ Failed to retrieve user key: {str(e)}"
+
+
+def test_connection(user_key, password):
+    """Step 2: Reconnect using the retrieved user key (like VBA)"""
+    try:
+        if not user_key:
+            return "⚠️ User key not found. Cannot proceed."
+
+        conn_str = (
+            f"mssql+pyodbc://{user_key}:{password}@{SERVER}/{DATABASE}?"
+            f"driver={DRIVER}&Encrypt=yes&TrustServerCertificate=no"
+        )
         engine = create_engine(conn_str, pool_pre_ping=True)
 
         with engine.connect() as connection:
@@ -27,10 +58,15 @@ def test_connection(username, password):
     except Exception as e:
         return f"❌ Connection failed: {str(e)}"
 
+
 # Button to test connection
 if st.button("Test Connection"):
     if username and password:
-        result = test_connection(username, password)
-        st.write(result)
+        user_key = get_user_key()
+        if isinstance(user_key, str) and user_key.startswith("❌"):
+            st.write(user_key)  # Display error if user key retrieval failed
+        else:
+            result = test_connection(user_key, password)
+            st.write(result)
     else:
         st.write("⚠️ Please enter both username and password.")