Fix performance regression caused by last commit.

(imported from commit f0abf93dbab92c5627903284fafad310de039fe6)
This commit is contained in:
Tim Abbott 2012-10-22 22:01:55 -04:00
parent 8c16295301
commit a9667d81a8
1 changed files with 6 additions and 2 deletions

View File

@ -421,7 +421,11 @@ def restore_saved_messages():
# Finally, create all the UserMessage objects
print datetime.datetime.now(), "Importing usermessages, part 1..."
all_messages = Message.objects.select_related().all()
personal_recipients = {}
for r in Recipient.objects.filter(type = Recipient.PERSONAL):
personal_recipients[r.id] = True
all_messages = Message.objects.all()
user_messages_to_create = []
messages_by_id = {}
@ -461,7 +465,7 @@ def restore_saved_messages():
um = UserMessage(user_profile=users_by_id[user_profile_id],
message=message)
user_messages_to_create.append(um)
if message.recipient.type == Recipient.PERSONAL:
if message.recipient_id in personal_recipients:
# Include the sender in huddle recipients
um = UserMessage(user_profile=users_by_id[message.sender_id],
message=message)