added database connection

database/connection.py

@@ -0,0 +1,46 @@
+import os
+import logging
+import psycopg2
+from psycopg2 import pool
+
+logger = logging.getLogger(__name__)
+
+# Connection pool for reusing connections (important for serverless!)
+_connection_pool = None
+
+def get_connection_pool():
+    global _connection_pool
+    if _connection_pool is None:
+        try:
+            DATABASE_URL = os.environ.get("DATABASE_URL")
+            _connection_pool = psycopg2.pool.SimpleConnectionPool(
+                1,  # minimum connections
+                5,  # maximum connections
+                DATABASE_URL
+            )
+            logger.info("Database connection pool created")
+        except Exception as e:
+            logger.error(f"Failed to create connection pool: {e}")
+            return None
+    return _connection_pool
+
+def get_db_connection():
+    """Get a connection from the pool"""
+    try:
+        pool = get_connection_pool()
+        if pool:
+            conn = pool.getconn()
+            return conn
+        return None
+    except Exception as e:
+        logger.error(f"Database connection error: {e}")
+        return None
+
+def release_db_connection(conn):
+    """Return connection to the pool"""
+    try:
+        pool = get_connection_pool()
+        if pool and conn:
+            pool.putconn(conn)
+    except Exception as e:
+        logger.error(f"Error releasing connection: {e}")

database/models.py

@@ -0,0 +1,26 @@
+import logging
+import database.connection as connection
+
+logger = logging.getLogger(__name__)
+
+
+class testModel:
+    def getAllUsers():
+        conn = None
+        try:
+            conn = connection.get_db_connection()
+            if not conn:
+                return None
+            
+            cursor = conn.cursor()
+            cursor.execute("SELECT * FROM students");
+            result = cursor.fetchall()
+            cursor.close()
+
+            return result if result else None
+        except Exception as e:
+            logger.error(f"Database query error: {e}")
+            return None
+        finally:
+            if conn:
+                connection.release_db_connection(conn)