diff --git a/main.py b/main.py index cc6ff48..7ff30e6 100644 --- a/main.py +++ b/main.py @@ -31,7 +31,15 @@ def evaluate(inp, count): if len(inp) > count: data = ("msg", "ER01", "Tolika kostkami nemůžete hodit. Zkuste to znovu:", count) return data - else: + else: + for x in inp: + if int(x) > count: + if count == 1: + range = "1" + else: + range = f"1–{count}" + data = ("msg", "ER02", f"Zvolte prosím kostky v platném rozsahu ({range}):", count) + return data count = len(inp) data = ("count", count) return data @@ -46,7 +54,7 @@ def game(data: tuple): data = evaluate(inp, count) game(data) elif data[0] == "msg": - if data[1] == "ER01": + if data[1] == "ER01" or data[1] == "ER02": msg = data[2] count = data[3] inp = input(msg)