def delete_from_favourites(perf_id): if 'username' not in session: return redirect('/login') user_id = session['user_id'] users_model = UsersModel(db.get_connection()) favourites = users_model.get_favourites(user_id).split(',') if str(perf_id) not in favourites: return redirect('/performances') del favourites[favourites.index(str(perf_id))] users_model.edit_favourites(user_id, ','.join(favourites)) return redirect('/performances')
def add_to_favourites(perf_id): if 'username' not in session: return redirect('/login') user_id = session['user_id'] performances_model = PerformancesModel(db.get_connection()) exists = performances_model.exists(perf_id)[0] if not exists: return redirect('/performances') users_model = UsersModel(db.get_connection()) if str(perf_id) in users_model.get_favourites( session['user_id']).split(','): return redirect('/performances') favourites = users_model.get_favourites(user_id) if len(favourites) > 0: favourites += ',{}'.format(perf_id) else: favourites += str(perf_id) users_model.edit_favourites(user_id, favourites) return redirect('/favourites')