90 lines
1.8 KiB
Python
90 lines
1.8 KiB
Python
from graphene import Schema, ObjectType, String, Int, Field, List, Mutation
|
|
|
|
class UserType(ObjectType):
|
|
id = Int()
|
|
name = String()
|
|
age = Int()
|
|
|
|
class CreateUser(Mutation):
|
|
class Arguments:
|
|
name = String()
|
|
age = Int()
|
|
|
|
user = Field(UserType)
|
|
|
|
@staticmethod
|
|
def mutate(info, name, age):
|
|
user = {"id": len(Query.users) + 1, "name": name, "age": age}
|
|
Query.users.append(user)
|
|
return CreateUser(user=user)
|
|
|
|
class Query(ObjectType):
|
|
user = Field(UserType, user_id=Int())
|
|
users_by_min_age = List(UserType, min_age=Int())
|
|
|
|
users = [
|
|
{"id": 1, "name": "Tyrel Souza", "age": 35},
|
|
{"id": 2, "name": "Lauren Feldman", "age": 38},
|
|
{"id": 3, "name": "Astrid Feldman", "age": 0},
|
|
]
|
|
|
|
@staticmethod
|
|
def resolve_user(root, info, user_id):
|
|
matched_users = [user for user in Query.users if user["id"] == user_id]
|
|
return matched_users[0] if matched_users else None
|
|
|
|
@staticmethod
|
|
def resolve_users_by_min_age(root, info, min_age):
|
|
return [user for user in Query.users if user["age"] >= min_age]
|
|
|
|
|
|
class Mutation(ObjectType):
|
|
create_user = CreateUser.Field()
|
|
|
|
class UpdateUser(Mutation):
|
|
class Arguments:
|
|
name
|
|
|
|
|
|
schema = Schema(query=Query, mutation=Mutation)
|
|
|
|
gql = """
|
|
query {
|
|
user(userId: 4) {
|
|
id
|
|
name
|
|
age
|
|
}
|
|
}
|
|
"""
|
|
|
|
gql2 = """
|
|
mutation{
|
|
createUser(name: "Jill", age: 61) {
|
|
user {
|
|
id
|
|
name
|
|
age
|
|
}
|
|
}
|
|
}
|
|
"""
|
|
gql3 = """
|
|
query {
|
|
usersByMinAge(minAge: 0) {
|
|
id
|
|
name
|
|
age
|
|
}
|
|
}
|
|
"""
|
|
|
|
|
|
if __name__ == "__main__":
|
|
res = schema.execute(gql)
|
|
print(res)
|
|
res = schema.execute(gql2)
|
|
print(res)
|
|
res = schema.execute(gql3)
|
|
print(res)
|