minor changes in demo/solar_system/correct-first_3_planets.py

Miguel Barão
1 parent f9bea841
Showing 1 changed file with 10 additions and 9 deletions Show diff stats
demo/solar_system/correct-first_3_planets.py
@@ -2,26 +2,27 @@
 import re
 import sys
+import time
 s = sys.stdin.read()
-# set of words converted to lowercase
-answer = set(re.findall(r'[\w]+', s.lower()))
-answer.difference_update({'e', 'a', 'planeta', 'planetas'})  # ignore these
+ans = set(re.findall(r'[\w]+', s.lower()))  # convert answer to lowercase
+ans.difference_update({'e', 'a', 'o', 'planeta', 'planetas'})  # ignore words
 # correct set of planets
 planets = {'mercúrio', 'vénus', 'terra'}
-correct = set.intersection(answer, planets)     # the ones I got right
-wrong = set.difference(answer, planets)         # the ones I got wrong
+correct = set.intersection(ans, planets)     # the ones I got right
+wrong = set.difference(ans, planets)         # the ones I got wrong
 grade = (len(correct) - len(wrong)) / len(planets)
+comments = 'Certo' if grade == 1.0 else 'as iniciais dos planetas são M, V e T'
+
 out = f'''---
-grade: {grade}'''
+grade: {grade}
+comments: {comments}'''
-if grade < 1.0:
-    out += '\ncomments: Vou dar uma ajuda, as iniciais são M, V e T...'
+time.sleep(5)
 print(out)
-exit(0)