X-Git-Url: https://git.mdrn.pl/edumed.git/blobdiff_plain/93b539c6a58516dbb4eb3aa1b3405121e8023fc7..b154d06ff9d6b7fa711607bc02d03fb9db5c6d33:/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
new file mode 100644
index 0000000..239fe6b
--- /dev/null
+++ b/wtem/management/commands/wtem_generate_keys.py
@@ -0,0 +1,26 @@
+from django.core.management.base import BaseCommand, CommandError
+
+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):
+        new = 0
+        skipped = 0
+
+        for wtem_contact in Contact.objects.filter(form_tag = 'wtem'):
+            for student in wtem_contact.body['student']:
+                if not Submission.objects.filter(email = student['email']).exists():
+                    args = dict()
+                    for attr in ['first_name', 'last_name', 'email']:
+                        args[attr] = student[attr]
+                    args['contact'] = wtem_contact
+                    Submission.create(**args)
+                    new += 1
+                else:
+                    skipped += 1
+
+        self.stdout.write('New: ' + str(new) + ', skipped: ' + str(skipped))