13 lines
341 B
Python
13 lines
341 B
Python
|
|
from fastapi import Depends
|
|
from app.db.session import get_tenant_db,get_user_db
|
|
from app.core import tenantCacheService
|
|
|
|
def get_db(tenant:str = "1",tenantdb = Depends(get_tenant_db)):
|
|
db_url = tenantCacheService.get_tenant_db(tenantdb,tenant)
|
|
db = get_user_db(db_url)
|
|
try:
|
|
yield db
|
|
finally:
|
|
db.close()
|