from typing import List, Optional
from pydantic import BaseModel


class CategorisationTriggerRequest(BaseModel):
    portfolio_id: int

    scenarios_to_consider: Optional[List[int]] = None
    scenario_priority_order: Optional[List[int]] = None

    min_property_id: Optional[int] = None
    max_property_id: Optional[int] = None

    subtask_id: Optional[str] = None


# {"portfolio_id": 556, "scenarios_to_consider": [1039,1041], "scenario_priority_order": [1041,1039]}