mirror of https://github.com/jumpserver/jumpserver
19 lines
485 B
Python
19 lines
485 B
Python
from rest_framework.request import Request
|
|
|
|
from users.models import User
|
|
from common.http import is_true
|
|
|
|
from perms.utils.user_permission import UserPermTreeRefreshUtil
|
|
|
|
|
|
__all__ = ['RebuildTreeMixin']
|
|
|
|
|
|
class RebuildTreeMixin:
|
|
user: User
|
|
|
|
def get(self, request: Request, *args, **kwargs):
|
|
force = is_true(request.query_params.get('rebuild_tree'))
|
|
UserPermTreeRefreshUtil(self.user).refresh_if_need(force)
|
|
return super().get(request, *args, **kwargs)
|