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

from constants.common import APOLLO_API_CHARTS_MONITORING, Service


class ApolloApiClientConfig(BaseConfig):
    service = Service.APOLLO

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


class ApolloApiClient(ApolloApiClientConfig, BaseApiClient):
    def get_charts_monitoring(self, **params):
        return self.send_request(APOLLO_API_CHARTS_MONITORING, **params)

    def put_charts_monitoring(self, body, **params):
        return self.send_request(APOLLO_API_CHARTS_MONITORING, method="PUT", data=body, **params)