Typing (#88)

* typing

Author: quequitejerina

lessons/day_01/python/resolvers.py

@@ -1,5 +1,8 @@
+from typing import Any
 from ariadne import QueryType, ObjectType
 from random import choice
+
+from graphql import GraphQLResolveInfo
 from models import Skill
 from data import session
 from datetime import datetime
@@ -13,17 +16,17 @@
 
 # Top level resolver
 @query.field("randomSkill")
-def resolve_random_skill(_, info):
+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))
 
 
 # Field level resolvers
 @skill.field("now")
-def resolve_now(_, info):
+def resolve_now(_, info: GraphQLResolveInfo) -> datetime:
     return datetime.now()
 
 
 @skill.field("parent")
-def resolve_parent(obj, info):
+def resolve_parent(obj: Any, info: GraphQLResolveInfo) -> Skill:
     return obj.parent_skill

lessons/day_02/python/resolvers.py

@@ -1,5 +1,8 @@
+from typing import Any, List
 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
@@ -14,48 +17,52 @@
 
 # Top level resolvers
 @query.field("randomSkill")
-def resolve_random_skill(_, info):
+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):
+def resolve_random_person(_, info: GraphQLResolveInfo) -> Person | 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, id=None):
-    return session.query(Person).filter_by(id=id) if id else session.query(Person).all()
+def resolve_persons(_, info: GraphQLResolveInfo, id: int | None = None) -> List[Person]:
+    return (
+        session.query(Person).filter_by(id=id).all()
+        if id
+        else session.query(Person).all()
+    )
 
 
 # Field level resolvers
 @skill.field("now")
-def resolve_now(_, info):
+def resolve_now(_, info: GraphQLResolveInfo) -> datetime:
     return datetime.now()
 
 
 @skill.field("parent")
-def resolve_parent(obj, info):
+def resolve_parent(obj: Any, info: GraphQLResolveInfo) -> Skill:
     return obj.parent_skill
 
 
 @person.field("fullName")
-def resolve_full_name(obj, info):
-    return f'{obj.name} {obj.surname}'
+def resolve_full_name(obj: Any, info: GraphQLResolveInfo) -> str:
+    return f"{obj.name} {obj.surname}"
 
 
 @person.field("friends")
-def resolve_friends(obj, info):
+def resolve_friends(obj: Any, info: GraphQLResolveInfo) -> Person:
     return obj.friends
 
 
 @person.field("skills")
-def resolve_skills(obj, info):
+def resolve_skills(obj: Any, info: GraphQLResolveInfo):
     return obj.skills
 
 
 @person.field("favSkill")
-def resolve_fav_skill(obj, info):
+def resolve_fav_skill(obj: Any, info: GraphQLResolveInfo):
     return obj.person_favSkill

lessons/day_03/python/resolvers.py

@@ -1,5 +1,8 @@
+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
@@ -11,66 +14,70 @@
 skill = ObjectType("Skill")
 person = ObjectType("Person")
 
+ID = TypeVar("ID", int, str)
+
 
 # Top level resolvers
 @query.field("randomSkill")
-def resolve_random_skill(_, info):
+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):
+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, id=None):
+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, id=None):
+def resolve_person(_, info: GraphQLResolveInfo, id: ID | None = None):
     return session.query(Person).get(id)
 
 
 @query.field("skills")
-def resolve_skills(_, info, id=None):
+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, id=None):
+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):
+def resolve_now(_, info: GraphQLResolveInfo) -> datetime:
     return datetime.now()
 
 
 @skill.field("parent")
-def resolve_parent(obj, info):
+def resolve_parent(obj, info: GraphQLResolveInfo) -> Skill:
     return obj.parent_skill
 
 
 @person.field("fullName")
-def resolve_full_name(obj, info):
-    return f'{obj.name} {obj.surname}'
+def resolve_full_name(obj, info: GraphQLResolveInfo) -> str:
+    return f"{obj.name} {obj.surname}"
 
 
 @person.field("friends")
-def resolve_friends(obj, info, id=None):
+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, id=None):
+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):
+def resolve_fav_skill(obj, info: GraphQLResolveInfo) -> Skill:
     return obj.person_favSkill

lessons/day_04/python/resolvers.py

@@ -1,5 +1,8 @@
+from typing import Any, Dict, List
 from ariadne import QueryType, ObjectType, EnumType
 from random import choice
+
+from graphql import GraphQLResolveInfo
 from models import Skill, Person
 from data import session
 from datetime import datetime
@@ -13,73 +16,81 @@
 eye_color = EnumType(
     "EyeColor",
     {
-        'BLUE': 'blue',
-        'GREEN': 'green',
-        'BROWN': 'brown',
-        'BLACK': 'black',
+        "BLUE": "blue",
+        "GREEN": "green",
+        "BROWN": "brown",
+        "BLACK": "black",
     },
 )
 
 
 # Top level resolvers
 @query.field("randomSkill")
-def resolve_random_skill(_, info):
+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):
+def resolve_random_person(_, info: GraphQLResolveInfo) -> Person | None:
     records = [person.id for person in session.query(Person.id)]
     return session.query(Person).get(choice(records))
 
 
 @query.field("person")
-def resolve_person(_, info, input=None):
+def resolve_person(
+    _, info: GraphQLResolveInfo, input: Dict | None = None
+) -> Person | None:
     return session.query(Person).filter_by(**input).first() if input else None
 
 
 @query.field("persons")
-def resolve_persons(_, info, input={}):
+def resolve_persons(_, info: GraphQLResolveInfo, input: Dict = {}) -> List[Person]:
     return session.query(Person).filter_by(**input).all()
 
 
 @query.field("skill")
-def resolve_skill(_, info, input=None):
+def resolve_skill(
+    _, info: GraphQLResolveInfo, input: Dict | None = None
+) -> Skill | None:
     return session.query(Skill).filter_by(**input).first() if input else None
 
 
 @query.field("skills")
-def resolve_skills(_, info, input={}):
+def resolve_skills(_, info: GraphQLResolveInfo, input: Dict = {}) -> List[Skill]:
     return session.query(Skill).filter_by(**input).all()
 
 
 # Field level resolvers
 @skill.field("now")
-def resolve_now(_, info):
+def resolve_now(_, info: GraphQLResolveInfo) -> datetime:
     return datetime.now()
 
 
 @skill.field("parent")
-def resolve_parent(obj, info):
+def resolve_parent(obj: Any, info: GraphQLResolveInfo) -> Skill:
     return obj.parent_skill
 
 
 @person.field("fullName")
-def resolve_full_name(obj, info):
-    return f'{obj.name} {obj.surname}'
+def resolve_full_name(obj: Any, info: GraphQLResolveInfo) -> str:
+    return f"{obj.name} {obj.surname}"
 
 
 @person.field("friends")
-def resolve_friends(obj, info, input={}):
+def resolve_friends(
+    obj: Any, info: GraphQLResolveInfo, input: Dict = {}
+) -> List[Person]:
     return obj.friends.filter_by(**input).all()
 
 
 @person.field("skills")
-def resolve_person_skills(obj, info, input={}):
+def resolve_person_skills(
+    obj: Any, info: GraphQLResolveInfo, input: Dict = {}
+) -> List[Skill]:
     return obj.skills.filter_by(**input).all()
 
 
 @person.field("favSkill")
-def resolve_fav_skill(obj, info):
+def resolve_fav_skill(obj: Any, info: GraphQLResolveInfo) -> Skill:
     return obj.person_favSkill