diff --git a/main.py b/main.py index 106525d..cc6ff48 100644 --- a/main.py +++ b/main.py @@ -28,22 +28,36 @@ def clear(): def evaluate(inp, count): inp = str(inp) if re.search("^\d+$", inp): - count = len(inp) - return count + if len(inp) > count: + data = ("msg", "ER01", "Tolika kostkami nemůžete hodit. Zkuste to znovu:", count) + return data + else: + count = len(inp) + data = ("count", count) + return data -def game(count): - new_throw = throw(count) - clear() - print(show_throw(new_throw)) - inp = input("Vyberte, kterými kostkami chcete hodit znovu (např. 135):") - count = evaluate(inp, count) - game(count) +def game(data: tuple): + if data[0] == "count": + count = data[1] + new_throw = throw(count) + clear() + print(show_throw(new_throw)) + inp = input("Vyberte, kterými kostkami chcete hodit znovu (např. 135):") + data = evaluate(inp, count) + game(data) + elif data[0] == "msg": + if data[1] == "ER01": + msg = data[2] + count = data[3] + inp = input(msg) + data = evaluate(inp, count) + game(data) def main(): print("Vítejte ve hře v kostky!") input("Prosím, stiskněte enter pro první hod:") - count = 6 - game(count) + data = ("count", 6) + game(data) if __name__ == "__main__": main()