from django.core.management.base import BaseCommand
from django.contrib.auth.models import Group, Permission
from django.contrib.contenttypes.models import ContentType

class Command(BaseCommand):
    help = 'Set up customer groups and permissions'

    def handle(self, *args, **options):
        # Create Customer group
        customer_group, created = Group.objects.get_or_create(name='Customer')
        
        if created:
            self.stdout.write(
                self.style.SUCCESS('Created Customer group')
            )
        else:
            self.stdout.write(
                self.style.WARNING('Customer group already exists')
            )
        
        # You can add specific permissions for customers here if needed
        # For example, permissions related to their own orders, profiles, etc.
        
        self.stdout.write(
            self.style.SUCCESS('Customer group setup completed')
        )