Skip to content
Snippets Groups Projects
Commit 8149bc53 authored by Piotr Gawron's avatar Piotr Gawron
Browse files

indentation, unused variables removed

parent bef9abe5
No related branches found
No related tags found
1 merge request!336Pylint checks
from getpass import getpass
from django.core.management.base import BaseCommand
from django.db import IntegrityError
from ...models import Worker, User, Location, Language, WorkerStudyRole
from web.models.worker_study_role import ROLE_CHOICES_TECHNICIAN
from web.models.constants import GLOBAL_STUDY_ID
from web.models.worker_study_role import ROLE_CHOICES_TECHNICIAN
from ...models import Worker, User, Location, Language, WorkerStudyRole
class Command(BaseCommand):
help = 'creates super worker (superuser + worker)'
......@@ -13,7 +16,7 @@ class Command(BaseCommand):
parser.add_argument('-e', '--email', type=str, required=True)
parser.add_argument('-f', '--first-name', type=str, required=True)
parser.add_argument('-l', '--last-name', type=str, required=True)
def handle(self, *args, **kwargs):
first_name = kwargs['first_name']
last_name = kwargs['last_name']
......@@ -40,10 +43,11 @@ class Command(BaseCommand):
languages = Language.objects.all()
worker.languages.set(languages)
worker.save()
workerStudyRole, _ = WorkerStudyRole.objects.update_or_create(worker=worker,
study_id=GLOBAL_STUDY_ID, name=ROLE_CHOICES_TECHNICIAN)
WorkerStudyRole.objects.update_or_create(worker=worker,
study_id=GLOBAL_STUDY_ID,
name=ROLE_CHOICES_TECHNICIAN)
except IntegrityError:
self.stderr.write('Worker already exists')
return
self.stderr.write(f'Superworker {username} created')
\ No newline at end of file
self.stderr.write(f'Superworker {username} created')
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment