industrial-model 0.1.27__py3-none-any.whl → 0.1.29__py3-none-any.whl

industrial_model/models/entities.py

@@ -1,4 +1,5 @@
 from abc import abstractmethod
+from datetime import date, datetime
 from typing import (
     Any,
     ClassVar,
@@ -10,6 +11,8 @@ from typing import (
 
 from pydantic import PrivateAttr
 
+from industrial_model.statements import Column
+
 from .base import DBModelMetaclass, RootModel
 
 
@@ -45,6 +48,7 @@ class ViewInstanceConfig(TypedDict, total=False):
     view_external_id: str | None
     instance_spaces: list[str] | None
     instance_spaces_prefix: str | None
+    view_code: str | None
 
 
 class ViewInstance(InstanceId, metaclass=DBModelMetaclass):
@@ -56,6 +60,54 @@ class ViewInstance(InstanceId, metaclass=DBModelMetaclass):
     def get_view_external_id(cls) -> str:
         return cls.view_config.get("view_external_id") or cls.__name__
 
+    def generate_model_id(
+        self,
+        fields: list[str] | list[Column] | list[Any],
+        view_code_as_prefix: bool = True,
+        separator: str = "-",
+    ) -> str:
+        if not fields:
+            raise ValueError("Fields list cannot be empty")
+        field_values = self._get_field_values(fields)
+
+        view_code = self.view_config.get("view_code")
+
+        result = separator.join(field_values)
+        return (
+            f"{view_code}{separator}{result}"
+            if view_code_as_prefix and view_code
+            else result
+        )
+
+    def _get_field_values(
+        self, fields: list[str] | list[Column] | list[Any]
+    ) -> list[str]:
+        field_values: list[str] = []
+        for field in fields:
+            if not isinstance(field, str | Column):
+                field_type = type(field).__name__
+                raise TypeError(
+                    f"Expected field to be a string or Column, got {field_type}"
+                )
+            field_name = self.get_field_name(
+                field.property if isinstance(field, Column) else field
+            )
+            if field_name is None:
+                raise ValueError(f"Field {field} not found in the model")
+            field_entry = getattr(self, field_name)
+
+            field_entry_str = ""
+            if isinstance(field_entry, str):
+                field_entry_str = field_entry
+            elif isinstance(field_entry, date | datetime):
+                field_entry_str = field_entry.isoformat()
+            elif isinstance(field_entry, InstanceId):
+                field_entry_str = field_entry.external_id
+            else:
+                field_entry_str = str(field_entry)
+            field_values.append(field_entry_str.replace(" ", ""))
+        return field_values
+
 
 class WritableViewInstance(ViewInstance):
     @abstractmethod

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/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/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/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.dist-info → industrial_model-0.1.29.dist-info}/METADATA

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

{industrial_model-0.1.27.dist-info → industrial_model-0.1.29.dist-info}/RECORD

@@ -20,13 +20,14 @@ industrial_model/engines/async_engine.py,sha256=-sQv2vn93bBmTZOxY_C1r40YP7IMl1ND
 industrial_model/engines/engine.py,sha256=lECOpjN6fGvKt28b441isT_u4UNF3LeLiJH-zwHmkYw,3798
 industrial_model/models/__init__.py,sha256=AzJ0CyPK5PvUCX45FFtybl13tkukUvk2UAF_90s_LQ8,742
 industrial_model/models/base.py,sha256=iGhDjXqA5ULEQIFHtkMi7WYJl0nQq1wi8_zqOr-Ep78,1649
-industrial_model/models/entities.py,sha256=tHOFjS-9XsaXVrZ-x0tZR1DWL2Cc8MHe82qsIazqmnM,2811
+industrial_model/models/entities.py,sha256=BC85P8THwvVmcNpYq2spJuFJtlK8Oegkmd0kx3CI_wU,4704
 industrial_model/models/schemas.py,sha256=EoLK9GYdS-0DQ98myTP3wOU9YpWIJOfDSLOYZUy3xEY,4559
-industrial_model/queries/__init__.py,sha256=0cF-KTTGHLhdFKe3T_5cEVHsTN95iAV7R6FUutvuoaU,317
-industrial_model/queries/models.py,sha256=XS-50Ja11fVM5G7XTvrFf4OSUD3QO58rAKkkFBQ3Xbg,3255
-industrial_model/queries/params.py,sha256=U2LCdmDs_jrROvJIsWdopAmZAhXrK2gDr4kuDlVK7Ug,1384
+industrial_model/queries/__init__.py,sha256=lA83zOxMRgBgkseWJgK9kCr1vD8D8iSWs9NGGRnoYKk,355
+industrial_model/queries/models.py,sha256=VK69c4L0b0miPrKvOQBB8A01SPxZXYThrquv6gw5OGY,2544
+industrial_model/queries/params.py,sha256=50qY5BO5onLsXorhcv-7qCKhJaMO94UzhKLCmZKY55s,1667
+industrial_model/queries/utils.py,sha256=uP6PLh9IVHDK6J8x444zHWPmyV4PkxdLO-PMc6qWItc,1505
 industrial_model/statements/__init__.py,sha256=rjLRo2KoazHQaOpmPkxbI3_Nm8NCkJxjpuqow6IZVSc,4221
 industrial_model/statements/expressions.py,sha256=4ZZOcZroI5-4xRw4PXIRlufi0ARndE5zSbbxLDpR2Ec,4816
-industrial_model-0.1.27.dist-info/METADATA,sha256=6Y23wxqT9bOGAos8ivWJTLH3Renj_fxrB1ZxZiR_JCQ,6858
-industrial_model-0.1.27.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-industrial_model-0.1.27.dist-info/RECORD,,
+industrial_model-0.1.29.dist-info/METADATA,sha256=vNGyeERDw-3hlRpxgxKjGERKp8VhwNPBhYRzoqWlArA,6858
+industrial_model-0.1.29.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+industrial_model-0.1.29.dist-info/RECORD,,