diff --git a/app.py b/app.py index 6bbceeb..922209d 100644 --- a/app.py +++ b/app.py @@ -177,6 +177,14 @@ def habit_create(): except ValueError: errors['times'] = 'Die Anzahl muss eine Zahl sein.' + # Check that times is greater than 0 + if times <= 0: + errors['times'] = 'Die Anzahl muss größer als 0 sein.' + + # Check that unit is valid + if unit not in ['Tag', 'Woche', 'Monat', 'Jahr']: + errors['unit'] = 'Die Einheit ist ungültig.' + if errors: return render_template( 'habit.html',