X-Git-Url: https://git.mdrn.pl/edumed.git/blobdiff_plain/b154d06ff9d6b7fa711607bc02d03fb9db5c6d33..3ca90d4aed87ed8b39c4984a290de6a3b898a7bd:/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 239fe6b..0e51e01 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='olimpiada').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]
@@ -21,6 +25,7 @@ class Command(BaseCommand):
                     Submission.create(**args)
                     new += 1
                 else:
+                    self.stdout.write('skipping ' + student['email'] + ': already exists.')
                     skipped += 1
 
         self.stdout.write('New: ' + str(new) + ', skipped: ' + str(skipped))