rb-commons 0.7.14__py3-none-any.whl → 0.7.15__py3-none-any.whl

rb_commons/orm/managers.py

@@ -43,6 +43,16 @@ def query_mutator(func: F) -> F:
     return wrapper
 
 
+AGG_MAP: dict[str, Callable[[Any], Any]] = {
+    "sum": func.sum,
+    "avg": func.avg,
+    "mean": func.avg,
+    "min": func.min,
+    "max": func.max,
+    "count": func.count,
+    "first": lambda c: c,
+}
+
 class BaseManager(Generic[ModelType]):
     model: Type[ModelType]
 
@@ -525,27 +535,116 @@ class BaseManager(Generic[ModelType]):
         self._filtered = True
         return self
 
-    @query_mutator
-    def sort_by(self, tokens: Sequence[str]) -> "BaseManager[ModelType]":
+    def _infer_default_agg(self, column) -> str:
+        try:
+            from sqlalchemy import Integer, BigInteger, SmallInteger, Float, Numeric
+            if hasattr(column, "type") and isinstance(column.type, (Integer, BigInteger, SmallInteger, Float, Numeric)):
+                return "sum"
+        except Exception:
+            pass
+        return "max"
+    def _order_expr_for_path(self, token: str):
         """
-        Dynamically apply ORDER BY clauses based on a list of "field" or "-field" tokens.
-        Unknown fields are collected for Python-side sorting later.
+        token grammar:
+          [-]<path>[:<agg>][!first|!last]
+            <path> := "field" | "relation__field" (one hop)
+            <agg>  := sum|avg|min|max|count|first  (required for uselist=True; optional otherwise)
+        Examples:
+          "category__title"
+          "-reviews__rating:avg!last"
+          "stocks__sold:sum"
         """
-        self._invalid_sort_tokens = []
-        self._order_by = []
-        model = self.model
 
-        for tok in tokens:
-            if not tok:
-                continue
-            direction = desc if tok.startswith("-") else asc
-            name = tok.lstrip("-")
-            col = getattr(model, name, None)
+        # strip leading '-' (handled by caller), and parse nulls placement
+        core = token.lstrip("-")
+        nulls_placement = None
+        if core.endswith("!first"):
+            core, nulls_placement = core[:-6], "first"
+        elif core.endswith("!last"):
+            core, nulls_placement = core[:-5], "last"
+
+        # split aggregate suffix if present
+        if ":" in core:
+            path, agg_name = core.split(":", 1)
+            agg_name = agg_name.lower().strip()
+        else:
+            path, agg_name = core, None
+
+        # base column on the model (no relation hop)
+        if "__" not in path and "." not in path:
+            col = getattr(self.model, path, None)
             if col is None:
-                self._invalid_sort_tokens.append(tok)
-                continue
-            self._order_by.append(direction(col))
+                raise ValueError(f"Invalid order_by field '{path}' for {self.model.__name__}")
+            expr = col
+            if nulls_placement == "first":
+                expr = expr.nullsfirst()
+            elif nulls_placement == "last":
+                expr = expr.nullslast()
+            return expr
+
+        # relation hop (exactly one)
+        parts = re.split(r"\.|\_\_", path)
+        if len(parts) != 2:
+            raise ValueError(f"Only one relation hop supported in order_by: {path!r}")
+
+        rel_name, col_name = parts
+        rel_attr = getattr(self.model, rel_name, None)
+        if rel_attr is None or not hasattr(rel_attr, "property"):
+            raise ValueError(f"Invalid relationship '{rel_name}' on {self.model.__name__}")
+
+        target_mapper = rel_attr.property.mapper
+        target_cls = target_mapper.class_
+        target_col = getattr(target_cls, col_name, None)
+        if target_col is None:
+            raise ValueError(f"Invalid column '{col_name}' on related model {target_cls.__name__}")
+
+        primaryjoin = rel_attr.property.primaryjoin
+        uselist = rel_attr.property.uselist
+
+        # One-to-many (or many-to-many via association): require aggregate (or infer)
+        if uselist:
+            agg_name = agg_name or self._infer_default_agg(target_col)
+            agg_fn = AGG_MAP.get(agg_name)
+            if agg_fn is None:
+                raise ValueError(f"Unsupported aggregate '{agg_name}' in order_by for {path!r}")
+
+            # SELECT agg(related.col) WHERE primaryjoin  (correlated)
+            subq = (
+                select(agg_fn(target_col))
+                .where(primaryjoin)
+                .correlate(self.model)  # tie to outer row
+                .scalar_subquery()
+            )
+            expr = subq
 
