from typing import TYPE_CHECKING

from core_domain import HumanTaskContract, HumanTaskCreateContract, HumanTaskStatus
from core_shared import JSONValue
from pydantic import BaseModel, Field

if TYPE_CHECKING:
    from app.db.models import HumanTask


class HumanTaskCreateRequest(HumanTaskCreateContract):
    pass


class HumanTaskClaimRequest(BaseModel):
    claimed_by: str


class HumanTaskCompleteRequest(BaseModel):
    status: HumanTaskStatus
    response_payload_json: dict[str, JSONValue] = Field(default_factory=dict)


class HumanTaskResponse(HumanTaskContract):
    @classmethod
    def from_entity(cls, entity: "HumanTask") -> "HumanTaskResponse":
        return cls.model_validate(entity, from_attributes=True)