Added read/unread pms

flaskbb/pms/models.py

@@ -25,8 +25,9 @@ class PrivateMessage(db.Model):
     date_created = db.Column(db.DateTime, default=datetime.utcnow())
     trash = db.Column(db.Boolean, nullable=False, default=False)
     draft = db.Column(db.Boolean, nullable=False, default=False)
+    unread = db.Column(db.Boolean, nullable=False, default=True)
 
-    user = db.relationship("User", lazy="joined", foreign_keys=[user_id])
+    user = db.relationship("User", backref="pms", lazy="joined", foreign_keys=[user_id])
     from_user = db.relationship("User", lazy="joined", foreign_keys=[from_user_id])
     to_user = db.relationship("User", lazy="joined", foreign_keys=[to_user_id])
 

flaskbb/pms/views.py

@@ -33,10 +33,12 @@ def inbox():
 
 
 @pms.route("/message/<int:id>")
-@pms.route("/message/<int:id>/view")
 @login_required
 def view_message(id):
     message = PrivateMessage.query.filter_by(id=id).first()
+    if message.unread:
+        message.unread=False
+        db.session.commit()
     return render_template("pms/view_message.html", message=message)