KK-72 | Add event mutation (#37)

* Add mutation API for create events/venues/occurrences

Author: quyenlq

common/models.py

@@ -2,6 +2,8 @@
 
 from django.db import models
 from django.utils.translation import ugettext_lazy as _
+from parler.managers import TranslatableQuerySet as ParlerTranslatableQuerySet
+from parler.models import TranslatableModel as ParlerTranslatableModel
 
 
 class TimestampedModel(models.Model):
@@ -19,3 +21,20 @@ class UUIDPrimaryKeyModel(models.Model):
 
     class Meta:
         abstract = True
+
+
+class TranslatableQuerySet(ParlerTranslatableQuerySet):
+    def create_translatable_object(self, **kwargs):
+        translations = kwargs.pop("translations")
+        obj = self.create(**kwargs)
+        for translation in translations:
+            language_code = translation.pop("language_code")
+            obj.create_translation(language_code=language_code, **translation)
+        return obj
+
+
+class TranslatableModel(ParlerTranslatableModel):
+    objects = TranslatableQuerySet.as_manager()
+
+    class Meta:
+        abstract = True

events/admin.py

@@ -4,6 +4,11 @@
 from .models import Event, Occurrence
 
 
+class OccurrencesInline(admin.StackedInline):
+    model = Occurrence
+    extra = 1
+
+
 @admin.register(Event)
 class EventAdmin(TranslatableAdmin):
     list_display = (
@@ -17,6 +22,9 @@ class EventAdmin(TranslatableAdmin):
     )
     list_display_links = ("id", "name")
     exclude = ("id", "created_at", "updated_at")
+    inlines = [
+        OccurrencesInline,
+    ]
 
 
 @admin.register(Occurrence)

events/models.py

@@ -1,8 +1,8 @@
 from django.db import models
 from django.utils.translation import ugettext_lazy as _
-from parler.models import TranslatableModel, TranslatedFields
+from parler.models import TranslatedFields
 
-from common.models import TimestampedModel
+from common.models import TimestampedModel, TranslatableModel
 from venues.models import Venue
 
 

events/schema.py

@@ -1,11 +1,14 @@
 import graphene
 from django.apps import apps
+from django.db import transaction
 from graphene import relay
 from graphene_django import DjangoConnectionField, DjangoObjectType
-from graphql_jwt.decorators import login_required
+from graphql_jwt.decorators import login_required, staff_member_required
+from graphql_relay import from_global_id
 
 from events.models import Event, Occurrence
-from venues.schema import VenueNode
+from kukkuu.exceptions import KukkuuGraphQLError
+from venues.models import Venue
 
 EventTranslation = apps.get_model("events", "EventTranslation")
 
@@ -34,8 +37,9 @@ def get_node(cls, info, id):
 
 
 class OccurrenceNode(DjangoObjectType):
-    venue = graphene.Field(VenueNode)
-    event = graphene.Field(EventNode)
+    venue_id = graphene.GlobalID()
+    event_id = graphene.GlobalID()
+    time = graphene.DateTime()
 
     @classmethod
     @login_required
@@ -53,9 +57,72 @@ class Meta:
         interfaces = (relay.Node,)
 
 
+class EventTranslationsInput(graphene.InputObjectType):
+    name = graphene.String(required=True)
+    short_description = graphene.String()
+    description = graphene.String()
+    language_code = graphene.String(required=True)
+
+
+class AddEventMutation(graphene.relay.ClientIDMutation):
+    class Input:
+        translations = graphene.List(EventTranslationsInput)
+        duration = graphene.Int()
+        participants_per_invite = graphene.String(required=True)
+        capacity_per_occurrence = graphene.Int(required=True)
+        published_at = graphene.DateTime()
+
+    event = graphene.Field(EventNode)
+
+    @classmethod
+    @staff_member_required
+    @transaction.atomic
+    def mutate_and_get_payload(cls, root, info, **kwargs):
+        # TODO: Add validation
+        event = Event.objects.create_translatable_object(**kwargs)
+        # TODO: Add support for image upload
+        return AddEventMutation(event=event)
+
+
+class AddOccurrenceMutation(graphene.relay.ClientIDMutation):
+    class Input:
+        time = graphene.DateTime(required=True)
+        event_id = graphene.GlobalID(required=True)
+        venue_id = graphene.GlobalID(required=True)
+
+    occurrence = graphene.Field(OccurrenceNode)
+
+    @classmethod
+    @staff_member_required
+    @transaction.atomic
+    def mutate_and_get_payload(cls, root, info, **kwargs):
+        # TODO: Validate data
+        event_id = from_global_id(kwargs["event_id"])[1]
+        try:
+            event = Event.objects.get(pk=event_id)
+            kwargs["event_id"] = event_id
+        except Event.DoesNotExist as e:
+            raise KukkuuGraphQLError(e)
+
+        venue_id = from_global_id(kwargs["venue_id"])[1]
+        try:
+            venue = Venue.objects.get(pk=venue_id)
+            kwargs["venue_id"] = venue_id
+        except Venue.DoesNotExist as e:
+            raise KukkuuGraphQLError(e)
+
+        occurrence = Occurrence.objects.create(**kwargs, event=event, venue=venue)
+        return AddOccurrenceMutation(occurrence=occurrence)
+
+
 class Query:
     events = DjangoConnectionField(EventNode)
     occurrences = DjangoConnectionField(OccurrenceNode)
 
     event = relay.Node.Field(EventNode)
     occurrence = relay.Node.Field(OccurrenceNode)
+
+
+class Mutation:
+    add_event = AddEventMutation.Field()
+    add_occurrence = AddOccurrenceMutation.Field()

events/tests/snapshots/snap_test_api.py

@@ -164,3 +164,37 @@
         }
     }
 }
