realityone 9 лет назад
Родитель
Сommit
9f0ca9137c
2 измененных файлов с 5 добавлено и 5 удалено
  1. 3 2
      flaskbb/message/forms.py
  2. 2 3
      flaskbb/message/models.py

+ 3 - 2
flaskbb/message/forms.py

@@ -47,9 +47,10 @@ class ConversationForm(Form):
             shared_id=shared_id,
             from_user_id=from_user,
             to_user_id=to_user,
-            user_id=user_id
+            user_id=user_id,
+            unread=unread
         )
-        message = Message(message=self.message.data)
+        message = Message(message=self.message.data, user_id=from_user)
         return conversation.save(message=message)
 
 

+ 2 - 3
flaskbb/message/models.py

@@ -98,12 +98,11 @@ class Message(db.Model):
     def save(self, conversation=None):
         """Saves a private message.
 
-        :param conversation_id: The id of the conversation to which the message
-                                belongs to.
+        :param conversation: The  conversation to which the message
+                             belongs to.
         """
         if conversation is not None:
             self.conversation_id = conversation.id
-            self.user_id = conversation.from_user_id
             self.date_created = datetime.utcnow()
 
         db.session.add(self)