попытка сделать движок

Author: DenisShutov

brain_games/engine.py

@@ -0,0 +1,35 @@
+import prompt
+
+from brain_games.cli import welcome_user
+from brain_games.scripts.brain_games import greet
+
+
+def get_conditions():
+    raise NotImplementedError("Define get_conditions() in game module")
+
+
+def engine_game():
+    greet()
+    name = welcome_user()
+    description = ''
+    print(description)
+    
+    answer = 0
+    rounds = 3
+
+    while answer < rounds:
+        question, correct_answer = get_conditions()
+        print(f'Question: {question}')
+
+        user_answer = prompt.string('Your answer: ')
+
+        if user_answer == correct_answer:
+            print('Correct!')
+            answer += 1
+        else:
+            print(f"'{user_answer}' is wrong answer ;(."
+                  f" Correct answer was '{correct_answer}'.")
+            print(f"Let's try again, {name}!")
+            break
+    if answer == rounds:
+        print(f'Congratulations, {name}!')

brain_games/games/__init__.py

@@ -0,0 +1,38 @@
+from random import choice, randrange
+
+import prompt
+
+from brain_games.cli import welcome_user
+
+
+def brain_calc():
+    name = welcome_user()
+    print('What is the result of the expression?')
+
+    answer = 0
+    while answer < 3:
+        n1 = randrange(1, 51)
+        n2 = randrange(1, 51)
+        operator = choice(['+', '-', '*'])
+
+        if operator == "+":
+            res = n1 + n2
+            print(f'Question: {n1} + {n2}')
+        elif operator == "-":
+            res = n1 - n2
+            print(f'Question: {n1} - {n2}')
+        elif operator == "*":
+            res = n1 * n2
+            print(f'Question: {n1} * {n2}')
+        
+        user_answer = prompt.integer('Your answer: ')
+        if user_answer == res:
+            print('Correct!')
+            answer += 1
+        else:
+            print(f"'{user_answer}' is wrong answer ;(." 
+                  f"Correct answer was '{res}.'")
+            print(f"Let's try again, {name}!")
+            break
+    if answer == 3:
+        print(f'Congratulations, {name}!')