servo-client

"""licenses namespace (006, FR-012)."""

from __future__ import annotations

from typing import TYPE_CHECKING, Any

from servo.resources.usage import Grant

if TYPE_CHECKING:  # pragma: no cover
    from servo.client import Servo


class Licenses:
    def __init__(self, client: "Servo"):
        self._client = client

    def grant(self, model: "Any") -> Grant:
        """Grant the org commercial-use rights for a model -> a typed :class:`Grant`."""
        from servo.resources.model import Model

        slug = model.slug if isinstance(model, Model) else model
        data = self._client._post("/v1/license-grants", {"model": slug})
        return Grant(self._client, {"status": "active", "expires_at": None, **data})