mirror of https://github.com/jumpserver/jumpserver
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
34 lines
929 B
34 lines
929 B
from rest_framework.views import APIView |
|
from rest_framework.request import Request |
|
from rest_framework.response import Response |
|
|
|
from users.models import User |
|
from common.utils import get_logger |
|
from common.permissions import UserConfirmation |
|
from common.mixins.api import RoleUserMixin, RoleAdminMixin |
|
from authentication.const import ConfirmType |
|
from authentication import errors |
|
|
|
logger = get_logger(__file__) |
|
|
|
|
|
class WeComQRUnBindBase(APIView): |
|
user: User |
|
|
|
def post(self, request: Request, **kwargs): |
|
user = self.user |
|
|
|
if not user.wecom_id: |
|
raise errors.WeComNotBound |
|
|
|
user.wecom_id = None |
|
user.save() |
|
return Response() |
|
|
|
|
|
class WeComQRUnBindForUserApi(RoleUserMixin, WeComQRUnBindBase): |
|
permission_classes = (UserConfirmation.require(ConfirmType.ReLogin),) |
|
|
|
|
|
class WeComQRUnBindForAdminApi(RoleAdminMixin, WeComQRUnBindBase): |
|
user_id_url_kwarg = 'user_id'
|
|
|