Misago/misago/acl/forms.py

from django.utils.translation import ugettext_lazy as _

from misago.core import forms

from misago.acl.models import Role
from misago.acl.providers import providers


class RoleForm(forms.ModelForm):
    name = forms.CharField(label=_("Role name"))

    class Meta:
        model = Role
        fields = ['name']


def get_permissions_forms(role, data=None):
    """
    Utility function for building forms in admin
    """
    role_permissions = role.permissions

    forms = []
    for extension, module in providers.list():
        try:
            module.change_permissions_form
        except AttributeError:
            message = "'%s' object has no attribute '%s'"
            raise AttributeError(
                message % (extension, 'change_permissions_form'))

        FormType = module.change_permissions_form(role)

        if FormType:
            if data:
                forms.append(FormType(data, prefix=extension))
            else:
                forms.append(FormType(initial=role_permissions.get(extension),
                                      prefix=extension))

    return forms