amenezes/isilon-client

from isilon.exceptions import TokenRetrieveException


class Credentials:
    def __init__(self, client) -> None:
        self._client = client

    async def token(self, headers: dict = {}) -> str:
        headers.update(
            {"X-Storage-User": f"{self._client.account}:{self._client.user}"}
        )
        headers.update({"X-Storage-Pass": f"{self._client.password}"})
        async with self._client.http.get(
            f"{self._client.address}/auth/v1.0", headers=headers
        ) as resp:
            try:
                return str(resp.headers["X-Auth-Token"])
            except Exception:
                raise TokenRetrieveException("Failed to obtain authentication token.")

    async def x_auth_token(self, headers: dict = {}) -> dict:
        token = await self.token(headers)
        return {"X-Auth-Token": token}

    def __repr__(self) -> str:
        return f"Credentials(account='{self._client.account}', user='{self._client.user}', password='{self._client.password}')"