Skip to content

Feature: Add geometry filters #710

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 11 commits into
base: main
Choose a base branch
from
Open

Feature: Add geometry filters #710

Show file tree
Hide file tree
Changes from 6 commits
Commits
Show all changes
11 commits
Select commit Hold shift + click to select a range
e3ed66b
Add Geometry filter
Evan-Dabeeo Feb 16, 2025
fb3a065
Add GeometryLooupFilter
Evan-Dabeeo Feb 17, 2025
3fc6b2c
Refactor intersects parameter
Evan-Dabeeo Feb 17, 2025
2e3e35b
Undo remove field_type_map
Evan-Dabeeo Feb 17, 2025
744c260
Add geometry filters
Evan-Dabeeo Feb 18, 2025
a0bfa66
Remove unused code
Evan-Dabeeo Feb 18, 2025
17ce3f8
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Feb 18, 2025
a097404
Merge branch 'strawberry-graphql:main' into feature/GeometryFilter
shmoon-kr Apr 6, 2025
95dab27
Add tests and etc.
Evan-Dabeeo Apr 6, 2025
e68b416
Add filter to the test case
Evan-Dabeeo Apr 6, 2025
2b1db9c
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Apr 6, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 2 additions & 0 deletions poetry.lock
View file
Open in desktop

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

9 changes: 9 additions & 0 deletions strawberry_django/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -71,3 +71,12 @@
"relay",
"type",
]

from django.core.exceptions import ImproperlyConfigured

try:
from .fields.filter_types import GeometryFilterLookup
__all__.append("GeometryFilterLookup")
except ImproperlyConfigured:
# If gdal is not available, skip.
pass
40 changes: 40 additions & 0 deletions strawberry_django/fields/filter_types.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -5,15 +5,20 @@
Generic,
Optional,
TypeVar,
Annotated,
TYPE_CHECKING,
)

import strawberry
from django.core.exceptions import ImproperlyConfigured
from django.db.models import Q
from strawberry import UNSET

from strawberry_django.filters import resolve_value

from .filter_order import filter_field
if TYPE_CHECKING:
from .types import Geometry

T = TypeVar("T")

Expand Down Expand Up @@ -123,3 +128,38 @@ class DatetimeFilterLookup(DateFilterLookup[T], TimeFilterLookup[T]):
str: FilterLookup,
uuid.UUID: FilterLookup,
}


try:
from django.contrib.gis.geos import GEOSGeometry
except ImproperlyConfigured:
# If gdal is not available, skip.
pass
else:
@strawberry.input
class GeometryFilterLookup(Generic[T]):
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

issue (complexity): Consider dynamically generating the geometry fields to reduce code duplication.

Consider reducing the repetitive field definitions by generating the geometry fields dynamically. For example, you can define a list of method names and then use a helper to build your fields. This will reduce the nesting and repetitive code without changing functionality. For instance:

try:
    from django.contrib.gis.geos import GEOSGeometry
except ImproperlyConfigured:
    # If gdal is not available, skip.
    pass
else:
    GEO_FIELD_TYPE = Optional[Annotated["Geometry", strawberry.lazy(".types")]]
    geometry_methods = [
        "bbcontains", "bboverlaps", "contained", "contains", "contains_properly",
        "coveredby", "covers", "crosses", "disjoint", "equals", "exacts",
        "intersects", "overlaps", "touches", "within", "left", "right",
        "overlaps_left", "overlaps_right", "overlaps_above", "overlaps_below",
        "strictly_above", "strictly_below",
    ]

    def _build_geometry_fields():
        fields = {name: UNSET for name in geometry_methods}
        fields.update({
            "isempty": filter_field(description=f"Test whether it's empty. {_SKIP_MSG}"),
            "isvalid": filter_field(description=f"Test whether it's valid. {_SKIP_MSG}"),
        })
        return fields

    GeometryFields = _build_geometry_fields()
    GeometryFilterLookup = strawberry.input(
        type("GeometryFilterLookup", (Generic[T],), GeometryFields)
    )

This refactoring maintains all functionality while reducing the duplication and nesting in your class definition.

