Initial Commit 🚀 🚀

api/db/base_client.py

@@ -0,0 +1,34 @@
+from typing import Any, Dict, List
+
+from sqlalchemy import text
+from sqlalchemy.ext.asyncio import async_sessionmaker, create_async_engine
+
+from api.constants import DATABASE_URL
+
+
+class BaseDBClient:
+    def __init__(self):
+        self.engine = create_async_engine(DATABASE_URL)
+        self.async_session = async_sessionmaker(bind=self.engine)
+
+    async def execute_raw_query(
+        self, query: str, params: Dict[str, Any] = None
+    ) -> List[Dict[str, Any]]:
+        """
+        Execute a raw SQL query and return results as a list of dictionaries.
+
+        Args:
+            query: The SQL query to execute
+            params: Optional dictionary of query parameters
+
+        Returns:
+            List of dictionaries containing the query results
+        """
+        async with self.async_session() as session:
+            result = await session.execute(text(query), params or {})
+            rows = result.fetchall()
+            if rows:
+                # Convert rows to dictionaries
+                columns = result.keys()
+                return [dict(zip(columns, row)) for row in rows]
+            return []