muffin-rest 5.1.3__tar.gz → 5.1.5__tar.gz

{muffin_rest-5.1.3 → muffin_rest-5.1.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: muffin-rest
-Version: 5.1.3
+Version: 5.1.5
 Summary: The package provides enhanced support for writing REST APIs with Muffin framework
 Home-page: https://github.com/klen/muffin-rest
 License: MIT

{muffin_rest-5.1.3 → muffin_rest-5.1.5}/muffin_rest/peewee/filters.py

@@ -2,13 +2,14 @@
 from __future__ import annotations
 
 import operator
-from typing import Any, Callable, Optional, Tuple, Type, Union, cast
-from warnings import warn
+from typing import Any, Callable, Tuple, Type, Union, cast
 
 from peewee import Field, ModelSelect
 
 from muffin_rest.filters import Filter, Filters
 
+from .utils import get_model_field_by_name
+
 
 class PWFilter(Filter):
     """Support Peewee."""
@@ -53,8 +54,6 @@ class PWFilters(Filters):
 
         handler = cast(PWRESTHandler, self.handler)
         if isinstance(obj, PWFilter):
-            if obj.field is None:
-                obj.field = get_model_field_by_name(handler, obj.name)
             return obj
 
         if isinstance(obj, Field):
@@ -71,22 +70,3 @@ class PWFilters(Filters):
         if schema_field is None and field:
             schema_field = handler.meta.Schema._declared_fields.get(field.name)
         return self.MUTATE_CLASS(name, field=field, schema_field=schema_field, **meta)
-
-
-def get_model_field_by_name(handler, name) -> Optional[Field]:
-    """Get model field by name."""
-    fields = handler.meta.model._meta.fields
-    candidate = fields.get(name)
-    if candidate:
-        return candidate
-
-    for field in fields.values():
-        if field.column_name == name:
-            return field
-
-    warn(
-        f"{handler.__qualname__} {handler.meta.model} has no field {name}",
-        category=RuntimeWarning,
-        stacklevel=5,
-    )
-    return None

{muffin_rest-5.1.3 → muffin_rest-5.1.5}/muffin_rest/peewee/sorting.py

@@ -8,6 +8,8 @@ from peewee import Field
 
 from muffin_rest.sorting import Sort, Sorting
 
+from .utils import get_model_field_by_name
+
 if TYPE_CHECKING:
     from .types import TVCollection
 
@@ -15,9 +17,7 @@ if TYPE_CHECKING:
 class PWSort(Sort):
     """Sorter for Peewee."""
 
-    async def apply(
-        self, collection: TVCollection, *, desc: bool = False, **_
-    ) -> TVCollection:
+    async def apply(self, collection: TVCollection, *, desc: bool = False, **_) -> TVCollection:
         """Sort the collection."""
         return collection.order_by_extend(self.field if not desc else self.field.desc())
 
@@ -39,14 +39,15 @@ class PWSorting(Sorting):
             return obj
 
         handler = cast(PWRESTHandler, self.handler)
-        model_meta = handler.meta.model._meta  # type: ignore[]
 
         if isinstance(obj, Field):
             name, field = obj.name, obj
 
         else:
             name = obj
-            field = meta.pop("field", model_meta.fields.get(name))
+            field = meta.pop("field", None) or name
+            if isinstance(field, str):
+                field = get_model_field_by_name(handler, field, stacklevel=6)
 
         if field:
             sort = self.MUTATE_CLASS(name, field=field, **meta)

muffin_rest-5.1.5/muffin_rest/peewee/utils.py

@@ -0,0 +1,27 @@
+"""Support filters for Peewee ORM."""
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, Optional
+from warnings import warn
+
+if TYPE_CHECKING:
+    from peewee import Field
+
+
+def get_model_field_by_name(handler, name: str, stacklevel=5) -> Optional[Field]:
+    """Get model field by name."""
+    fields = handler.meta.model._meta.fields
+    candidate = fields.get(name)
+    if candidate:
+        return candidate
+
+    for field in fields.values():
+        if field.column_name == name:
+            return field
+
+    warn(
+        f"{handler.__qualname__} {handler.meta.model} has no field {name}",
+        category=RuntimeWarning,
+        stacklevel=stacklevel,
+    )
+    return None

{muffin_rest-5.1.3 → muffin_rest-5.1.5}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "muffin-rest"
-version = "5.1.3"
+version = "5.1.5"
 description = "The package provides enhanced support for writing REST APIs with Muffin framework"
 readme = "README.rst"
 homepage = "https://github.com/klen/muffin-rest"