forms.py 1.2 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546
  1. from django import forms
  2. from django.utils.translation import ugettext_lazy as _
  3. from .models import Role
  4. from .providers import providers
  5. class RoleForm(forms.ModelForm):
  6. name = forms.CharField(label=_("Role name"))
  7. class Meta:
  8. model = Role
  9. fields = ['name']
  10. def get_permissions_forms(role, data=None):
  11. """
  12. Utility function for building forms in admin
  13. """
  14. role_permissions = role.permissions
  15. perms_forms = []
  16. for extension, module in providers.list():
  17. try:
  18. module.change_permissions_form
  19. except AttributeError:
  20. message = "'%s' object has no attribute '%s'"
  21. raise AttributeError(message % (extension, 'change_permissions_form'))
  22. FormType = module.change_permissions_form(role)
  23. if FormType:
  24. if data:
  25. perms_forms.append(FormType(
  26. data,
  27. prefix=extension,
  28. ))
  29. else:
  30. perms_forms.append(
  31. FormType(
  32. initial=role_permissions.get(extension),
  33. prefix=extension,
  34. )
  35. )
  36. return perms_forms