File size: 281 Bytes
be1eeae
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
from pydantic import BaseModel, validator


class IndexPayload(BaseModel):
    source: str = "database"

    @validator("source")
    def valid_source(cls, v):
        if v not in ["database", "hsx"]:
            raise ValueError('source must be database or hsx')
        return v