mirror of https://github.com/doccano/doccano.git
pythonannotation-tooldatasetsactive-learningtext-annotationdatasetnatural-language-processingdata-labelingmachine-learning
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
23 lines
905 B
23 lines
905 B
from api.models import Role
|
|
from django.core.management.base import BaseCommand
|
|
from django.db import DatabaseError
|
|
from django.conf import settings
|
|
|
|
|
|
class Command(BaseCommand):
|
|
help = 'Non-interactively create default roles'
|
|
|
|
def handle(self, *args, **options):
|
|
try:
|
|
role_names = [settings.ROLE_PROJECT_ADMIN, settings.ROLE_ANNOTATOR, settings.ROLE_ANNOTATION_APPROVER]
|
|
except KeyError as key_error:
|
|
self.stderr.write(self.style.ERROR(f'Missing Key: "{key_error}"'))
|
|
for role_name in role_names:
|
|
role = Role()
|
|
role.name = role_name
|
|
try:
|
|
role.save()
|
|
except DatabaseError as db_error:
|
|
self.stderr.write(self.style.ERROR(f'Datbase Error: "{db_error}"'))
|
|
else:
|
|
self.stdout.write(self.style.SUCCESS(f'Role created successfully "{role_name}"'))
|