File size: 1,448 Bytes
6bff5d9 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 | """JSON IR (intermediate representation) Pydantic models.
See ARCHITECTURE.md §7 for the schema.
Initial scope: single-table; filter, group_by, agg, order_by, limit.
Joins, having, offset, boolean tree filters are deferred to later versions.
"""
from typing import Any, Literal
from pydantic import BaseModel, Field
FilterOp = Literal[
"=", "!=", "<", "<=", ">", ">=",
"in", "not_in", "is_null", "is_not_null",
"like", "between",
]
AggFn = Literal["count", "count_distinct", "sum", "avg", "min", "max"]
ValueType = Literal["int", "decimal", "string", "datetime", "date", "bool"]
SortDir = Literal["asc", "desc"]
class ColumnSelect(BaseModel):
kind: Literal["column"] = "column"
column_id: str
alias: str | None = None
class AggSelect(BaseModel):
kind: Literal["agg"] = "agg"
fn: AggFn
column_id: str | None = None
alias: str | None = None
SelectItem = ColumnSelect | AggSelect
class FilterClause(BaseModel):
column_id: str
op: FilterOp
value: Any
value_type: ValueType
class OrderByClause(BaseModel):
column_id: str
dir: SortDir = "asc"
class QueryIR(BaseModel):
ir_version: str = "1.0"
source_id: str
table_id: str
select: list[SelectItem]
filters: list[FilterClause] = Field(default_factory=list)
group_by: list[str] = Field(default_factory=list)
order_by: list[OrderByClause] = Field(default_factory=list)
limit: int | None = None
|