Initial 2

example/cashbag_conform/filter.py

@@ -1,6 +1,16 @@
+from datetime import datetime
+
 from televend_core.databases.base_filter import BaseFilter
-from televend_core.databases.common.filters.filters import EQ, IN, filterfield
+from televend_core.databases.common.filters.filters import (
+    EQ,
+    GT,
+    IN,
+    IS_NOT_NULL,
+    LT,
+    filterfield,
+)
 from televend_core.databases.televend_repositories.cashbag_conform.model import CashBagConform
+from televend_core.databases.televend_repositories.machine.model import Machine
 
 
 class CashBagConformFilter(BaseFilter):
@@ -9,6 +19,18 @@ class CashBagConformFilter(BaseFilter):
     alive: bool | None = filterfield(operator=EQ, default=True)
     ids: list[int] | None = filterfield(field="id", operator=IN)
     machine_ids: list[int] | None = filterfield(field="machine_id", operator=IN)
-    cashflow_collections_ids: list[int] | None = filterfield(
-        field="cashflow_collections_id", operator=IN
+    company_id: int | None = filterfield(
+        field="owner_id",
+        operator=EQ,
+        joins=[Machine],
+        filter_on_model_cls=Machine,
     )
+    cashflow_collections_ids: list[int] | None = filterfield(
+        field="cashflow_collection_id", operator=IN
+    )
+    is_assigned_to_collection: bool | None = filterfield(
+        field="cashflow_collection_id", operator=IS_NOT_NULL
+    )
+    is_counted: bool | None = filterfield(field="count_timestamp", operator=IS_NOT_NULL)
+    count_timestamp_gt: datetime | None = filterfield(field="count_timestamp", operator=GT)
+    count_timestamp_lt: datetime | None = filterfield(field="count_timestamp", operator=LT)

example/cashbag_conform/load_options.py

@@ -1,6 +1,9 @@
+from sqlalchemy.orm import Query
+
 from televend_core.databases.base_load_options import LoadOptions
 from televend_core.databases.common.load_options import joinload
 from televend_core.databases.televend_repositories.cashbag_conform.model import CashBagConform
+from televend_core.databases.televend_repositories.custom_user.model import CustomUser
 
 
 class CashBagConformLoadOptions(LoadOptions):
@@ -10,3 +13,16 @@ class CashBagConformLoadOptions(LoadOptions):
     load_machine: bool = joinload(relations=["machine"])
     load_cashbag: bool = joinload(relations=["cashbag"])
     load_denominations: bool = joinload(relations=["denominations"])
+
+    load_count_user_auth_user: bool = False
+
+    def _apply_custom_joins(self, query: Query) -> Query:
+        if self.load_count_user_auth_user:
+            query = self._add_joins_to_options(
+                query=query,
+                fields=[
+                    CashBagConform.count_user,
+                    CustomUser.auth_user,
+                ],
+            )
+        return query

example/cashbag_conform/mapper.py

@@ -0,0 +1,26 @@
+    mapper_registry.map_imperatively(
+        class_=CashBagConform,
+        local_table=CASHBAG_CONFORM_TABLE,
+        properties={
+            "cashflow_collection": relationship(
+                CashFlowCollection, lazy=relationship_loading_strategy.value
+            ),
+            "cashbag": relationship(CashBag, lazy=relationship_loading_strategy.value),
+            "machine": relationship(Machine, lazy=relationship_loading_strategy.value),
+            "count_user": relationship(
+                CustomUser,
+                lazy=relationship_loading_strategy.value,
+                foreign_keys=CASHBAG_CONFORM_TABLE.columns.count_user_id,
+            ),
+            "collect_user": relationship(
+                CustomUser,
+                lazy=relationship_loading_strategy.value,
+                foreign_keys=CASHBAG_CONFORM_TABLE.columns.collect_user_id,
+            ),
+            "denominations": relationship(
+                CashBagConformDenomination,
+                back_populates="cashbag_conform",
+                lazy=relationship_loading_strategy.value,
+            ),
+        },
+    )

example/filters.py

@@ -0,0 +1,122 @@
+class FieldOperator(ABC):
+    model_cls: Type[Base]
+
+    def __init__(self, model_cls: Type[Base]) -> None:
+        self.model_cls = model_cls
+
+    @abstractmethod
+    def apply_query(self, field: str, value: Any, query: Query) -> Query: ...
+
+    def to_query(self, field: str, value: Any, query: Query) -> Query:
+        return self.apply_query(field, value, query)
+
+
+class EQ(FieldOperator):
+    """ColumnOperators.__eq__() (Python “==” operator)"""
+
+    def apply_query(self, field: str, value: Any, query: Query) -> Query:
+        return query.where(getattr(self.model_cls, field) == value)
+
+
+class NOT_EQ(FieldOperator):
+    """ColumnOperators.__ne__() (Python “!=” operator)"""
+
+    def apply_query(self, field: str, value: Any, query: Query) -> Query:
+        return query.where(getattr(self.model_cls, field) != value)
+
+
+class GT(FieldOperator):
+    """ColumnOperators.__gt__() (Python “>” operator)"""
+
+    def apply_query(self, field: str, value: Comparable, query: Query) -> Query:
+        return query.where(getattr(self.model_cls, field) > value)
+
+
+class GE(FieldOperator):
+    """ColumnOperators.__ge__() (Python “>=” operator)"""
+
+    def apply_query(self, field: str, value: Comparable, query: Query) -> Query:
+        return query.where(getattr(self.model_cls, field) >= value)
+
+
+class LT(FieldOperator):
+    """ColumnOperators.__lt__() (Python “<” operator)"""
+
+    def apply_query(self, field: str, value: Comparable, query: Query) -> Query:
+        return query.where(getattr(self.model_cls, field) < value)
+
+
+class LE(FieldOperator):
+    """ColumnOperators.__le__() (Python “<=” operator)"""
+
+    def apply_query(self, field: str, value: Comparable, query: Query) -> Query:
+        return query.where(getattr(self.model_cls, field) <= value)
+
+
+class IN(FieldOperator):
+    """IN is available most typically by passing a list of values to the ColumnOperators.in_() method"""
+
+    def apply_query(self, field: str, value: Iterable[Any], query: Query) -> Query:
+        return query.where(getattr(self.model_cls, field).in_(value))
+
+
+class NOT_IN(FieldOperator):
+    """NOT IN is available via the ColumnOperators.not_in() operator"""
+
+    def apply_query(self, field: str, value: Iterable[Any], query: Query) -> Query:
+        return query.where(~getattr(self.model_cls, field).in_(value))
+
+
+class LIKE(FieldOperator):
+    """ColumnOperators.like(). Expects string value."""
+
+    def apply_query(self, field: str, value: str, query: Query) -> Query:
+        return query.where(getattr(self.model_cls, field).like(f"%{value}%"))
+
+
+class ILIKE(FieldOperator):
+    """ColumnOperators.ilike(). Expects string value."""
+
+    def apply_query(self, field: str, value: str, query: Query) -> Query:
+        return query.where(getattr(self.model_cls, field).ilike(f"%{value}%"))
+
+
+class IS_NULL(FieldOperator):
+    """Test value for NULL / NOT NULL"""
+
+    def apply_query(self, field: str, value: bool, query: Query) -> Query:
+        if value:
+            return query.where(getattr(self.model_cls, field).is_(None))
+        else:
+            return query.where(getattr(self.model_cls, field).is_not(None))
+
+
+class IS_NOT_NULL(FieldOperator):
+    """Test value for NULL / NOT NULL"""
+
+    def apply_query(self, field: str, value: bool, query: Query) -> Query:
+        if value:
+            return query.where(getattr(self.model_cls, field).is_not(None))
+        else:
+            return query.where(getattr(self.model_cls, field).is_(None))
+
+
+class IEXACT(FieldOperator):
+    """ColumnOperators.__eq__() over strings but comparison is case-insensitive. Expects string value."""
+
+    def apply_query(self, field: str, value: str, query: Query) -> Query:
+        return query.where(func.lower(getattr(self.model_cls, field)) == value.lower())
+
+
+class REGEX(FieldOperator):
+    """Case-sensitive regex operator. Expects string regex pattern."""
+
+    def apply_query(self, field: str, value: str, query: Query) -> Query:
+        return query.where(getattr(self.model_cls, field).op("~")(value))
+
+
+class IREGEX(FieldOperator):
+    """Case-insensitive regex operator. Expects string regex pattern."""
+
+    def apply_query(self, field: str, value: str, query: Query) -> Query:
+        return query.where(getattr(self.model_cls, field).op("~*")(value))