Skip to content
Snippets Groups Projects
Commit eb76bbd2 authored by kevgu916's avatar kevgu916
Browse files

Puts

parent 7a038dc0
No related branches found
No related tags found
No related merge requests found
...@@ -141,12 +141,16 @@ class Notification(db.Model): ...@@ -141,12 +141,16 @@ class Notification(db.Model):
user = db.relationship('User', backref=db.backref('notifications', lazy=True)) user = db.relationship('User', backref=db.backref('notifications', lazy=True))
def create_notification(user_id, message): def create_notification(user_id, message, username):
# Hämta användaren från databasen # Hämta användaren från databasen
# Det är bara relevant att spara 20 notifikationer, därför raderar vi om det är över 30.
user = db.session.get(User, user_id) user = db.session.get(User, user_id)
if user: if user:
# Kolla om username matchar användarens username (alltså att användaren försöker skapa notis till sig själv)
if user.username == username:
# Om samma användare, gör inget
return
# Skapa en ny notis # Skapa en ny notis
new_notification = Notification(user_id=user_id, message=message) new_notification = Notification(user_id=user_id, message=message)
...@@ -162,6 +166,7 @@ def create_notification(user_id, message): ...@@ -162,6 +166,7 @@ def create_notification(user_id, message):
db.session.delete(oldest_notification) db.session.delete(oldest_notification)
db.session.commit() db.session.commit()
# Klass för kommentarer # Klass för kommentarer
class Comment(db.Model): class Comment(db.Model):
id = db.Column(db.Integer, primary_key=True) id = db.Column(db.Integer, primary_key=True)
...@@ -417,7 +422,7 @@ def follow(): ...@@ -417,7 +422,7 @@ def follow():
follower.follow(following) follower.follow(following)
db.session.commit() db.session.commit()
create_notification(following.id, f"{follower.username} just started following you") create_notification(following.id, f"{follower.username} just started following you", follower.username)
return jsonify({"message": f"{follower.username} is now following {following.username}"}), 200 return jsonify({"message": f"{follower.username} is now following {following.username}"}), 200
...@@ -603,7 +608,7 @@ def add_comment(): ...@@ -603,7 +608,7 @@ def add_comment():
db.session.commit() db.session.commit()
# Skapa en notis för användaren som har reviewn. # Skapa en notis för användaren som har reviewn.
create_notification(review.user_id, f"{user.username} commented '{comment_text}' on your review.") create_notification(review.user_id, f"{user.username} commented '{comment_text}' on your review.", user.username)
return jsonify({ return jsonify({
"message": "Comment added successfully", "message": "Comment added successfully",
...@@ -696,7 +701,7 @@ def like_review(): ...@@ -696,7 +701,7 @@ def like_review():
db.session.commit() db.session.commit()
# Skapa en notis för användaren som har reviewn. # Skapa en notis för användaren som har reviewn.
create_notification(review.user_id, f"{user.username} liked your review '{review.drink_name}'") create_notification(review.user_id, f"{user.username} liked your review '{review.drink_name}'", user.username)
return jsonify({"message": f"Review {review.id} liked by {user.username}"}), 200 return jsonify({"message": f"Review {review.id} liked by {user.username}"}), 200
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment