Mutations

Author: zsl-abyss

hackernews/hackernews/schema.py

@@ -6,5 +6,7 @@
 class Query(links.schema.Query, graphene.ObjectType):
         pass
 
+class Mutation(links.schema.Mutation, graphene.ObjectType):
+    pass
 
-schema = graphene.Schema(query=Query)
+schema = graphene.Schema(query=Query, mutation=Mutation)

hackernews/links/schema.py

@@ -14,3 +14,31 @@ class Query(graphene.ObjectType):
 
     def resolve_links(self, info, **kwargs):
         return Link.objects.all()
+
+
+#1
+class CreateLink(graphene.Mutation):
+    id = graphene.Int()
+    url = graphene.String()
+    description = graphene.String()
+
+    #2
+    class Arguments:
+        url = graphene.String()
+        description = graphene.String()
+
+    #3
+    def mutate(self, info, url, description):
+        link = Link(url=url, description=description)
+        link.save()
+
+        return CreateLink(
+            id=link.id,
+            url=link.url,
+            description=link.description,
+        )
+
+
+#4
+class Mutation(graphene.ObjectType):
+    create_link = CreateLink.Field()