9ed9c39ac8
rbac: fix error when looking up permissions for now uninstalled apps (#8068) Signed-off-by: Jens Langhammer <jens@goauthentik.io> Co-authored-by: Jens L <jens@goauthentik.io>
81 lines
3.1 KiB
Python
81 lines
3.1 KiB
Python
"""common RBAC serializers"""
|
|
from typing import Optional
|
|
|
|
from django.apps import apps
|
|
from django_filters.filters import NumberFilter
|
|
from django_filters.filterset import FilterSet
|
|
from guardian.models import UserObjectPermission
|
|
from guardian.shortcuts import get_objects_for_user
|
|
from rest_framework.fields import SerializerMethodField
|
|
from rest_framework.mixins import ListModelMixin
|
|
from rest_framework.viewsets import GenericViewSet
|
|
|
|
from authentik.api.pagination import SmallerPagination
|
|
from authentik.rbac.api.rbac_assigned_by_users import UserObjectPermissionSerializer
|
|
|
|
|
|
class ExtraUserObjectPermissionSerializer(UserObjectPermissionSerializer):
|
|
"""User permission with additional object-related data"""
|
|
|
|
app_label_verbose = SerializerMethodField()
|
|
model_verbose = SerializerMethodField()
|
|
|
|
object_description = SerializerMethodField()
|
|
|
|
def get_app_label_verbose(self, instance: UserObjectPermission) -> str:
|
|
"""Get app label from permission's model"""
|
|
try:
|
|
return apps.get_app_config(instance.content_type.app_label).verbose_name
|
|
except LookupError:
|
|
return instance.content_type.app_label
|
|
|
|
def get_model_verbose(self, instance: UserObjectPermission) -> str:
|
|
"""Get model label from permission's model"""
|
|
try:
|
|
return apps.get_model(
|
|
instance.content_type.app_label, instance.content_type.model
|
|
)._meta.verbose_name
|
|
except LookupError:
|
|
return f"{instance.content_type.app_label}.{instance.content_type.model}"
|
|
|
|
def get_object_description(self, instance: UserObjectPermission) -> Optional[str]:
|
|
"""Get model description from attached model. This operation takes at least
|
|
one additional query, and the description is only shown if the user/role has the
|
|
view_ permission on the object"""
|
|
app_label = instance.content_type.app_label
|
|
model = instance.content_type.model
|
|
try:
|
|
model_class = apps.get_model(app_label, model)
|
|
except LookupError:
|
|
return None
|
|
objects = get_objects_for_user(instance.user, f"{app_label}.view_{model}", model_class)
|
|
obj = objects.first()
|
|
if not obj:
|
|
return None
|
|
return str(obj)
|
|
|
|
class Meta(UserObjectPermissionSerializer.Meta):
|
|
fields = UserObjectPermissionSerializer.Meta.fields + [
|
|
"app_label_verbose",
|
|
"model_verbose",
|
|
"object_description",
|
|
]
|
|
|
|
|
|
class UserPermissionFilter(FilterSet):
|
|
"""User-assigned permission filter"""
|
|
|
|
user_id = NumberFilter("user__id", required=True)
|
|
|
|
|
|
class UserPermissionViewSet(ListModelMixin, GenericViewSet):
|
|
"""Get a users's assigned object permissions"""
|
|
|
|
serializer_class = ExtraUserObjectPermissionSerializer
|
|
ordering = ["user__username"]
|
|
pagination_class = SmallerPagination
|
|
# The filtering is done in the filterset,
|
|
# which has a required filter that does the heavy lifting
|
|
queryset = UserObjectPermission.objects.select_related("content_type", "user").all()
|
|
filterset_class = UserPermissionFilter
|