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')
