70 lines
2.1 KiB
Python
70 lines
2.1 KiB
Python
from api.src.queries.cities import CityFilter, CitySort, CityGraphType, City
|
|
from api.src.queries.user import User, UserFilter, UserSort, UserGraphType
|
|
from cpl.api.middleware.request import get_request
|
|
from cpl.auth.schema import AuthUserDao, AuthUser
|
|
from cpl.graphql.schema.filter.filter import Filter
|
|
from cpl.graphql.schema.graph_type import GraphType
|
|
from cpl.graphql.schema.query import Query
|
|
from cpl.graphql.schema.sort.sort import Sort
|
|
from cpl.graphql.schema.sort.sort_order import SortOrder
|
|
|
|
users = [User(i, f"User {i}") for i in range(1, 101)]
|
|
cities = [City(i, f"City {i}") for i in range(1, 101)]
|
|
|
|
# class AuthUserFilter(Filter[AuthUser]):
|
|
# def __init__(self):
|
|
# Filter.__init__(self)
|
|
# self.field("id", int)
|
|
# self.field("username", str)
|
|
#
|
|
#
|
|
# class AuthUserSort(Sort[AuthUser]):
|
|
# def __init__(self):
|
|
# Sort.__init__(self)
|
|
# self.field("id", SortOrder)
|
|
# self.field("username", SortOrder)
|
|
#
|
|
# class AuthUserGraphType(GraphType[AuthUser]):
|
|
#
|
|
# def __init__(self):
|
|
# GraphType.__init__(self)
|
|
#
|
|
# self.int_field(
|
|
# "id",
|
|
# resolver=lambda root: root.id,
|
|
# )
|
|
# self.string_field(
|
|
# "username",
|
|
# resolver=lambda root: root.username,
|
|
# )
|
|
|
|
class HelloQuery(Query):
|
|
def __init__(self):
|
|
Query.__init__(self)
|
|
self.string_field(
|
|
"message",
|
|
resolver=lambda name: f"Hello {name} {get_request().state.request_id}",
|
|
).with_argument("name", str, "Name to greet", "world")
|
|
|
|
self.collection_field(
|
|
UserGraphType,
|
|
"users",
|
|
UserFilter,
|
|
UserSort,
|
|
resolver=lambda: users,
|
|
)
|
|
self.collection_field(
|
|
CityGraphType,
|
|
"cities",
|
|
CityFilter,
|
|
CitySort,
|
|
resolver=lambda: cities,
|
|
)
|
|
# self.dao_collection_field(
|
|
# AuthUserGraphType,
|
|
# AuthUserDao,
|
|
# "authUsers",
|
|
# AuthUserFilter,
|
|
# AuthUserSort,
|
|
# )
|