Use prereg.users.set() in do_invite_users.

This is a bit more idiomatic for many-to-many relationships.
This commit is contained in:
Steve Howell
2017-09-17 09:49:23 -07:00
committed by Tim Abbott
parent 8ad7133351
commit 9046efb71a

View File

@@ -3534,11 +3534,9 @@ def do_invite_users(user_profile, invitee_emails, streams, body=None):
# The logged in user is the referrer.
prereg_user = PreregistrationUser(email=email, referred_by=user_profile)
# We save twice because you cannot associate a ManyToMany field
# on an unsaved object.
prereg_user.save()
prereg_user.streams = streams
prereg_user.save()
stream_ids = [stream.id for stream in streams]
prereg_user.streams.set(stream_ids)
event = {"email": prereg_user.email, "referrer_id": user_profile.id, "email_body": body}
queue_json_publish("invites", event,