+
+snapshots["test_add_event_staff_user 1"] = {
+    "data": {
+        "addEvent": {
+            "event": {
+                "capacityPerOccurrence": 30,
+                "duration": 1000,
+                "image": "",
+                "participantsPerInvite": "FAMILY",
+                "publishedAt": "1986-12-12T16:40:48+00:00",
+                "translations": [
+                    {
+                        "description": "desc",
+                        "languageCode": "FI",
+                        "name": "Event test",
+                        "shortDescription": "Short desc",
+                    }
+                ],
+            }
+        }
+    }
+}
+
+snapshots["test_add_occurrence_staff_user 1"] = {
+    "data": {
+        "addOccurrence": {
+            "occurrence": {
+                "event": {"id": "RXZlbnROb2RlOjg="},
+                "time": "1986-12-12T16:40:48+00:00",
+                "venue": {"id": "VmVudWVOb2RlOjU="},
+            }
+        }
+    }
+}

events/tests/test_api.py

@@ -1,8 +1,11 @@
+from copy import deepcopy
+
 import pytest
 from graphql_relay import to_global_id
 
 from common.tests.utils import assert_permission_denied
 from events.factories import EventFactory, OccurrenceFactory
+from venues.factories import VenueFactory
 
 
 @pytest.fixture(autouse=True)
