From 4d2b354bc00e635f72db0a7c689ee0eee99ff94e Mon Sep 17 00:00:00 2001 From: Ramon Wenger Date: Wed, 4 Sep 2019 16:08:41 +0200 Subject: [PATCH] Add command for adding teachers --- .../management/commands/create_teacher.py | 40 +++++++++++++++++++ 1 file changed, 40 insertions(+) create mode 100644 server/core/management/commands/create_teacher.py diff --git a/server/core/management/commands/create_teacher.py b/server/core/management/commands/create_teacher.py new file mode 100644 index 00000000..b7cd2471 --- /dev/null +++ b/server/core/management/commands/create_teacher.py @@ -0,0 +1,40 @@ +import csv + +from django.core.management import BaseCommand +import os +from django.conf import settings + +from users.models import User, SchoolClass, Role, UserRole + + +class Command(BaseCommand): + def add_arguments(self, parser): + parser.add_argument('first_name') + parser.add_argument('last_name') + parser.add_argument('email') + + def handle(self, *args, **options): + first_name = options['first_name'] + last_name = options['last_name'] + email = options['email'] + self.stdout.write('Creating user {} {} with email {}'.format(first_name, last_name, email)) + try: + school_class_name = 'Klasse {} {}'.format(first_name, last_name) + + user, created = User.objects.get_or_create(email=email, username=email) + user.first_name = first_name + user.last_name = last_name + user.set_password(User.objects.make_random_password()) + user.save() + + self.stdout.write("Assigning teacher role") + teacher = Role.objects.get(key='teacher') + UserRole.objects.get_or_create(user=user, role=teacher) + + school, _ = SchoolClass.objects.get_or_create(name=school_class_name) + user.school_classes.add(school) + + self.stdout.write("") + + except Exception as e: + self.stdout.write(e)