iralex-badmin/database.py

35 lines
687 B
Python

from sqlalchemy import create_engine
from sqlalchemy.ext.asyncio import AsyncSession, create_async_engine
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker
from sqlalchemy.engine import URL
connection_url = URL.create(
"postgresql",
username="iralex",
password="thestrongpassword!!!!!2",
host="ira-lex.postgres.database.azure.com",
port=5432,
database="iralex",
)
engine = create_engine(
connection_url
)
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
Base = declarative_base()
def get_db():
db = SessionLocal()
try:
yield db
finally:
db.close()