Terralego/django-terra-accounts

from django.contrib.auth.base_user import BaseUserManager
from django.contrib.contenttypes.models import ContentType
from django.db.models import Manager, ObjectDoesNotExist
from django.conf import settings


class TerraUserManager(BaseUserManager):

    def _create_user(self, email, password, **extra_fields):
        email = self.normalize_email(email)
        user = self.model(email=email, **extra_fields)
        user.set_password(password)
        user.save(using=self._db)
        return user

    def create_user(self, email, password, **extra_fields):
        extra_fields.setdefault('is_staff', False)
        extra_fields.setdefault('is_superuser', False)
        return self._create_user(email, password, **extra_fields)

    def create_superuser(self, email, password, **extra_fields):
        extra_fields.setdefault('is_staff', True)
        extra_fields.setdefault('is_superuser', True)

        return self._create_user(email, password, **extra_fields)


class ReadModelManager(Manager):
    def get_user_read(self, user, obj):
        contenttype = ContentType.objects.get_for_model(obj.__class__)
        try:
            return self.get(
                user=user,
                contenttype=contenttype,
                identifier=obj.pk)
        except ObjectDoesNotExist:
            return None

    def read_object(self, user, obj):
        read = self.get_user_read(user, obj)
        if read:
            read.read_instance()
        else:
            read = self.create(
                user=user,
                model_object=obj
            )

        return read


class TerraPermissionManager(Manager):
    def get_queryset(self):
        disabled_modules = settings.TERRA_APPLIANCE_SETTINGS.get('disabled_modules', [])
        return super().get_queryset().exclude(module__in=disabled_modules)