From 8149bc539671030c471ff11ed4a725d2d5671731 Mon Sep 17 00:00:00 2001
From: Piotr Gawron <piotr.gawron@uni.lu>
Date: Thu, 19 Aug 2021 13:11:12 +0200
Subject: [PATCH] indentation, unused variables removed
---
smash/web/management/commands/superworker.py | 16 ++++++++++------
1 file changed, 10 insertions(+), 6 deletions(-)
diff --git a/smash/web/management/commands/superworker.py b/smash/web/management/commands/superworker.py
index f34e7ebe..58b26880 100644
--- a/smash/web/management/commands/superworker.py
+++ b/smash/web/management/commands/superworker.py
@@ -1,9 +1,12 @@
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')
--
GitLab