added filters

kammer/filters.py

@@ -1,17 +1,39 @@
+from typing import Optional
+
 import strawberry
-from komment import models, scalars, enums
+import strawberry_django
+from kammer import enums, models, scalars
 from strawberry import auto
-from typing import Optional
 from strawberry_django.filters import FilterLookup
-import strawberry_django
 
 
-@strawberry_django.filter(models.Comment)
-class CommentFilter:
-    name: Optional[FilterLookup[str]] | None
+@strawberry_django.filter(models.Message)
+class MessageFilter:
     ids: list[strawberry.ID] | None
+    search: Optional[str] | None
 
     def filter_ids(self, queryset, info):
         if self.ids is None:
             return queryset
         return queryset.filter(id__in=self.ids)
+
+    def filter_search(self, queryset, info):
+        if self.search is None:
+            return queryset
+        return queryset.filter(text__icontains=self.search)
+
+
+@strawberry_django.filter(models.Room)
+class RoomFilter:
+    search: Optional[str] | None
+    ids: list[strawberry.ID] | None
+
+    def filter_ids(self, queryset, info):
+        if self.ids is None:
+            return queryset
+        return queryset.filter(id__in=self.ids)
+
+    def filter_search(self, queryset, info):
+        if self.search is None:
+            return queryset
+        return queryset.filter(title__icontains=self.search)

kammer/models.py

@@ -30,6 +30,10 @@ class Room(models.Model):
         help_text="The users that have pinned the workspace",
     )
 
+    @property
+    def messages(self):
+        return Message.objects.filter(agent__room=self).all()
+
 
 class Agent(models.Model):
     room = models.ForeignKey(Room, on_delete=models.CASCADE)

kammer/types.py

@@ -13,31 +13,28 @@
 from strawberry.experimental import pydantic
 
 
-@strawberry_django.type(models.Structure)
+@strawberry_django.type(models.Structure, pagination=True)
 class Structure:
     id: strawberry.ID
     object: strawberry.ID
     identifier: str
 
 
-@strawberry_django.type(models.Room)
+@strawberry_django.type(models.Room, pagination=True, filters=filters.RoomFilter)
 class Room:
     id: strawberry.ID
     title: str
     description: str
+    messages: list["Message"]
 
-    @strawberry_django.field()
-    def messages(self, info) -> list["Message"]:
-        return models.Message.objects.filter(agent__room=self).all()
 
-
-@strawberry_django.type(models.Agent)
+@strawberry_django.type(models.Agent, pagination=True)
 class Agent:
     id: strawberry.ID
     room: Room
 
 
-@strawberry_django.type(models.Message)
+@strawberry_django.type(models.Message, pagination=True, filters=filters.MessageFilter)
 class Message:
     id: strawberry.ID
     title: str