Skip to content

Commit 9977523

Browse files
Create db.py
1 parent 2699692 commit 9977523

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

aura-core/db.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,12 @@
1+
from sqlalchemy.ext.asyncio import create_async_engine, AsyncSession
2+
from sqlalchemy.orm import sessionmaker
3+
import os
4+
5+
DATABASE_URL = os.getenv("DATABASE_URL")
6+
7+
engine = create_async_engine(DATABASE_URL, echo=True, future=True)
8+
SessionLocal = sessionmaker(engine, class_=AsyncSession, expire_on_commit=False)
9+
10+
async def get_db():
11+
async with SessionLocal() as session:
12+
yield session

0 commit comments

Comments
 (0)