from core.database import SessionLocal_db1, SessionLocal_db2
from sqlalchemy.orm import Session

# Dependency to get the first database session
def get_db1():
    db1 = SessionLocal_db1()
    try:
        yield db1
    finally:
        db1.close()

# Dependency to get the second database session
def get_db2():
    db2 = SessionLocal_db2()
    try:
        yield db2
    finally:
        db2.close()