models/ichimoku.py

from pydantic import BaseModel, root_validator


class IchimokuPayload(BaseModel):
    symbol: str
    count_back: int = 200
    symbol_type: str = "stock"
    conversion_period: int = 9
    base_period: int = 26
    span_b_period: int = 52
    displacement: int = 26

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