def post_favourites(user_id):
# Get the user or return 404 not found
user = User.query.get_or_404(user_id)
# Get the data
data = request.get_json()
# Check that there is data
if not data:
abort(400)
# Check that the data has the correct fields
if not (('movie_tv_id' in data and isinstance(data['movie_tv_id'], int)
and data['movie_tv_id']) or
('media_type' in data and isinstance(data['media_type'], str)
and data['media_type'])):
abort(400)
print(data)
try:
favourite = Favourite(data['movie_tv_id'], data['media_type'])
user.favourites.append(favourite)
favourite.insert()
return jsonify({
"success": True,
"status_code": 200,
"favourite": favourite.format()
})
except Exception as e:
print('Error ', e)
abort(422)
def add(self, song_id):
song = Song.objects.get(id=song_id)
user = User.objects.get(id=self.user_id)
favourite = Favourite(Song=song, User=user)
favourite.save()
return song_id
def post(self):
data = request.get_json()
print("data:",data)
rec = Favourite(user_id = data["user_id"],
post_id = data["post_id"],
created_date = datetime.datetime.now())
db.session.add(rec)
db.session.commit()
return { "message": "{} is your favourite post".format(data["post_id"]),
"code": 200}
def bookmark(pid):
user = g.user
if Favourite.query.filter_by(id=str(user.id) + ':' + str(pid)).first():
flash('The post was already in your collection.')
else:
fav = Favourite(user.id, pid)
db.session.add(fav)
db.session.commit()
flash('The post was added in your collection.')
return redirect(url_for('list_post'))
def add_favourite(plant_id):
if Favourite.query.filter_by(plant_id=plant_id).one_or_none():
flash("You have already added that plant to your favourites list.", category="warning")
return redirect(request.referrer)
if not check_logged_in():
return redirect(url_for('users.login'))
if session.get('user_id') is not None:
favourite = Favourite(session['user_id'], plant_id)
db.session.add(favourite)
db.session.commit()
flash("Plant added to favourites list!", category="success")
return redirect(request.referrer)
else:
return redirect(url_for('users.login'))
def add_favourite(uid, rid):
"""
Inserts a row into the Favourite table.
Args:
uid: A user ID that corresponds to a user in the User table. A integer.
rid: A restaurant ID that corresponds to a restaurant in the Restaurant
table. A integer.
Returns:
None.
"""
fav = Favourite(uid=uid, rid=rid)
db.session.add(fav)
db.session.commit()
