return value unification

This commit is contained in:
knst-kotov
2021-08-04 13:38:56 +00:00
parent 9467810632
commit a0e99e5ba9
5 changed files with 50 additions and 103 deletions

View File

@@ -1,4 +1,4 @@
from ariadne import MutationType, QueryType, SubscriptionType, ScalarType, InterfaceType, UnionType
from ariadne import MutationType, QueryType, SubscriptionType, ScalarType
query = QueryType()
@@ -12,23 +12,4 @@ datetime_scalar = ScalarType("DateTime")
def serialize_datetime(value):
return value.isoformat()
class ApiError:
def __init__(self, message):
self.message = message
error_interface = InterfaceType("ErrorInterface")
@error_interface.type_resolver
def resolve_search_result_type(obj, *_):
if isinstance(obj, ApiError):
return "ApiError"
sign_in_result = UnionType("SignInResult")
register_user_result = UnionType("RegisterUserResult")
results = [sign_in_result, register_user_result]
resolvers = [query, mutation, subscription, datetime_scalar, error_interface]
resolvers.extend(results)
resolvers = [query, mutation, subscription, datetime_scalar]