diff --git a/server.py b/server.py index 4112082..0a12660 100644 --- a/server.py +++ b/server.py @@ -48,9 +48,18 @@ def purchasePlaces(): competition = [c for c in competitions if c['name'] == request.form['competition']][0] club = [c for c in clubs if c['name'] == request.form['club']][0] placesRequired = int(request.form['places']) - competition['numberOfPlaces'] = int(competition['numberOfPlaces'])-placesRequired - flash('Great-booking complete!') - return render_template('welcome.html', club=club, competitions=competitions) + points = int(club['points']) + if placesRequired <= points: + competition['numberOfPlaces'] = int(competition['numberOfPlaces']) - placesRequired + club['points'] = int(club['points']) - placesRequired + if competition['numberOfPlaces'] < 0: + competition['numberOfPlaces'] = 0 + flash('Great-booking complete!') + else: + flash("You don't have enough points") + return render_template('welcome.html', club=club, + competitions=competitions) + # TODO: Add route for points display