industrial-model 0.1.27__tar.gz → 0.1.28__tar.gz

{industrial_model-0.1.27 → industrial_model-0.1.28}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: industrial-model
-Version: 0.1.27
+Version: 0.1.28
 Summary: Industrial Model ORM
 Author-email: Lucas Alves <lucasrosaalves@gmail.com>
 Classifier: Programming Language :: Python

{industrial_model-0.1.27 → industrial_model-0.1.28}/industrial_model/queries/__init__.py

@@ -1,5 +1,5 @@
 from .models import BaseAggregationQuery, BasePaginatedQuery, BaseQuery, BaseSearchQuery
-from .params import NestedQueryParam, QueryParam, SortParam
+from .params import BoolQueryParam, NestedQueryParam, QueryParam, SortParam
 
 __all__ = [
     "BaseQuery",
@@ -9,4 +9,5 @@ __all__ = [
     "SortParam",
     "QueryParam",
     "NestedQueryParam",
+    "BoolQueryParam",
 ]

{industrial_model-0.1.27 → industrial_model-0.1.28}/industrial_model/queries/models.py

@@ -7,24 +7,17 @@ from industrial_model.statements import (
     SearchStatement,
     Statement,
     aggregate,
+    search,
+    select,
 )
 
-from .params import NestedQueryParam, QueryParam, SortParam
+from .utils import extract_base_statement_params
 
 
 class BaseQuery(RootModel):
     def to_statement(self, entity: type[TViewInstance]) -> Statement[TViewInstance]:
-        statement = Statement(entity)
-
-        for key, item in self.__class__.model_fields.items():
-            values = getattr(self, key)
-            if values is None:
-                continue
-            for metadata_item in item.metadata:
-                if isinstance(metadata_item, SortParam):
-                    statement.sort(values, metadata_item.direction)
-                elif isinstance(metadata_item, QueryParam | NestedQueryParam):
-                    statement.where(metadata_item.to_expression(values))
+        statement = select(entity)
+        _set_base_statement_params(self, statement)
 
         return statement
 
@@ -49,17 +42,8 @@ class BaseSearchQuery(RootModel):
     def to_statement(
         self, entity: type[TViewInstance]
     ) -> SearchStatement[TViewInstance]:
-        statement = SearchStatement(entity)
-
-        for key, item in self.__class__.model_fields.items():
-            values = getattr(self, key)
-            if values is None:
-                continue
-            for metadata_item in item.metadata:
-                if isinstance(metadata_item, SortParam):
-                    statement.sort(values, metadata_item.direction)
-                elif isinstance(metadata_item, QueryParam | NestedQueryParam):
-                    statement.where(metadata_item.to_expression(values))
+        statement = search(entity)
+        _set_base_statement_params(self, statement)
         if self.query:
             statement.query_by(self.query, self.query_properties)
         statement.limit(self.limit)
@@ -78,13 +62,7 @@ class BaseAggregationQuery(RootModel):
     ) -> AggregationStatement[TAggregatedViewInstance]:
         statement = aggregate(entity, self.aggregate)
 
-        for key, item in self.__class__.model_fields.items():
-            values = getattr(self, key)
-            if values is None:
-                continue
-            for metadata_item in item.metadata:
-                if isinstance(metadata_item, QueryParam | NestedQueryParam):
-                    statement.where(metadata_item.to_expression(values))
+        _set_base_statement_params(self, statement)
 
         if self.group_by_properties:
             statement.group_by(*self.group_by_properties)
@@ -96,3 +74,19 @@ class BaseAggregationQuery(RootModel):
             statement.limit(self.limit)
 
         return statement
+
+
+def _set_base_statement_params(
+    entity: RootModel,
+    statement: Statement[TViewInstance]
+    | SearchStatement[TViewInstance]
+    | AggregationStatement[TAggregatedViewInstance],
+) -> None:
+    filters_, sort_params = extract_base_statement_params(entity)
+    statement.where(*filters_)
+
+    if isinstance(statement, AggregationStatement):
+        return
+
+    for sort_value, direction in sort_params:
+        statement.sort(sort_value, direction)

{industrial_model-0.1.27 → industrial_model-0.1.28}/industrial_model/queries/params.py

@@ -2,6 +2,7 @@ from dataclasses import dataclass
 from typing import Any
 
 from industrial_model.constants import (
+    BOOL_EXPRESSION_OPERATORS,
     LEAF_EXPRESSION_OPERATORS,
     SORT_DIRECTION,
 )
@@ -49,6 +50,17 @@ class NestedQueryParam:
         )
 
 
+@dataclass
+class BoolQueryParam:
+    operator: BOOL_EXPRESSION_OPERATORS
+
+    def to_expression(self, filters: list[Expression]) -> Expression:
+        return BoolExpression(
+            operator=self.operator,
+            filters=filters,
+        )
+
+
 @dataclass
 class SortParam:
     direction: SORT_DIRECTION

industrial_model-0.1.28/industrial_model/queries/utils.py

@@ -0,0 +1,42 @@
+from typing import Any
+
+from industrial_model.constants import SORT_DIRECTION
+from industrial_model.models import RootModel
+from industrial_model.statements import Expression
+
+from .params import BoolQueryParam, NestedQueryParam, QueryParam, SortParam
+
+
+def extract_base_statement_params(
+    entity: RootModel,
+) -> tuple[list[Expression], list[tuple[Any, SORT_DIRECTION]]]:
+    filters_: list[Expression] = []
+    sort_params: list[tuple[Any, SORT_DIRECTION]] = []
+
+    for key, item in entity.__class__.model_fields.items():
+        values = getattr(entity, key)
+        if values is None:
+            continue
+        for metadata_item in item.metadata:
+            if isinstance(metadata_item, SortParam):
+                sort_params.append((values, metadata_item.direction))
+            elif isinstance(
+                metadata_item, QueryParam | NestedQueryParam | BoolQueryParam
+            ):
+                filter_ = _extract_expression(metadata_item, values)
+                if filter_ is not None:
+                    filters_.append(filter_)
+    return filters_, sort_params
+
+
+def _extract_expression(
+    metadata_item: QueryParam | NestedQueryParam | BoolQueryParam, values: Any
+) -> Expression | None:
+    if isinstance(metadata_item, QueryParam | NestedQueryParam):
+        return metadata_item.to_expression(values)
+
+    if not isinstance(values, RootModel):
+        return None
+
+    filters_, _ = extract_base_statement_params(values)
+    return metadata_item.to_expression(filters_)

{industrial_model-0.1.27 → industrial_model-0.1.28}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "industrial-model"
-version = "0.1.27"
+version = "0.1.28"
 description = "Industrial Model ORM"
 readme = "README.md"
 requires-python = ">=3.11"