Browse Source

Attach actual user objects not their IDs when resurrecting PM thread

Alec Nikolas Reiter 7 years ago
parent
commit
a1f124bb67
1 changed files with 5 additions and 4 deletions
  1. 5 4
      flaskbb/message/views.py

+ 5 - 4
flaskbb/message/views.py

@@ -21,11 +21,10 @@ from flaskbb.extensions import db
 from flaskbb.message.forms import ConversationForm, MessageForm
 from flaskbb.message.forms import ConversationForm, MessageForm
 from flaskbb.message.models import Conversation, Message
 from flaskbb.message.models import Conversation, Message
 from flaskbb.user.models import User
 from flaskbb.user.models import User
-from flaskbb.utils.helpers import (format_quote, register_view,
+from flaskbb.utils.helpers import (format_quote, real, register_view,
                                    render_template, time_utcnow)
                                    render_template, time_utcnow)
 from flaskbb.utils.settings import flaskbb_config
 from flaskbb.utils.settings import flaskbb_config
 
 
-
 logger = logging.getLogger(__name__)
 logger = logging.getLogger(__name__)
 
 
 message = Blueprint("message", __name__)
 message = Blueprint("message", __name__)
@@ -108,8 +107,10 @@ class ViewConversation(MethodView):
             # then we have to change the id's a bit.
             # then we have to change the id's a bit.
             if current_user.id == conversation.to_user_id:
             if current_user.id == conversation.to_user_id:
                 to_user_id = conversation.from_user_id
                 to_user_id = conversation.from_user_id
+                to_user = conversation.from_user
             else:
             else:
                 to_user_id = conversation.to_user_id
                 to_user_id = conversation.to_user_id
+                to_user = conversation.to_user
 
 
             form.save(conversation=conversation, user_id=current_user.id)
             form.save(conversation=conversation, user_id=current_user.id)
 
 
@@ -125,8 +126,8 @@ class ViewConversation(MethodView):
             if conversation is None:
             if conversation is None:
                 conversation = Conversation(
                 conversation = Conversation(
                     subject=old_conv.subject,
                     subject=old_conv.subject,
-                    from_user_id=current_user.id,
-                    to_user=to_user_id,
+                    from_user=real(current_user),
+                    to_user=to_user,
                     user_id=to_user_id,
                     user_id=to_user_id,
                     shared_id=old_conv.shared_id
                     shared_id=old_conv.shared_id
                 )
                 )