import json
from .Service import DEFAULT_TIMEOUT_S, APIService


class TokenService(APIService):
    def __init__(self, username, api_key):
        super(TokenService, self).__init__(username, api_key)

    def generate_auth_token(self, callback=None, timeout=DEFAULT_TIMEOUT_S):
        url = self._make_url("/auth-token/generate")
        headers = dict(self._headers)
        headers["Content-Type"] = "application/json"
        data = json.dumps({"username": self._username})

        return self._make_request(
            url,
            "POST",
            headers,
            params=None,
            data=data,
            callback=callback,
            timeout=timeout,
        )