20 lines
532 B
Python
20 lines
532 B
Python
|
|
from fastapi import Depends,Request
|
|
from app.db.session import get_tenant_db,get_user_db
|
|
from app.core import tenantCacheService
|
|
from app.db.session import tenantdb
|
|
|
|
def get_db(request: Request,tenant:str = "1",tenantdb = Depends(get_tenant_db)):
|
|
db_url = tenantCacheService.get_tenant_db(tenantdb,tenant)
|
|
db = get_user_db(db_url)
|
|
try:
|
|
request.state.tenant = tenant
|
|
request.state.db = db
|
|
yield db
|
|
finally:
|
|
db.close()
|
|
|
|
|
|
def get_log_db():
|
|
db = tenantdb.get_db()
|
|
return db |