def post(self, request, *args, **kwargs):
""" Return a mapping of participant IDs to membership statuses. """
postdata = json.loads(self.request.body)
par_pks = postdata.get('participant_ids')
if not isinstance(par_pks, list):
return JsonResponse({'message': 'Bad request'}, status=400)
# Span databases to map from participants -> users -> email addresses
participants = models.Participant.objects.filter(pk__in=par_pks)
user_to_par = dict(participants.values_list('user_id', 'pk'))
email_addresses = EmailAddress.objects.filter(
user_id__in=user_to_par, verified=True
)
email_to_user = dict(email_addresses.values_list('email', 'user_id'))
# Gives email -> membership info for all matches
matches = geardb_utils.matching_memberships(email_to_user)
# Default to blank memberships in case not found
participant_memberships = {
pk: geardb_utils.repr_blank_membership() for pk in par_pks
}
# Update participants where matching membership information was found
for email, membership in matches.items():
par_pk = user_to_par[email_to_user[email]]
# We might overwrite a previous membership record, but that will
# only happen if the user has memberships under 2+ emails
# (Older memberships come first, so this will safely yield the newest)
participant_memberships[par_pk] = membership
return JsonResponse({'memberships': participant_memberships})
def test_just_one_record(self):
""" When requesting records under many emails, just one is returned.
(Provided that the primary email is included in the lookup list)
"""
person_id = self.create_tim()
alternate_emails = [f'tim@{i}.example.com' for i in range(3)]
for email in alternate_emails:
self.record_alternate_email(person_id, email)
# When we request just the alternate emails, it returns one for each
self.assertEqual(len(geardb.matching_memberships(alternate_emails)), 3)
# However, so long as the primary is included, we'll just have one
all_emails = ['*****@*****.**'] + alternate_emails
self.expect_under_email('*****@*****.**', lookup=all_emails)
def test_just_one_record(self):
""" When requesting records under many emails, just one is returned.
(Provided that the primary email is included in the lookup list)
"""
person_id = self.create_tim()
alternate_emails = [f'tim@{i}.example.com' for i in range(3)]
for email in alternate_emails:
self.record_alternate_email(person_id, email)
# When we request just the alternate emails, it returns one for each
self.assertEqual(len(geardb.matching_memberships(alternate_emails)), 3)
# However, so long as the primary is included, we'll just have one
all_emails = ['*****@*****.**'] + alternate_emails
self.expect_under_email('*****@*****.**', lookup=all_emails)
def test_alternate_email(self):
""" We can look up participants by other emails. """
person_id = self.create_tim()
# First, there is no known membership for the other email
self.assertEqual(geardb.matching_memberships(['*****@*****.**']), OrderedDict())
# Then, after tying the alternate email to the main account, results!
self.record_alternate_email(person_id, '*****@*****.**')
self.expect_under_email('*****@*****.**')
# Importantly, we can still look up by the main email address!
self.expect_under_email('*****@*****.**')
# If looking up by both emails, the primary email is reported
# (Importantly, only one membership is returned)
self.expect_under_email('*****@*****.**', lookup=['*****@*****.**', '*****@*****.**'])
def test_alternate_email(self):
""" We can look up participants by other emails. """
person_id = self.create_tim()
# First, there is no known membership for the other email
self.assertEqual(geardb.matching_memberships(['*****@*****.**']),
OrderedDict())
# Then, after tying the alternate email to the main account, results!
self.record_alternate_email(person_id, '*****@*****.**')
self.expect_under_email('*****@*****.**')
# Importantly, we can still look up by the main email address!
self.expect_under_email('*****@*****.**')
# If looking up by both emails, the primary email is reported
# (Importantly, only one membership is returned)
self.expect_under_email('*****@*****.**',
lookup=['*****@*****.**', '*****@*****.**'])
def test_empty_emails(self):
""" Passing an empty list of emails will return zero matches.
There's no need to hit the database.
"""
self.assertEqual(OrderedDict(), geardb_utils.matching_memberships([]))
def expect_under_email(self, email, lookup=None):
expected = self.just_tim
expected['membership']['email'] = email
lookup_emails = lookup or [email]
results = geardb.matching_memberships(lookup_emails) # (OrderedDict)
self.assertEqual({email: expected}, dict(results))
def test_no_people_record(self):
""" Without a match, nothing is returned. """
matches = geardb.matching_memberships(['*****@*****.**'])
self.assertEqual(matches, OrderedDict())
def one_match(self, email):
matches = geardb.matching_memberships([email])
self.assertEqual(len(matches), 1)
return matches[email]
def expect_under_email(self, email, lookup=None):
expected = self.just_tim
expected['membership']['email'] = email
lookup_emails = lookup or [email]
results = geardb.matching_memberships(lookup_emails) # (OrderedDict)
self.assertEqual({email: expected}, dict(results))
def test_no_people_record(self):
""" Without a match, nothing is returned. """
matches = geardb.matching_memberships(['*****@*****.**'])
self.assertEqual(matches, OrderedDict())
def one_match(self, email):
matches = geardb.matching_memberships([email])
self.assertEqual(len(matches), 1)
return matches[email]