diff --git a/arkindex/documents/management/commands/bootstrap.py b/arkindex/documents/management/commands/bootstrap.py index 1fc4a6878644386da017ced2bd37f690bb3eba8a..ff620514061f5ff8dac3db315da3cf9a7e162821 100644 --- a/arkindex/documents/management/commands/bootstrap.py +++ b/arkindex/documents/management/commands/bootstrap.py @@ -8,7 +8,14 @@ from django.db.models import Q from django.db.utils import IntegrityError from rest_framework.authtoken.models import Token -from arkindex.dataimport.models import Repository, Worker, WorkerVersion, WorkerVersionGPUUsage, WorkerVersionState +from arkindex.dataimport.models import ( + Repository, + Worker, + WorkerType, + WorkerVersion, + WorkerVersionGPUUsage, + WorkerVersionState, +) from arkindex.images.models import ImageServer from arkindex.users.models import User from ponos.models import Farm @@ -171,10 +178,14 @@ class Command(BaseCommand): else: self.success(f'Repository for {IMPORT_WORKER_REPO} exists') + worker_type, _ = WorkerType.objects.get_or_create( + slug='import', + defaults={'display_name': 'Import'} + ) worker, created = Worker.objects.update_or_create( repository=repo, slug=IMPORT_WORKER_SLUG, - defaults={'name': 'File import', 'type': 'import'}, + defaults={'name': 'File import', 'type': worker_type}, ) if created: self.success(f'Created {IMPORT_WORKER_SLUG} worker')