dev

domain-api/app/core/db.py

@@ -0,0 +1,20 @@
+from contextlib import contextmanager
+
+import psycopg2
+
+from app.core.config import settings
+
+
+@contextmanager
+def get_db():
+    conn = psycopg2.connect(
+        host=settings.db_host,
+        port=settings.db_port,
+        dbname=settings.db_database,
+        user=settings.db_user,
+        password=settings.db_password,
+    )
+    try:
+        yield conn
+    finally:
+        conn.close()