"""API Client for ows-assets."""

from typing import Any

import requests


class AssetsAPIClient:
    """API Client for ows-assets."""

    def __init__(self, base_url: str, session: Any | None = None) -> None:
        """Initialize APIClient class."""
        self.base_url = base_url
        self.session = session

    def upload_token(
        self, headers: dict[str, Any], asset_type: str
    ) -> requests.Response:
        """Execute a POST against upload-token."""
        endpoint = "{}/upload-token".format(self.base_url)
        data = {"asset_type": asset_type}
        return requests.post(endpoint, headers=headers, json=data)