21 lines
391 B
Python
21 lines
391 B
Python
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()
|