Update the bootstrap Django command to support WorkerType

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')