X-Git-Url: https://git.mdrn.pl/edumed.git/blobdiff_plain/ecf1bb43a75eeacf9dbec78fd6f5a1d3d526dbd8..d93fe7b28289de7d67080cdc635f025e077449bd:/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 0fc537b..f49c8b7 100644 --- a/wtem/management/commands/wtem_generate_keys.py +++ b/wtem/management/commands/wtem_generate_keys.py @@ -1,18 +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): - - 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]