X-Git-Url: https://git.mdrn.pl/edumed.git/blobdiff_plain/61bff0638c1eba23b9bee153eafa1d9c05749be5..f4cb3b0ea0d7be31ee483dbb0e4486ea85437fad:/wtem/management/commands/wtem_generate_keys.py diff --git a/wtem/management/commands/wtem_generate_keys.py b/wtem/management/commands/wtem_generate_keys.py index 75847ae..f49c8b7 100644 --- a/wtem/management/commands/wtem_generate_keys.py +++ b/wtem/management/commands/wtem_generate_keys.py @@ -1,19 +1,23 @@ -from django.core.management.base import BaseCommand, CommandError +# -*- coding: utf-8 -*- +from django.core.management.base import BaseCommand from contact.models import Contact from wtem.models import Submission class Command(BaseCommand): - help = 'Sends personalized links to WTEM contestants' - def handle(self, *args, **options): + def handle(self, *ids, **options): new = 0 skipped = 0 - for wtem_contact in Contact.objects.filter(form_tag = 'wtem'): + query = Contact.objects.filter(form_tag='wtem').order_by('-created_at') + if ids: + query = query.filter(pk__in=ids) + + for wtem_contact in query: for student in wtem_contact.body['student']: - if not Submission.objects.filter(email = student['email']).exists(): + if not Submission.objects.filter(email=student['email']).exists(): args = dict() for attr in ['first_name', 'last_name', 'email']: args[attr] = student[attr]