Avoid unwanted duplication

CentreForDigitalHumanities · Dec 18, 2024 · 497f1e6 · 497f1e6
1 parent b115871
commit 497f1e6
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/backend/event/types/EpisodeType.py b/backend/event/types/EpisodeType.py
@@ -6,13 +6,15 @@
 from event.models import Episode, EpisodeCategory
 from event.types.EpisodeCategoryType import EpisodeCategoryType
 from person.models import AgentDescription
+from space.models import SpaceDescription
 
 
 class EpisodeType(EntityDescriptionType, DjangoObjectType):
     categories = List(NonNull(EpisodeCategoryType), required=True)
     agents = List(
         NonNull("person.types.AgentDescriptionType.AgentDescriptionType"), required=True
     )
+    spaces = List(NonNull("space.types.SpaceDescriptionType.SpaceDescriptionType"), required=True)
 
     class Meta:
         model = Episode
@@ -43,6 +45,12 @@ def resolve_agents(
         # to that agent, for some reason.
         return parent.agents.distinct()
 
+    @staticmethod
+    def resolve_spaces(
+        parent: Episode, info: ResolveInfo
+    ) -> QuerySet[SpaceDescription]:
+        return parent.spaces.distinct()
+
     @staticmethod
     def resolve_categories(
         parent: Episode, info: ResolveInfo