fix: swagger

This commit is contained in:
xinwen
2022-02-21 18:51:11 +08:00
committed by Jiangjie.Bai
parent ad2cb233d7
commit f460916e84
8 changed files with 24 additions and 5 deletions

View File

@@ -25,6 +25,9 @@ class RoleAdminMixin:
@lazyproperty
def user(self):
user_id = self.kwargs.get(self.user_id_url_kwarg)
if hasattr(self, 'swagger_fake_view') and not user_id:
return self.request.user # NOQA
user_model = get_user_model()
return user_model.objects.get(id=user_id)