from django.core.management.base import BaseCommand

from authentication.models.roles import Roles


class Command(BaseCommand):
    help = "Populate database with roles"

    def handle(self, *args, **options):
        Roles.objects.create(pk=1, description="Administrador")
        Roles.objects.create(pk=2, description="Dependencia")
        self.stdout.write(self.style.SUCCESS("Data successfully seeded"))