from django.core.management.base import BaseCommand
from django.contrib.auth.models import Group

from emr.permissions import ROLE_ADMIN, ROLE_DOCTOR, ROLE_NURSE

class Command(BaseCommand):
    help = "Create default EMR roles (Doctor, Nurse, Admin) as Django Groups."

    def handle(self, *args, **options):
        for name in (ROLE_ADMIN, ROLE_DOCTOR, ROLE_NURSE):
            Group.objects.get_or_create(name=name)
        self.stdout.write(self.style.SUCCESS("EMR roles ensured: Admin, Doctor, Nurse"))