from apollo_utils.service.base_api_client.client import BaseApiClient
from apollo_utils.service.base_api_client.config import BaseConfig

from constants.common import DELPHI_DSP_PATH, DSP, Service


class DspApiClientConfig(BaseConfig):
    service = Service.DSP

    def __init__(self):
        self._load_attrs()
        super().__init__()


class DspApiClient(DspApiClientConfig, BaseApiClient):
    def get_charts_data_health_status(self, dsp: DSP, **params):
        return self.send_request(f"api/delphi/{DELPHI_DSP_PATH[dsp]}/charts/data-health/status", **params)

    def get_charts_tracks(self, dsp: DSP, **params):
        return self.send_request(f"api/delphi/{DELPHI_DSP_PATH[dsp]}/charts/tracks", **params)