from typing import List

from apollo_utils.job.clients.base.client import BaseApiClient

from src.clients.base_config import ApiConfig
from src.config import config
from src.constants import Service
from src.utils.api import request_in_chunks, sum_get_users


class AtlasDataClientConfig(ApiConfig):
    _loaded_attrs = ("host",)

    service = Service.ATLAS


class AtlasDataClient(AtlasDataClientConfig, BaseApiClient):
    def set_token(self, token: str):
        self._apikey = f"Bearer {token}"
        if self._session:
            self._add_header(self._session, self._get_header())

    @request_in_chunks(config.ATLAS_CHECK_LIMIT, "user_id_list", sum_get_users)
    def get_users(self, user_id_list: List[str]) -> List[dict]:
        return self.send_request("api/v2/users/status_lookup", method="GET", id=user_id_list, get_json=True)