feat(mbql): add Query to compile MBQL

Signed-off-by: Charles Larivier <charles@dribbble.com>

Author: Charles Larivier

src/metabase/mbql/__init__.py

@@ -0,0 +1 @@
+from metabase.mbql.query import Query

src/metabase/mbql/aggregations.py

@@ -0,0 +1,32 @@
+from metabase.mbql.base import Mbql
+
+
+class Aggregation(Mbql):
+    mbql: str
+
+    def compile(self):
+        return [self.mbql]
+
+
+class ColumnAggregation(Aggregation):
+    def __init__(self, field_id: int):
+        self.field_id = field_id
+
+    def compile(self):
+        return [self.mbql, ["field", self.field_id, None]]
+
+
+class Count(Aggregation):
+    mbql = "count"
+
+
+class Sum(ColumnAggregation):
+    mbql = "sum"
+
+
+class Max(ColumnAggregation):
+    mbql = "max"
+
+
+class Min(ColumnAggregation):
+    mbql = "min"

src/metabase/mbql/base.py

@@ -0,0 +1,18 @@
+from typing import List
+
+
+class Mbql:
+    def compile(self):
+        raise NotImplementedError()
+
+    def __repr__(self):
+        return str(self.compile())
+
+
+class Field(Mbql):
+    def __init__(self, id: int, option=None):
+        self.id = id
+        self.option = option
+
+    def compile(self) -> List:
+        return ["field", self.id, self.option]

src/metabase/mbql/groupby.py

@@ -0,0 +1,44 @@
+from enum import Enum
+
+from metabase.mbql.base import Field
+
+
+class TemporalOption(Enum):
+    MINUTE = {"temporal-unit": "minute"}
+    HOUR = {"temporal-unit": "hour"}
+    DAY = {"temporal-unit": "day"}
+    WEEK = {"temporal-unit": "week"}
+    MONTH = {"temporal-unit": "month"}
+    QUARTER = {"temporal-unit": "quarter"}
+    YEAR = {"temporal-unit": "year"}
+    MINUTE_OF_HOUR = {"temporal-unit": "minute-of-hour"}
+    HOUR_OF_DAY = {"temporal-unit": "hour-of-day"}
+    DAY_OF_WEEK = {"temporal-unit": "day-of-week"}
+    DAY_OF_MONTH = {"temporal-unit": "day-of-month"}
+    DAY_OF_YEAR = {"temporal-unit": "day-of-year"}
+    WEEK_OF_YEAR = {"temporal-unit": "week-of-year"}
+    MONTH_OF_YEAR = {"temporal-unit": "month-of-year"}
+    QUARTER_OF_YEAR = {"temporal-unit": "quarter-of-year"}
+
+
+class BinOption(Enum):
+    AUTO = {"binning": {"strategy": "default"}}
+    BINS_10 = {"binning": {"strategy": "num-bins", "num-bins": 10}}
+    BINS_50 = {"binning": {"strategy": "num-bins", "num-bins": 50}}
+    BINS_100 = {"binning": {"strategy": "num-bins", "num-bins": 100}}
+    NONE = None
+
+
+class GroupBy(Field):
+    def __init__(self, field_id: int, option=None):
+        super(GroupBy, self).__init__(id=field_id, option=option)
+
+
+class TemporalGroupBy(GroupBy):
+    def __init__(self, field_id: int, option: TemporalOption):
+        super(TemporalGroupBy, self).__init__(field_id=field_id, option=option.value)
+
+
+class BinnedGroupBy(GroupBy):
+    def __init__(self, field_id: int, option: BinOption):
+        super(BinnedGroupBy, self).__init__(field_id=field_id, option=option.value)

src/metabase/mbql/query.py

@@ -0,0 +1,21 @@
+from dataclasses import dataclass, field
+from typing import List, Union
+
+from metabase.mbql.aggregations import Aggregation
+from metabase.mbql.groupby import GroupBy
+
+
+@dataclass
+class Query:
+    table_id: int
+    aggregations: List[Union[Aggregation, Metric]]
+    group_by: List[GroupBy] = field(default_factory=list)
+    filters: List[Filter] = field(default_factory=list)
+
+    def compile(self):
+        return {
+            "source-table": self.table_id,
+            "aggregation": [agg.compile() for agg in self.aggregations],
+            "breakout": [],
+            "filter": [],
+        }