added get_postgres_session()

app/sql.py

@@ -8,12 +8,14 @@ from sqlalchemy.ext.asyncio import create_async_engine, AsyncSession
 from sqlalchemy.orm import sessionmaker
 from sqlalchemy import MetaData, text
 
+from contextlib import asynccontextmanager
+
 from app.core.config import settings
 
 logger = logging.getLogger(__name__)
 
 # Database URL validation
-DATABASE_URI = settings.POSTGRES_URI
+DATABASE_URI = settings.DATABASE_URL
 if not DATABASE_URI:
     logger.error("POSTGRES_URI is empty or missing from settings")
     raise ValueError("POSTGRES_URI is not set. Check environment variables.")
@@ -305,3 +307,28 @@ try:
         return _is_postgres_connected()
 except Exception:
     pass
+
+@asynccontextmanager
+async def get_postgres_session():
+    """
+    Get PostgreSQL session context manager.
+    
+    Usage:
+        async with get_postgres_session() as session:
+            await session.execute(...)
+            await session.commit()
+    """
+    if not async_session:
+        logger.warning("PostgreSQL session maker not initialized")
+        yield None
+        return
+    
+    session = async_session()
+    try:
+        yield session
+    except Exception as e:
+        await session.rollback()
+        logger.error(f"Session error, rolled back: {e}")
+        raise
+    finally:
+        await session.close()