refactor: carrega DATABASE_URL do .env

app/database.py

@@ -1,15 +1,18 @@
-from sqlalchemy import create_engine
 from sqlalchemy.ext.asyncio import AsyncSession, create_async_engine
 from sqlalchemy.orm import sessionmaker, declarative_base
-from contextlib import contextmanager
+from contextlib import asynccontextmanager
+from dotenv import load_dotenv
+import os
 
-# database.py
-DATABASE_URL = "postgresql+asyncpg://fatura:102030@ic-postgresql-FtOY:5432/producao"
+load_dotenv()
+
+DATABASE_URL = os.getenv("DATABASE_URL")
 
 async_engine = create_async_engine(DATABASE_URL, echo=False, future=True)
 AsyncSessionLocal = sessionmaker(bind=async_engine, class_=AsyncSession, expire_on_commit=False)
 Base = declarative_base()
 
+@asynccontextmanager
 async def get_session():
     async with AsyncSessionLocal() as session:
-        yield session
+        yield session