fix random queries (#85)

Author: quequitejerina

lessons/day_01/python/resolvers.py

@@ -1,5 +1,5 @@
 from ariadne import QueryType, ObjectType
-from random import randint
+from random import choice
 from models import Skill
 from data import session
 from datetime import datetime
@@ -14,9 +14,8 @@
 # Top level resolver
 @query.field("randomSkill")
 def resolve_random_skill(_, info):
-    records = session.query(Skill).count()
-    random_id = str(randint(1, records))
-    return session.query(Skill).get(random_id)
+    records = [skill.id for skill in session.query(Skill.id)]
+    return session.query(Skill).get(choice(records))
 
 
 # Field level resolvers

lessons/day_02/python/resolvers.py

@@ -1,5 +1,5 @@
 from ariadne import QueryType, ObjectType
-from random import randint
+from random import choice
 from models import Skill, Person
 from data import session
 from datetime import datetime
@@ -15,16 +15,14 @@
 # Top level resolvers
 @query.field("randomSkill")
 def resolve_random_skill(_, info):
-    records = session.query(Skill).count()
-    random_id = str(randint(1, records))
-    return session.query(Skill).get(random_id)
+    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):
-    records = session.query(Person).count()
-    random_id = str(randint(1, records))
-    return session.query(Person).get(random_id)
+    records = [person.id for person in session.query(Person.id)]
+    return session.query(Person).get(choice(records))
 
 
 @query.field("persons")

lessons/day_03/python/resolvers.py

@@ -1,5 +1,5 @@
 from ariadne import QueryType, ObjectType
-from random import randint
+from random import choice
 from models import Skill, Person
 from data import session
 from datetime import datetime
@@ -15,16 +15,14 @@
 # Top level resolvers
 @query.field("randomSkill")
 def resolve_random_skill(_, info):
-    records = session.query(Skill).count()
-    random_id = str(randint(1, records))
-    return session.query(Skill).get(random_id)
+    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):
-    records = session.query(Person).count()
-    random_id = str(randint(1, records))
-    return session.query(Person).get(random_id)
+    records = [person.id for person in session.query(Person.id)]
+    return session.query(Person).get(choice(records))
 
 
 @query.field("persons")

lessons/day_04/python/resolvers.py

@@ -1,5 +1,5 @@
 from ariadne import QueryType, ObjectType, EnumType
-from random import randint
+from random import choice
 from models import Skill, Person
 from data import session
 from datetime import datetime
@@ -24,16 +24,14 @@
 # Top level resolvers
 @query.field("randomSkill")
 def resolve_random_skill(_, info):
-    records = session.query(Skill).count()
-    random_id = str(randint(1, records))
-    return session.query(Skill).get(random_id)
+    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):
-    records = session.query(Person).count()
-    random_id = str(randint(1, records))
-    return session.query(Person).get(random_id)
+    records = [person.id for person in session.query(Person.id)]
+    return session.query(Person).get(choice(records))
 
 
 @query.field("person")

lessons/day_05/python/resolvers.py

@@ -1,5 +1,5 @@
 from ariadne import QueryType, ObjectType, EnumType, MutationType
-from random import randint
+from random import choice
 from models import Skill, Person
 from data import session
 from datetime import datetime
@@ -26,16 +26,14 @@
 # Top level resolvers
 @query.field("randomSkill")
 def resolve_random_skill(_, info):
-    records = session.query(Skill).count()
-    random_id = str(randint(1, records))
-    return session.query(Skill).get(random_id)
+    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):
-    records = session.query(Person).count()
-    random_id = str(randint(1, records))
-    return session.query(Person).get(random_id)
+    records = [person.id for person in session.query(Person.id)]
+    return session.query(Person).get(choice(records))
 
 
 @query.field("person")

lessons/day_06/python/resolvers.py

@@ -1,6 +1,12 @@
-from ariadne import QueryType, ObjectType, EnumType, MutationType, \
-    InterfaceType, UnionType
-from random import randint
+from ariadne import (
+    QueryType,
+    ObjectType,
+    EnumType,
+    MutationType,
+    InterfaceType,
+    UnionType
+)
+from random import choice
 from models import Skill, Person
 from data import session
 from datetime import datetime
@@ -74,16 +80,14 @@ def resolve_global_search_type(obj, *_):
 # Top level resolvers
 @query.field("randomSkill")
 def resolve_random_skill(_, info):
-    records = session.query(Skill).count()
-    random_id = str(randint(1, records))
-    return session.query(Skill).get(random_id)
+    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):
-    records = session.query(Person).count()
-    random_id = str(randint(1, records))
-    return session.query(Person).get(random_id)
+    records = [person.id for person in session.query(Person.id)]
+    return session.query(Person).get(choice(records))
 
 
 @query.field("person")

lessons/day_07/python/resolvers.py

@@ -1,6 +1,12 @@
-from ariadne import QueryType, ObjectType, EnumType, MutationType, \
-    InterfaceType, UnionType
-from random import randint
+from ariadne import (
+    QueryType,
+    ObjectType,
+    EnumType,
+    MutationType,
+    InterfaceType,
+    UnionType
+)
+from random import choice
 from models import Skill, Person
 from data import session
 from datetime import datetime
@@ -74,16 +80,14 @@ def resolve_global_search_type(obj, *_):
 # Top level resolvers
 @query.field("randomSkill")
 def resolve_random_skill(_, info):
-    records = session.query(Skill).count()
-    random_id = str(randint(1, records))
-    return session.query(Skill).get(random_id)
+    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):
-    records = session.query(Person).count()
-    random_id = str(randint(1, records))
-    return session.query(Person).get(random_id)
+    records = [person.id for person in session.query(Person.id)]
+    return session.query(Person).get(choice(records))
 
 
 @query.field("person")