How I Built an AWS Trivia Game with Amazon Q-CLI—No Sweat!

import random
import time
import json
import os
import sys
from typing import List, Dict, Any, Optional

class AwsTrivia:
    def __init__(self):
        self.questions = []
        self.score = 0
        self.total_questions = 0
        self.categories = set()
        self.load_questions()

    def load_questions(self):
        """Load questions from the questions.json file or use default questions if file doesn't exist."""
        try:
            with open('aws_trivia_questions.json', 'r') as f:
                self.questions = json.load(f)
                for question in self.questions:
                    self.categories.add(question.get('category', 'General'))
        except FileNotFoundError:
            # Default questions if file doesn't exist
            self.questions = [
                {
                    "question": "What does EC2 stand for?",
                    "options": [
                        "Elastic Cloud Computing",
                        "Elastic Compute Cloud",
                        "Electronic Compute Cloud",
                        "Extended Cloud Computing"
                    ],
                    "answer": 1,  # 0-indexed
                    "category": "Compute",
                    "difficulty": "Easy"
                },
                {
                    "question": "Which AWS service is used for object storage?",
                    "options": [
                        "EBS",
                        "EFS",
                        "S3",
                        "FSx"
                    ],
                    "answer": 2,
                    "category": "Storage",
                    "difficulty": "Easy"
                },
                {
                    "question": "Which AWS service is used for NoSQL database?",
                    "options": [
                        "RDS",
                        "DynamoDB",
                        "Redshift",
                        "Aurora"
                    ],
                    "answer": 1,
                    "category": "Database",
                    "difficulty": "Easy"
                },
                {
                    "question": "What is the default timeout for a Lambda function?",
                    "options": [
                        "1 minute",
                        "3 seconds",
                        "3 minutes",
                        "15 minutes"
                    ],
                    "answer": 2,
                    "category": "Serverless",
                    "difficulty": "Medium"
                },
                {
                    "question": "Which AWS service is used for content delivery?",
                    "options": [
                        "Route 53",
                        "CloudFront",
                        "API Gateway",
                        "AppSync"
                    ],
                    "answer": 1,
                    "category": "Networking",
                    "difficulty": "Medium"
                },
                {
                    "question": "What is the maximum size of an object in S3?",
                    "options": [
                        "1 GB",
                        "100 GB",
                        "1 TB",
                        "5 TB"
                    ],
                    "answer": 3,
                    "category": "Storage",
                    "difficulty": "Medium"
                },
                {
                    "question": "Which AWS service is used for container orchestration?",
                    "options": [
                        "ECR",
                        "ECS",
                        "Fargate",
                        "Elastic Beanstalk"
                    ],
                    "answer": 1,
                    "category": "Containers",
                    "difficulty": "Medium"
                },
                {
                    "question": "What is the AWS Well-Architected Framework pillar that focuses on system recovery?",
                    "options": [
                        "Operational Excellence",
                        "Security",
                        "Reliability",
                        "Performance Efficiency"
                    ],
                    "answer": 2,
                    "category": "Architecture",
                    "difficulty": "Hard"
                },
                {
                    "question": "Which AWS service provides a virtual network dedicated to your AWS account?",
                    "options": [
                        "VPC",
                        "Direct Connect",
                        "Transit Gateway",
                        "Global Accelerator"
                    ],
                    "answer": 0,
                    "category": "Networking",
                    "difficulty": "Easy"
                },
                {
                    "question": "Which AWS service would you use for serverless data processing?",
                    "options": [
                        "EC2",
                        "Lambda",
                        "Batch",
                        "ECS"
                    ],
                    "answer": 1,
                    "category": "Serverless",
                    "difficulty": "Easy"
                }
            ]
            for question in self.questions:
                self.categories.add(question.get('category', 'General'))

    def save_questions(self):
        """Save questions to a JSON file."""
        with open('aws_trivia_questions.json', 'w') as f:
            json.dump(self.questions, f, indent=2)

    def add_question(self, question: str, options: List[str], answer: int,
                    category: str = "General", difficulty: str = "Medium"):
        """Add a new question to the trivia database."""
        new_question = {
            "question": question,
            "options": options,
            "answer": answer,
            "category": category,
            "difficulty": difficulty
        }
        self.questions.append(new_question)
        self.categories.add(category)
        self.save_questions()

    def filter_questions(self, category: Optional[str] = None, difficulty: Optional[str] = None) -> List[Dict[str, Any]]:
        """Filter questions by category and/or difficulty."""
        filtered = self.questions

        if category:
            filtered = [q for q in filtered if q.get('category', 'General') == category]

        if difficulty:
            filtered = [q for q in filtered if q.get('difficulty', 'Medium') == difficulty]

        return filtered

    def display_question(self, question_data: Dict[str, Any]) -> bool:
        """Display a question and get the user's answer. Return True if correct."""
        print("\n" + "="*60)
        print(f"\n{question_data['question']}")
        print("-"*40)

        for i, option in enumerate(question_data['options']):
            print(f"{i+1}. {option}")

        try:
            user_answer = int(input("\nYour answer (number): ")) - 1  # Convert to 0-indexed

            if user_answer == question_data['answer']:
                print("\n✅ Correct! Well done!")
                return True
            else:
                correct_option = question_data['options'][question_data['answer']]
                print(f"\n❌ Wrong! The correct answer is: {question_data['answer']+1}. {correct_option}")
                return False

        except (ValueError, IndexError):
            print("\n❌ Invalid input. The correct answer is:",
                  f"{question_data['answer']+1}. {question_data['options'][question_data['answer']]}")
            return False

    def play_game(self, num_questions: int = 5, category: Optional[str] = None, difficulty: Optional[str] = None):
        """Start the trivia game with the specified number of questions."""
        filtered_questions = self.filter_questions(category, difficulty)

        if not filtered_questions:
            print("No questions available with the selected filters.")
            return

        if num_questions > len(filtered_questions):
            print(f"Only {len(filtered_questions)} questions available with your filters.")
            num_questions = len(filtered_questions)

        selected_questions = random.sample(filtered_questions, num_questions)
        self.score = 0
        self.total_questions = num_questions

        print("\n" + "="*60)
        print(f"AWS TRIVIA GAME - {num_questions} Questions")
        if category:
            print(f"Category: {category}")
        if difficulty:
            print(f"Difficulty: {difficulty}")
        print("="*60)

        for i, question in enumerate(selected_questions):
            print(f"\nQuestion {i+1} of {num_questions}")
            if self.display_question(question):
                self.score += 1
            time.sleep(1)

        self.show_results()

    def show_results(self):
        """Display the final score and feedback."""
        print("\n" + "="*60)
        print(f"GAME OVER! Your score: {self.score}/{self.total_questions}")

        percentage = (self.score / self.total_questions) * 100

        if percentage == 100:
            print("Perfect score! You're an AWS expert! 🏆")
        elif percentage >= 80:
            print("Great job! You really know your AWS services! 🌟")
        elif percentage >= 60:
            print("Good work! You have solid AWS knowledge. 👍")
        elif percentage >= 40:
            print("Not bad! Keep learning about AWS. 📚")
        else:
            print("Time to brush up on your AWS knowledge! 💪")

        print("="*60)

    def show_menu(self):
        """Display the main menu and handle user input."""
        while True:
            print("\n" + "="*60)
            print("AWS TRIVIA GAME - MAIN MENU")
            print("="*60)
            print("1. Play Game")
            print("2. Add New Question")
            print("3. View Categories")
            print("4. Exit")

            try:
                choice = int(input("\nEnter your choice (1-4): "))

                if choice == 1:
                    self.play_game_menu()
                elif choice == 2:
                    self.add_question_menu()
                elif choice == 3:
                    self.view_categories()
                elif choice == 4:
                    print("\nThanks for playing AWS Trivia! Goodbye!")
                    sys.exit(0)
                else:
                    print("Invalid choice. Please enter a number between 1 and 4.")

            except ValueError:
                print("Invalid input. Please enter a number.")

    def play_game_menu(self):
        """Menu for configuring and starting a game."""
        print("\n" + "="*60)
        print("PLAY GAME")
        print("="*60)

        # Get number of questions
        while True:
            try:
                num_questions = int(input("How many questions would you like? (1-10): "))
                if 1 <= num_questions <= 10:
                    break
                print("Please enter a number between 1 and 10.")
            except ValueError:
                print("Please enter a valid number.")

        # Get category (optional)
        print("\nCategories:")
        categories_list = sorted(list(self.categories))
        for i, category in enumerate(categories_list):
            print(f"{i+1}. {category}")
        print(f"{len(categories_list)+1}. All Categories")

        try:
            cat_choice = int(input("\nSelect a category (or enter for all): ") or str(len(categories_list)+1))
            if 1 <= cat_choice <= len(categories_list):
                selected_category = categories_list[cat_choice-1]
            else:
                selected_category = None
        except ValueError:
            selected_category = None

        # Get difficulty (optional)
        difficulties = ["Easy", "Medium", "Hard"]
        print("\nDifficulty:")
        for i, diff in enumerate(difficulties):
            print(f"{i+1}. {diff}")
        print("4. All Difficulties")

        try:
            diff_choice = int(input("\nSelect difficulty (or enter for all): ") or 4)
            if 1 <= diff_choice <= 3:
                selected_difficulty = difficulties[diff_choice-1]
            else:
                selected_difficulty = None
        except ValueError:
            selected_difficulty = None

        # Start the game
        self.play_game(num_questions, selected_category, selected_difficulty)

    def add_question_menu(self):
        """Menu for adding a new question."""
        print("\n" + "="*60)
        print("ADD NEW QUESTION")
        print("="*60)

        question = input("Enter the question: ")

        options = []
        for i in range(4):
            option = input(f"Enter option {i+1}: ")
            options.append(option)

        while True:
            try:
                answer = int(input("Enter the correct answer number (1-4): "))
                if 1 <= answer <= 4:
                    break
                print("Please enter a number between 1 and 4.")
            except ValueError:
                print("Please enter a valid number.")

        # Get or create category
        print("\nExisting categories:")
        categories_list = sorted(list(self.categories))
        for i, category in enumerate(categories_list):
            print(f"{i+1}. {category}")
        print(f"{len(categories_list)+1}. Create new category")

        try:
            cat_choice = int(input("\nSelect a category: "))
            if 1 <= cat_choice <= len(categories_list):
                category = categories_list[cat_choice-1]
            else:
                category = input("Enter new category name: ")
        except ValueError:
            category = "General"

        # Get difficulty
        difficulties = ["Easy", "Medium", "Hard"]
        print("\nDifficulty:")
        for i, diff in enumerate(difficulties):
            print(f"{i+1}. {diff}")

        try:
            diff_choice = int(input("\nSelect difficulty (1-3): "))
            if 1 <= diff_choice <= 3:
                difficulty = difficulties[diff_choice-1]
            else:
                difficulty = "Medium"
        except ValueError:
            difficulty = "Medium"

        # Add the question
        self.add_question(question, options, answer-1, category, difficulty)
        print("\nQuestion added successfully!")

    def view_categories(self):
        """Display all available categories and their question counts."""
        print("\n" + "="*60)
        print("CATEGORIES")
        print("="*60)

        category_counts = {}
        for question in self.questions:
            cat = question.get('category', 'General')
            if cat in category_counts:
                category_counts[cat] += 1
            else:
                category_counts[cat] = 1

        for category, count in sorted(category_counts.items()):
            print(f"{category}: {count} questions")

        input("\nPress Enter to continue...")

def main():
    """Main function to start the AWS Trivia application."""
    print("\nWelcome to AWS Trivia!")
    trivia = AwsTrivia()
    trivia.show_menu()

if __name__ == "__main__":
    main()