Split Quiz logic into own file quiz.py

Author: davidindub

.vscode/settings.json

@@ -1,8 +1,8 @@
 {
-   "python.linting.pylintEnabled": true,
+   "python.linting.pylintEnabled": false,
    "python.linting.enabled": true,
-   "python.linting.pycodestyleEnabled": false,
-   "python.linting.flake8Enabled": true,
+   "python.linting.pycodestyleEnabled": true,
+   "python.linting.flake8Enabled": false,
    "python.terminal.activateEnvironment": false,
    "python.formatting.autopep8Path": "/home/gitpod/.pyenv/shims/autopep8",
    "python.linting.flake8Path": "/home/gitpod/.pyenv/shims/flake8",

quiz.py

@@ -0,0 +1,128 @@
+import requests
+import json
+import random
+from pprint import pprint
+
+ALL_CATEGORIES = json.loads(requests.get(
+    "https://opentdb.com/api_category.php").text)["trivia_categories"]
+
+SESSION_TOKEN = json.loads(requests.get(
+    "https://opentdb.com/api_token.php?command=request").text)["token"]
+
+
+def get_quiz_questions(num_qs, cat, diff):
+    """
+    Makes a call to the Quiz API and returns a list of
+    multiple choice questions with category, type, difficulty,
+    question, correct_answer and incorrect_answers
+    """
+
+    return json.loads(requests.get(
+        f"https://opentdb.com/api.php?amount={num_qs}&category={cat}&difficulty={diff}&type=multiple&token={SESSION_TOKEN}").text)["results"]  # noqa
+
+
+class Game:
+    """
+    Generates a Quiz Game,
+    Expects 5 parameters: Quiz Title, Number of Rounds,
+    Number of Questions in a round, a list of the categories
+    for each round, and the difficulty
+    """
+
+    def __init__(self, quiz_title, num_rounds, num_qs, categories, difficulty):
+        self.quiz_title = quiz_title
+        self.num_rounds = num_rounds
+        self.num_questions = num_qs
+        self.categories = categories
+        self.difficulty = difficulty
+
+        self.rounds = [Round(self.num_questions, self.categories[x],
+                             self.difficulty) for x in range(self.num_rounds)]
+        print(self.rounds)
+
+        for round in self.rounds:
+            print(round.__dict__)
+
+
+class Round:
+    """
+    Generates a Quiz Round
+    Expects three parameters: Num of Questions,
+    Category, and Difficulty
+    """
+
+    def __init__(self, num_qs, category, difficulty):
+
+        self.num_qs = num_qs
+        self.category = category
+        self.difficulty = difficulty
+
+        self.question_data = get_quiz_questions(num_qs, category, difficulty)
+
+        self.questions_list = [Question(**question)
+                               for question in self.question_data]
+
+
+class Question:
+    """
+    Creates a Quiz Question instance
+    """
+
+    def __init__(
+        self, category, type, difficulty, question,
+            correct_answer, incorrect_answers):
+
+        self.category = category
+        self.qtype = type
+        self.difficulty = difficulty
+        self.question = question
+        self.correct_answer = correct_answer
+        self.incorrect_answers = incorrect_answers
+
+    def ask(self):
+        return self.question
+
+
+def get_quiz_categories(categories):
+
+    print_cats = ""
+    for x in range(len(categories)):
+        print_cats += (f"{categories[x]['id']}: {categories[x]['name']} \n")
+
+    return print_cats
+
+
+def setup_new_quiz():
+    """
+    Asks user for inputs and creates a new quiz based on the inputs
+    """
+
+    # ADD VALIDATION TO INPUTS
+
+    # Add loops to ensure all inputs are completed
+
+    title = input("What is the name of this Quiz? \n")
+    rounds = int(input("How many rounds should the quiz have? \n"))
+    q_num = int(input("How many questions in each round? \n"))
+
+    # Possibly ask the categories for each round one by one and
+    # add to a list?
+
+    print(get_quiz_categories(ALL_CATEGORIES))
+
+    cats = []
+    for x in range(1, rounds+1):
+        cat = int(input(f"Chose a category for round {x}: \n"))
+        cats.append(cat)
+
+    #   cats = [int(x) for x in input(
+    #     f"Chose {rounds} categories from the list, "
+    #     "separated by commas \n").split(",")]
+
+    print(cats)
+
+    diff = input(
+        "What difficulty should the questions be? \n"
+        "Easy, Medium or Hard? \n").lower()
+
+    return Game(title, rounds, q_num, cats, diff)

run.py

@@ -1,83 +1,35 @@
-import requests
-import json
-import random
-from pprint import pprint
+import quiz
+from datetime import datetime
 
-categories = json.loads(requests.get(
-    "https://opentdb.com/api_category.php").text)["trivia_categories"]
+now = datetime.now()
 
+current_hour = int(now.strftime("%H"))
 
-def get_quiz_questions(num_qs, cat, diff):
-    """
-    Makes a call to the Quiz API and returns a list of
-    multiple choice questions with category, type, difficulty,
-    question, correct_answer and incorrect_answers
-    """
-    return json.loads(requests.get(
-        f"https://opentdb.com/api.php?amount={num_qs}&category={cat}&difficulty={diff}&type=multiple").text)["results"]
-
-
-class Game:
-    """
-    Generates a Quiz Game of several Rounds
-    """
-
-    def __init__(self, quiz_title, num_rounds):
-        self.quiz_title = quiz_title
-        self.num_rounds = num_rounds
-
-
-class Round:
-    """
-    Generates a Quiz Round
-    Expects three parameters: Num of Questions,
-    Category, and Difficulty
-    """
 
-    def __init__(self, num_qs, category, difficulty):
+def pick_greeting(hour):
+    if hour < 6:
+        return "Wow, you're up late!"
+    elif hour < 12:
+        return "Good morning!"
+    elif hour < 17:
+        return "Good afternoon."
+    else:
+        return "Good evening."
 
-        self.num_qs = num_qs
-        self.category = category
-        self.difficulty = difficulty
 
-        self.question_data = get_quiz_questions(num_qs, category, difficulty)
+greeting = pick_greeting(current_hour)
 
-        self.questions_list = [Question(**question)
-                               for question in self.question_data]
 
-
-class Question:
+def main():
     """
-    Creates a Quiz Question instance
+    Code to run on terminal boot
     """
 
-    def __init__(self, category, type, difficulty, question, correct_answer, incorrect_answers):
-
-        self.category = category
-        self.qtype = type
-        self.difficulty = difficulty
-        self.question = question
-        self.correct_answer = correct_answer
-        self.incorrect_answers = incorrect_answers
-
-    def ask(self):
-        return self.question
-
-
-def get_quiz_categories(categories):
-
-    print_cats = ""
-    for x in range(len(categories)):
-        print_cats += (f"{categories[x]['id']}: {categories[x]['name']} \n")
-
-    return print_cats
-
-
-def main():
+    print(f"{greeting} Welcome to QuizMaster 2022! \n")
 
-    print("Welcome to QuizMaster 2022! \n")
+    game_1 = quiz.setup_new_quiz()
 
-    round_1 = Round(5, 9, "easy")
+    print(game_1)
 
 
-main()
+main()