+        else:
+            if agg_name and agg_name != "first":
+                agg_fn = AGG_MAP.get(agg_name)
+                if agg_fn is None:
+                    raise ValueError(f"Unsupported aggregate '{agg_name}' in order_by for {path!r}")
+                select_expr = agg_fn(target_col)
+            else:
+                select_expr = target_col
+
+            sub = select(select_expr).where(primaryjoin).correlate(self.model)
+            if agg_name == "first":
+                sub = sub.limit(1)
+            expr = sub.scalar_subquery()
+
+        if nulls_placement == "first":
+            expr = expr.nullsfirst()
+        elif nulls_placement == "last":
+            expr = expr.nullslast()
+
+        return expr
+
+    @query_mutator
+    def sort_by(self, tokens):
+        self._order_by = []
+        for tok in tokens or []:
+            direction = desc if tok.startswith("-") else asc
+            name = tok.lstrip("-")
+            self._order_by.append(direction(self._order_expr_for_path(name)))
         return self
 
     def model_to_dict(self, instance: ModelType, exclude: set[str] = None) -> dict:

{rb_commons-0.7.14.dist-info → rb_commons-0.7.15.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: rb-commons
-Version: 0.7.14
+Version: 0.7.15
 Summary: Commons of project and simplified orm based on sqlalchemy.
 Home-page: https://github.com/RoboSell-organization/rb-commons
 Author: Abdulvoris

{rb_commons-0.7.14.dist-info → rb_commons-0.7.15.dist-info}/RECORD

@@ -14,7 +14,7 @@ rb_commons/http/exceptions.py,sha256=EGRMr1cRgiJ9Q2tkfANbf0c6-zzXf1CD6J3cmCaT_FA
 rb_commons/orm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 rb_commons/orm/enum.py,sha256=PRNSuP1X7ADJW1snnxRGcHj9mxE3JaO-a2AyaHmB6es,380
 rb_commons/orm/exceptions.py,sha256=1aMctiEwrPjyehoXVX1l6ML5ZOhmDkmBISzlTD5ey1Y,509
-rb_commons/orm/managers.py,sha256=BrpMvOkRoAPKtUV1SfakXMYd73-RhqIaoj1Wmru1gAs,20319
+rb_commons/orm/managers.py,sha256=wo6fq85kxGAzzd3DQfveg-yyCLtDuFOibq6isFNbFc4,24161
 rb_commons/orm/querysets.py,sha256=Q4iY_TKZItpNnrbCAvlSwyCxJikhaX5qVF3Y2rVgAYQ,1666
 rb_commons/orm/services.py,sha256=71eRcJ4TxZvzNz-hLXo12X4U7PGK54ZfbLAb27AjZi8,1589
 rb_commons/permissions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -24,7 +24,7 @@ rb_commons/schemes/jwt.py,sha256=ZKLJ5D3fcEmEKySjzbxEgUcza4K-oPoHr14_Z0r9Yic,249
 rb_commons/schemes/pagination.py,sha256=8VZW1wZGJIPR9jEBUgppZUoB4uqP8ORudHkMwvEJSxg,1866
 rb_commons/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 rb_commons/utils/media.py,sha256=pwwGyImI5BK-NCJkX0Q6w2Nm-QL9_CCQC7B7O7wz38I,1493
-rb_commons-0.7.14.dist-info/METADATA,sha256=e_AOmEqnBAOeEs8njJyzP-ge1IJTFW5_JrHmZHTUPjA,6571
-rb_commons-0.7.14.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-rb_commons-0.7.14.dist-info/top_level.txt,sha256=HPx_WAYo3_fbg1WCeGHsz3wPGio1ucbnrlm2lmqlJog,11
-rb_commons-0.7.14.dist-info/RECORD,,
+rb_commons-0.7.15.dist-info/METADATA,sha256=sKv6SUepDqAr5CzE7OC4UD6D6VFjH-GlwoiAQzJm95o,6571
+rb_commons-0.7.15.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+rb_commons-0.7.15.dist-info/top_level.txt,sha256=HPx_WAYo3_fbg1WCeGHsz3wPGio1ucbnrlm2lmqlJog,11
+rb_commons-0.7.15.dist-info/RECORD,,