resolvers.py

from typing import List, TypeVar
from ariadne import QueryType, ObjectType
from random import choice

from graphql import GraphQLResolveInfo
from models import Skill, Person
from data import session
from datetime import datetime


query = QueryType()

# Type definition
skill = ObjectType("Skill")
person = ObjectType("Person")

ID = TypeVar("ID", int, str)


# Top level resolvers
@query.field("randomSkill")
def resolve_random_skill(_, info: GraphQLResolveInfo) -> Skill | None:
    records = [skill.id for skill in session.query(Skill.id)]
    return session.query(Skill).get(choice(records))


@query.field("randomPerson")
def resolve_random_person(_, info: GraphQLResolveInfo) -> Skill | None:
    records = [person.id for person in session.query(Person.id)]
    return session.query(Person).get(choice(records))


@query.field("persons")
def resolve_persons(_, info: GraphQLResolveInfo, id: ID | None = None):
    return session.query(Person).filter_by(id=id) if id else session.query(Person).all()


@query.field("person")
def resolve_person(_, info: GraphQLResolveInfo, id: ID | None = None):
    return session.query(Person).get(id)


@query.field("skills")
def resolve_skills(_, info: GraphQLResolveInfo, id: ID | None = None):
    return session.query(Skill).filter_by(id=id) if id else session.query(Skill).all()


@query.field("skill")
def resolve_skill(_, info: GraphQLResolveInfo, id: ID | None = None):
    return session.query(Skill).get(id)


# Field level resolvers
@skill.field("now")
def resolve_now(_, info: GraphQLResolveInfo) -> datetime:
    return datetime.now()


@skill.field("parent")
def resolve_parent(obj, info: GraphQLResolveInfo) -> Skill:
    return obj.parent_skill


@person.field("fullName")
def resolve_full_name(obj, info: GraphQLResolveInfo) -> str:
    return f"{obj.name} {obj.surname}"


@person.field("friends")
def resolve_friends(obj, info: GraphQLResolveInfo, id: ID | None = None) -> Person:
    return obj.friends.filter_by(id=id) if id else obj.friends


@person.field("skills")
def resolve_person_skills(
    obj, info: GraphQLResolveInfo, id: ID | None = None
) -> List[Skill]:
    return obj.skills.filter_by(id=id) if id else obj.skills


@person.field("favSkill")
def resolve_fav_skill(obj, info: GraphQLResolveInfo) -> Skill:
    return obj.person_favSkill