38 lines
949 B
Python
38 lines
949 B
Python
from sqlalchemy import create_engine
|
|
from sqlalchemy.orm import sessionmaker, declarative_base, Session
|
|
from app.core import config
|
|
|
|
|
|
Base = declarative_base()
|
|
|
|
# engine = create_engine(
|
|
# config.SQLALCHEMY_DATABASE_URI,
|
|
# )
|
|
# SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
|
|
|
|
class Database:
|
|
def __init__(self, database_url: str):
|
|
self.database_url = database_url
|
|
self.engine = create_engine(self.database_url)
|
|
self.SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=self.engine)
|
|
self.Base = declarative_base()
|
|
|
|
def get_db(self):
|
|
db =self.SessionLocal()
|
|
return db
|
|
|
|
tenantdb = Database(config.SQLALCHEMY_DATABASE_URI)
|
|
|
|
|
|
def get_tenant_db():
|
|
db = tenantdb.get_db()
|
|
try:
|
|
yield db
|
|
finally:
|
|
db.close()
|
|
|
|
def get_user_db(database_url: str):
|
|
database = Database(database_url)
|
|
db = database.get_db()
|
|
return db
|