@@ -153,6 +156,65 @@ def autouse_db(db):
 }
 """
 
+ADD_EVENT_MUTATION = """
+mutation AddEvent($input: AddEventMutationInput!) {
+  addEvent(input: $input) {
+    event {
+      translations{
+        languageCode
+        name
+        description
+        shortDescription
+      }
+      duration
+      image
+      participantsPerInvite
+      capacityPerOccurrence
+      publishedAt
+    }
+  }
+}
+"""
+
+ADD_EVENT_VARIABLES = {
+    "input": {
+        "translations": [
+            {
+                "name": "Event test",
+                "shortDescription": "Short desc",
+                "description": "desc",
+                "languageCode": "fi",
+            }
+        ],
+        "duration": 1000,
+        # "image": None,
+        "participantsPerInvite": "family",
+        "capacityPerOccurrence": 30,
+        "publishedAt": "1986-12-12T16:40:48+00:00",
+    }
+}
+
+ADD_OCCURRENCE_MUTATION = """
+mutation AddOccurrence($input: AddOccurrenceMutationInput!) {
+  addOccurrence(input: $input) {
+    occurrence{
+      event{
+        id,
+      }
+      venue {
+        id
+      }
+        time
+    }
+  }
+}
+
+"""
+
+ADD_OCCURRENCE_VARIABLES = {
+    "input": {"eventId": "", "venueId": "", "time": "1986-12-12T16:40:48+00:00"}
+}
+
 
 def test_events_query_unauthenticated(api_client):
     executed = api_client.execute(EVENTS_QUERY)
@@ -212,3 +274,44 @@ def test_occurrence_query_normal_user(snapshot, user_api_client):
     executed = user_api_client.execute(OCCURRENCE_QUERY, variables=variables)
 
     snapshot.assert_match(executed)
+
+
+def test_add_event_permission_denied(api_client, user_api_client):
+    executed = api_client.execute(ADD_EVENT_MUTATION, variables=ADD_EVENT_VARIABLES)
+    assert_permission_denied(executed)
+
+    executed = user_api_client.execute(
+        ADD_EVENT_MUTATION, variables=ADD_EVENT_VARIABLES
+    )
+    assert_permission_denied(executed)
+
+
+def test_add_event_staff_user(snapshot, staff_api_client):
+    executed = staff_api_client.execute(
+        ADD_EVENT_MUTATION, variables=ADD_EVENT_VARIABLES
+    )
+    snapshot.assert_match(executed)
+
+
+def test_add_occurrence_permission_denied(api_client, user_api_client):
+    executed = api_client.execute(
+        ADD_OCCURRENCE_MUTATION, variables=ADD_OCCURRENCE_VARIABLES
+    )
+    assert_permission_denied(executed)
+
+    executed = user_api_client.execute(
+        ADD_OCCURRENCE_MUTATION, variables=ADD_OCCURRENCE_VARIABLES
+    )
+    assert_permission_denied(executed)
+
+
+def test_add_occurrence_staff_user(snapshot, staff_api_client):
+    event = EventFactory()
+    venue = VenueFactory()
+    occurrence_variables = deepcopy(ADD_OCCURRENCE_VARIABLES)
+    occurrence_variables["input"]["eventId"] = to_global_id("EventNode", event.id)
+    occurrence_variables["input"]["venueId"] = to_global_id("VenueNode", venue.id)
+    executed = staff_api_client.execute(
+        ADD_OCCURRENCE_MUTATION, variables=occurrence_variables
+    )
+    snapshot.assert_match(executed)

kukkuu/schema.py

@@ -6,6 +6,16 @@
 import venues.schema
 
 
+class Mutation(
+    children.schema.Mutation,
+    users.schema.Mutation,
+    events.schema.Mutation,
+    venues.schema.Mutation,
+    graphene.ObjectType,
+):
+    pass
+
+
 class Query(
     children.schema.Query,
     users.schema.Query,
@@ -16,8 +26,4 @@ class Query(
     pass
 
 
-class Mutation(children.schema.Mutation, users.schema.Mutation, graphene.ObjectType):
-    pass
-
-
 schema = graphene.Schema(query=Query, mutation=Mutation)

venues/admin.py

@@ -1,6 +1,7 @@
 from django.contrib import admin
 from parler.admin import TranslatableAdmin
 
+from events.admin import OccurrencesInline
 from venues.models import Venue
 
 
@@ -9,3 +10,7 @@ class VenueAdmin(TranslatableAdmin):
     list_display = ("id", "name", "created_at", "updated_at")
     list_display_links = ("id", "name")
     exclude = ("id", "created_at", "updated_at")
+
+    inlines = [
+        OccurrencesInline,
+    ]

venues/models.py

@@ -1,8 +1,8 @@
 from django.db import models
 from django.utils.translation import ugettext_lazy as _
-from parler.models import TranslatableModel, TranslatedFields
+from parler.models import TranslatedFields
 
-from common.models import TimestampedModel
+from common.models import TimestampedModel, TranslatableModel
 
 
 class Venue(TimestampedModel, TranslatableModel):