models/ichimoku.py

from pydantic import BaseModel, root_validator


class IchimokuPayload(BaseModel):
    symbol: str
    count_back: int = 365
    type: str = "stock"

    @root_validator
    def validate_dates(cls, values):
        if values['type'] not in ["index", "stock"]:
            raise ValueError('type should be index or stock')
        return values