22 lines
509 B
Python
22 lines
509 B
Python
import os
|
|
|
|
from sqlalchemy import create_engine
|
|
from sqlalchemy.orm import sessionmaker
|
|
|
|
|
|
def get_sqlalchemy_database_uri():
|
|
username = os.getenv("POSTGRES_USER")
|
|
password = os.getenv("POSTGRES_PASSWORD")
|
|
host = os.getenv("POSTGRES_HOST")
|
|
db_name = "mev_inspect"
|
|
return f"postgresql://{username}:{password}@{host}/{db_name}"
|
|
|
|
|
|
def get_engine():
|
|
return create_engine(get_sqlalchemy_database_uri())
|
|
|
|
|
|
def get_session():
|
|
Session = sessionmaker(bind=get_engine())
|
|
return Session()
|