Add calc.py, scripts/brain_calc.py, apdate pyproject.toml

Author: lytic11-web

brain_games/games/calc.py

@@ -0,0 +1,24 @@
+import secrets
+import operator
+
+
+def calculate(num1: int, num2: int, operation: str) -> int:
+    """Calculate the result of two numbers with given operation."""
+    operations = {
+        '+': operator.add,
+        '-': operator.sub,
+        '*': operator.mul
+    }
+    return operations[operation](num1, num2)
+
+
+def generate_round() -> tuple[str, str]:
+    """Generate a round for the calculator game."""
+    num1 = secrets.randbelow(50) + 1  # 1-50
+    num2 = secrets.randbelow(50) + 1  # 1-50
+    operation = secrets.choice(['+', '-', '*'])
+
+    question = f"{num1} {operation} {num2}"
+    correct_answer = str(calculate(num1, num2, operation))
+
+    return question, correct_answer

brain_games/scripts/brain_calc.py

@@ -0,0 +1,13 @@
+#!/usr/bin/env python3
+from brain_games.engine import run_game
+from brain_games.games.calc import generate_round
+
+DESCRIPTION = 'What is the result of the expression?'
+
+
+def main():
+    run_game(DESCRIPTION, generate_round)
+
+
+if __name__ == '__main__':
+    main()

pyproject.toml

@@ -23,3 +23,4 @@ dev = [
 [project.scripts]
 brain-games = "brain_games.scripts.brain_games:main"
 brain-even = "brain_games.scripts.brain_even:main"
+brain-calc = "brain_games.scripts.brain_calc:main"