import os
import psycopg_pool
from pathlib import Path
from dotenv import load_dotenv

load_dotenv(Path(__file__).parent.parent / ".env")

DSN = os.environ["DATABASE_URL"]

pool = psycopg_pool.AsyncConnectionPool(DSN, min_size=2, max_size=10, open=False)


async def get_conn():
    async with pool.connection() as conn:
        yield conn