init

backend/db.py

@@ -0,0 +1,19 @@
+# backend/db.py
+import os
+from psycopg2 import pool
+from dotenv import load_dotenv
+
+load_dotenv()
+
+DATABASE_URL = os.getenv('DATABASE_URL')
+
+# Initialise a connection pool (min 1, max 10 connections)
+postgres_pool = pool.ThreadedConnectionPool(1, 10, dsn=DATABASE_URL)
+
+def get_db_connection():
+    """Return a connection from the pool. Caller must close it after use."""
+    return postgres_pool.getconn()
+
+def release_db_connection(conn):
+    """Return the connection to the pool."""
+    postgres_pool.putconn(conn)