X-Git-Url: https://git.mdrn.pl/edumed.git/blobdiff_plain/118fd7ac41a586001b4c8f45c149731cc9a2e586..602e4759c725ce5ba0eb37d27852d7a3bf7b48e4:/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 8a14ff9..19b9582 100644
--- a/wtem/management/commands/wtem_generate_keys.py
+++ b/wtem/management/commands/wtem_generate_keys.py
@@ -6,11 +6,15 @@ 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').order_by('-created_at'):
+        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():
                     args = dict()