bbcontains: Optional[Annotated["Geometry", strawberry.lazy(".types")]] = UNSET
bboverlaps: Optional[Annotated["Geometry", strawberry.lazy(".types")]] = UNSET
contained: Optional[Annotated["Geometry", strawberry.lazy(".types")]] = UNSET
contains: Optional[Annotated["Geometry", strawberry.lazy(".types")]] = UNSET
contains_properly: Optional[Annotated["Geometry", strawberry.lazy(".types")]] = UNSET
coveredby: Optional[Annotated["Geometry", strawberry.lazy(".types")]] = UNSET
covers: Optional[Annotated["Geometry", strawberry.lazy(".types")]] = UNSET
crosses: Optional[Annotated["Geometry", strawberry.lazy(".types")]] = UNSET
disjoint: Optional[Annotated["Geometry", strawberry.lazy(".types")]] = UNSET
equals: Optional[Annotated["Geometry", strawberry.lazy(".types")]] = UNSET
exacts: Optional[Annotated["Geometry", strawberry.lazy(".types")]] = UNSET
intersects: Optional[Annotated["Geometry", strawberry.lazy(".types")]] = UNSET
isempty: Optional[bool] = filter_field(description=f"Test whether it's empty. {_SKIP_MSG}")
isvalid: Optional[bool] = filter_field(description=f"Test whether it's valid. {_SKIP_MSG}")
overlaps: Optional[Annotated["Geometry", strawberry.lazy(".types")]] = UNSET
touches: Optional[Annotated["Geometry", strawberry.lazy(".types")]] = UNSET
within: Optional[Annotated["Geometry", strawberry.lazy(".types")]] = UNSET
left: Optional[Annotated["Geometry", strawberry.lazy(".types")]] = UNSET
right: Optional[Annotated["Geometry", strawberry.lazy(".types")]] = UNSET
overlaps_left: Optional[Annotated["Geometry", strawberry.lazy(".types")]] = UNSET
overlaps_right: Optional[Annotated["Geometry", strawberry.lazy(".types")]] = UNSET
overlaps_above: Optional[Annotated["Geometry", strawberry.lazy(".types")]] = UNSET
overlaps_below: Optional[Annotated["Geometry", strawberry.lazy(".types")]] = UNSET
strictly_above: Optional[Annotated["Geometry", strawberry.lazy(".types")]] = UNSET
strictly_below: Optional[Annotated["Geometry", strawberry.lazy(".types")]] = UNSET
5 changes: 4 additions & 1 deletion strawberry_django/fields/types.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,7 @@
from strawberry.file_uploads.scalars import Upload
from strawberry.scalars import JSON
from strawberry.types.enum import EnumValueDefinition
from strawberry.types.scalar import ScalarWrapper
from strawberry.utils.str_converters import capitalize_first, to_camel_case

from strawberry_django import filters
Expand Down Expand Up @@ -348,7 +349,7 @@
Geometry = strawberry.scalar(
NewType("Geometry", geos.GEOSGeometry),
serialize=lambda v: v.tuple if isinstance(v, geos.GEOSGeometry) else v, # type: ignore
parse_value=lambda v: geos.GeometryCollection,
parse_value=lambda v: geos.GEOSGeometry(v),
description=(
"An arbitrary geographical object. One of Point, "
"LineString, LinearRing, Polygon, MultiPoint, MultiLineString, MultiPolygon."
Expand Down Expand Up @@ -557,9 +558,11 @@
):
if using_old_filters:
field_type = filters.FilterLookup[field_type]
elif type(field_type) is ScalarWrapper and field_type._scalar_definition.name in ("Point", "LineString", "LinearRing", "Polygon", "MultiPoint", "MultilineString", "MultiPolygon", "Geometry"):

Check failure on line 561 in strawberry_django/fields/types.py

View workflow job for this annotation

GitHub Actions / Typing 

Cannot access attribute "_scalar_definition" for class "type[bool]"
  Attribute "_scalar_definition" is unknown (reportAttributeAccessIssue)
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

suggestion: Consider using isinstance for type checking of ScalarWrapper.

Using 'type(field_type) is ScalarWrapper' might be too strict if subclasses of ScalarWrapper should also match. Switching to isinstance(field_type, ScalarWrapper) could provide a more flexible and robust check.

Suggested change
elif type(field_type) is ScalarWrapper and field_type._scalar_definition.name in ("Point", "LineString", "LinearRing", "Polygon", "MultiPoint", "MultilineString", "MultiPolygon", "Geometry"):
elif isinstance(field_type, ScalarWrapper) and field_type._scalar_definition.name in ("Point", "LineString", "LinearRing", "Polygon", "MultiPoint", "MultilineString", "MultiPolygon", "Geometry"):

field_type = filter_types.GeometryFilterLookup[field_type]
else:
field_type = filter_types.type_filter_map.get( # type: ignore
field_type, filter_types.FilterLookup

Check failure on line 565 in strawberry_django/fields/types.py

View workflow job for this annotation

GitHub Actions / Typing 

Argument of type "Any | type[bool] | ScalarWrapper" cannot be assigned to parameter "key" of type "type[ID] | type[bool] | type[date] | type[datetime] | type[time] | type[Decimal] | type[float] | type[int] | type[str] | type[UUID]" in function "get"
  Type "Any | type[bool] | ScalarWrapper" is not assignable to type "type[ID] | type[bool] | type[date] | type[datetime] | type[time] | type[Decimal] | type[float] | type[int] | type[str] | type[UUID]"
    Type "ScalarWrapper" is not assignable to type "type[ID] | type[bool] | type[date] | type[datetime] | type[time] | type[Decimal] | type[float] | type[int] | type[str] | type[UUID]"
      Type "ScalarWrapper" is not assignable to type "type[ID]"
      Type "ScalarWrapper" is not assignable to type "type[bool]"
      Type "ScalarWrapper" is not assignable to type "type[date]"
      Type "ScalarWrapper" is not assignable to type "type[datetime]"
      Type "ScalarWrapper" is not assignable to type "type[time]"
      Type "ScalarWrapper" is not assignable to type "type[Decimal]"
  ... (reportArgumentType)
)[field_type]

return field_type
Expand Down
Loading