Source code for depot_service.models.dataset_response

from pydantic.fields import Optional, Dict, List, Field
from pydantic import BaseModel


[docs]class SchemaResponse(BaseModel):
    type: str
    avro: str
    mapping: Optional[str]


[docs]class IcebergPartitionSpecResponse(BaseModel):
    index: int
    type: str
    column: str
    name: str


[docs]class IcebergResponse(BaseModel):
    specs: Optional[List[IcebergPartitionSpecResponse]]
    properties: Dict[str, str]
    snapshotId: str
    version: str


[docs]class PulsarResponse(BaseModel):
    partitions: int
    type: str


[docs]class DatasetResponse(BaseModel):
    schema_: Optional[SchemaResponse] = Field(..., alias='schema')
    iceberg: Optional[IcebergResponse]
    pulsar: Optional[PulsarResponse]