from database import get_db
from services.clinicDoctorsServices import ClinicDoctorsServices
from services.clinicServices import ClinicServices

class DashboardService:
    def __init__(self):
        self.db = next(get_db())
        self.clinicDoctorsServices = ClinicDoctorsServices()
        self.clinicServices = ClinicServices()
    
    def get_dashboard_counts(self, isSuperAdmin:bool):
        if isSuperAdmin:
            clinicCounts = self.clinicServices.get_clinic_count()
            return clinicCounts
        else:
            clinicDoctorsCount = self.clinicDoctorsServices.get_doctor_status_count()
            return clinicDoctorsCount