# pylint: disable=attribute-defined-outside-init,pointless-string-statement from django.db import models from django.dispatch import receiver from django.utils.encoding import python_2_unicode_compatible from misago.core import serializer from misago.core.signals import secret_key_changed from . import version as acl_version @python_2_unicode_compatible class BaseRole(models.Model): name = models.CharField(max_length=255) special_role = models.CharField(max_length=255, null=True, blank=True) pickled_permissions = models.TextField(null=True, blank=True) class Meta: abstract = True def __str__(self): return self.name def save(self, *args, **kwargs): if self.pk: acl_version.invalidate() return super(BaseRole, self).save(*args, **kwargs) def delete(self, *args, **kwargs): acl_version.invalidate() return super(BaseRole, self).delete(*args, **kwargs) @property def permissions(self): try: return self.permissions_cache except AttributeError: if self.pickled_permissions: self.permissions_cache = serializer.loads( self.pickled_permissions) else: self.permissions_cache = {} return self.permissions_cache @permissions.setter def permissions(self, permissions): self.permissions_cache = permissions self.pickled_permissions = serializer.dumps(permissions) class Role(BaseRole): pass """ Signal handlers """ @receiver(secret_key_changed) def update_roles_pickles(sender, **kwargs): for role in Role.objects.iterator(): if role.pickled_permissions: role.pickled_permissions = serializer.regenerate_checksum( role.pickled_permissions) role.save(update_fields=['pickled_permissions'])