Relate Links and Votes

hackernews/hackernews/schema.py

@@ -12,6 +12,11 @@ class Meta:
         model = Link
 
 
+class VoteType(DjangoObjectType):
+    class Meta:
+        model = Vote
+
+
 class UserType(DjangoObjectType):
     class Meta:
         model = get_user_model()
@@ -92,12 +97,16 @@ class Mutation(graphene.ObjectType):
 
 class Query(graphene.ObjectType):
     links = graphene.List(LinkType)
+    votes = graphene.List(VoteType)
     users = graphene.List(UserType)
     whoami = graphene.Field(UserType)
 
     def resolve_links(self, info, **kwargs):
         return Link.objects.all()
 
+    def resolve_votes(self, info, **kwargs):
+        return Vote.objects.all()
+
     def resolve_users(self, info):
         return get_user_model().objects.all()
 

hackernews/queries.curl

@@ -7,3 +7,5 @@ curl -s -X POST -H "Content-Type: application/json" -H "Authorization: JWT `cat
 curl -s -X POST -H "Content-Type: application/json" -d '{"query":"query{links{id, url, description}}"}' http://localhost:8000/graphql/
 curl -s -X POST -H "Content-Type: application/json" -d '{"query":"query{users{id, email, password}}"}' http://localhost:8000/graphql/
 curl -s -X POST -H "Content-Type: application/json" -H "Authorization: JWT `cat token`" -d '{"query":"query{whoami{username email}}"}' http://localhost:8000/graphql/
+curl -s -X POST -H "Content-Type: application/json" -d '{"query":"query{votes{user{username} link{url}}}"}' http://localhost:8000/graphql/
+curl -s -X POST -H "Content-Type: application/json" -d '{"query":"query{links{url votes{user{username}}}}"}' http://localhost:8000/graphql/