filter using schema

club/api.py

@@ -1,7 +1,5 @@
-from typing import Annotated
-
-from annotated_types import MinLen
 from django.db.models import Prefetch
+from ninja import Query
 from ninja.security import SessionAuth
 from ninja_extra import ControllerBase, api_controller, paginate, route
 from ninja_extra.pagination import PageNumberPaginationExtra
@@ -10,7 +8,7 @@ from ninja_extra.schemas import PaginatedResponseSchema
 from api.auth import ApiKeyAuth
 from api.permissions import CanAccessLookup, HasPerm
 from club.models import Club, Membership
-from club.schemas import ClubSchema, SimpleClubSchema
+from club.schemas import ClubSchema, ClubSearchFilterSchema, SimpleClubSchema
 
 
 @api_controller("/club")
@@ -23,8 +21,12 @@ class ClubController(ControllerBase):
         url_name="search_club",
     )
     @paginate(PageNumberPaginationExtra, page_size=50)
-    def search_club(self, search: Annotated[str, MinLen(1)]):
-        return Club.objects.filter(name__icontains=search).values()
+    def search_club(
+        self,
+        filters: Query[ClubSearchFilterSchema],
+    ):
+        clubs = Club.objects.all()
+        return filters.filter(clubs)
 
     @route.get(
         "/{int:club_id}",

club/schemas.py

@@ -1,9 +1,24 @@
-from ninja import ModelSchema
+from typing import Optional
+
+from django.db.models import Q
+from ninja import Field, FilterSchema, ModelSchema
 
 from club.models import Club, Membership
 from core.schemas import SimpleUserSchema
 
 
+class ClubSearchFilterSchema(FilterSchema):
+    search: Optional[str] = Field(None, q="name__icontains")
+    club_id: Optional[int] = Field(None, q="id")
+    is_active: Optional[bool] = None
+    parent_id: Optional[int] = None
+    parent_name: Optional[str] = Field(None, q="parent__name__icontains")
+    exclude_ids: Optional[list[int]] = None
+
+    def filter_exclude_ids(self, value: list[int]):
+        return ~Q(id__in=value)
+
+
 class SimpleClubSchema(ModelSchema):
     class Meta:
         model = Club