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(request: Request):
    db =  tenantdb.get_db()
    request.state.db = db 
    return db