1
+ from graphql import GraphQLError
1
2
import graphene
2
- import graphql_jwt
3
+
4
+ from graphql_jwt import ObtainJSONWebToken as ObtainToken
5
+ from graphql_jwt import Verify as VerifyToken
6
+ from graphql_jwt import Refresh as RefreshToken
3
7
from graphene_django import DjangoObjectType
4
8
from django .contrib .auth import get_user_model
5
9
from django .db .models import Q
6
- from graphql import GraphQLError
7
10
8
11
from links .models import Link as LinkModel
9
12
from links .models import Vote as VoteModel
@@ -34,7 +37,7 @@ class Arguments:
34
37
url = graphene .String ()
35
38
description = graphene .String ()
36
39
37
- def mutate (self , info , url , description ):
40
+ def mutate (parent , info , url , description ):
38
41
user = info .context .user
39
42
if user .is_anonymous :
40
43
raise GraphQLError ("Anonymous users can't create links!" )
@@ -56,7 +59,7 @@ class CreateVote(graphene.Mutation):
56
59
class Arguments :
57
60
linkId = graphene .Int ()
58
61
59
- def mutate (self , info , linkId ):
62
+ def mutate (parent , info , linkId ):
60
63
user = info .context .user
61
64
if user .is_anonymous :
62
65
raise GraphQLError ("You must be logged in to vote!" )
@@ -77,7 +80,7 @@ class Arguments:
77
80
password = graphene .String (required = True )
78
81
email = graphene .String (required = True )
79
82
80
- def mutate (self , info , username , password , email ):
83
+ def mutate (parent , info , username , password , email ):
81
84
user = get_user_model ()(
82
85
username = username ,
83
86
email = email ,
@@ -89,9 +92,9 @@ def mutate(self, info, username, password, email):
89
92
90
93
91
94
class Mutation (graphene .ObjectType ):
92
- obtainToken = graphql_jwt . ObtainJSONWebToken .Field ()
93
- verifyToken = graphql_jwt . Verify .Field ()
94
- refreshToken = graphql_jwt . Refresh .Field ()
95
+ obtainToken = ObtainToken .Field ()
96
+ verifyToken = VerifyToken .Field ()
97
+ refreshToken = RefreshToken .Field ()
95
98
createLink = CreateLink .Field ()
96
99
createVote = CreateVote .Field ()
97
100
createUser = CreateUser .Field ()
@@ -105,7 +108,7 @@ class Query(graphene.ObjectType):
105
108
users = graphene .List (User )
106
109
whoami = graphene .Field (User )
107
110
108
- def resolve_links (self , info , search = None , first = None , skip = None , ** kwargs ):
111
+ def resolve_links (parent , info , search = None , first = None , skip = None , ** kwargs ):
109
112
objects = LinkModel .objects
110
113
qs = []
111
114
if search is None :
@@ -122,13 +125,13 @@ def resolve_links(self, info, search=None, first=None, skip=None, **kwargs):
122
125
123
126
return qs
124
127
125
- def resolve_votes (self , info , ** kwargs ):
128
+ def resolve_votes (parent , info , ** kwargs ):
126
129
return VoteModel .objects .all ()
127
130
128
- def resolve_users (self , info ):
131
+ def resolve_users (parent , info ):
129
132
return get_user_model ().objects .all ()
130
133
131
- def resolve_whoami (self , info ):
134
+ def resolve_whoami (parent , info ):
132
135
user = info .context .user
133
136
if user .is_anonymous :
134
137
raise GraphQLError ("Not logged in!" )
0 commit comments