Simplifies Role Based Access Control in django-rest-framework.
You have more than one type of user in your data model and you have business logic that diverges depending on the type of user. You do not want to organize your API by role because that is not very RESTful. You do not want to manually type out a lot of conditional branching around user roles.
- You must have one Group for each role
- A User cannot belong to more than one of the Groups corresponding to each role
$ pip install django-rest-framework-rolesVIEWSET_METHOD_REGISTRYA tuple of DRF methods to override. Defaults to:
(
"get_queryset",
"get_serializer_class",
"perform_create",
"perform_update",
"perform_destroy",
)ROLE_GROUPSA tuple of Group names that correspond 1-to-1 with user roles. Defaults to:
[group.name.lower() for group in Group.objects.all()]It's recommended to define ROLE_GROUPS in settings to avoid
a database lookup on every request.
Add the mixin to any ViewSet:
from drf_roles.mixins import RoleViewSetMixin
class MyViewSet(RoleViewSetMixin, ModelViewSet):
# ...For each of the methods specified in VIEWSET_METHOD_REGISTRY a
role-scoped method will be generated on your ViewSet.
For example, let’s say you have three groups named Takers, Leavers &
Gods. Let’s also say you included "get_queryset" in the
VIEWSET_METHOD_REGISTRY.
When a Taker user hits an endpont on the ViewSet, the call to
get_queryset will be rerouted to a call to
get_queryset_for_takers.
When a Leaver user hits an endpont on the ViewSet, the call to
get_queryset will be rerouted to a call to
get_queryset_for_leavers.
When a God user hits an endpont on the ViewSet, the call to
get_queryset will be rerouted to a call to
get_queryset_for_gods.
You can implement each of these methods on your ViewSet to return a different queryset for each type of user.
You can also not implement one or more of these methods, in which case
the default call will be executed. For example, with our same set of
groups and with "get_serializer_class" included in the role
registry, let’s say you did not implement
get_serializer_class_for_takers. When a Taker user hits an
endpoint on the ViewSet, the default implementation of
get_serializer_class will be executed and return
serializer_class.
In this case, you would want to be sure that you have a
serializer_class defined on your ViewSet! Otherwise Django REST
Framework will complain. It is a good idea to always define a default
queryset and serializer_class with least privilege (e.g:
Model.objects.none()).
- Some projects require even further parameterization. For example, you may need to use a different serializer_class depending on the user's role and the request method.
- There may be a more pleasant way to express the parameterization in code. For example, it may be more pleasing to use nested classes instead of renaming the methods.