def post(self, request): serializer = self.serializer_class(data=request.data) serializer.is_valid(raise_exception=True) dingtalk_appkey = serializer.validated_data['DINGTALK_APPKEY'] dingtalk_agentid = serializer.validated_data['DINGTALK_AGENTID'] dingtalk_appsecret = serializer.validated_data.get( 'DINGTALK_APPSECRET') if not dingtalk_appsecret: secret = Setting.objects.filter(name='DINGTALK_APPSECRET').first() if secret: dingtalk_appsecret = secret.cleaned_value dingtalk_appsecret = dingtalk_appsecret or '' try: dingtalk = DingTalk(appid=dingtalk_appkey, appsecret=dingtalk_appsecret, agentid=dingtalk_agentid) dingtalk.send_text(['test'], 'test') return Response(status=status.HTTP_200_OK, data={'msg': _('Test success')}) except APIException as e: try: error = e.detail['errmsg'] except: error = e.detail return Response(status=status.HTTP_400_BAD_REQUEST, data={'error': error})
def get(self, request: HttpRequest): code = request.GET.get('code') redirect_url = request.GET.get('redirect_url') login_url = reverse('authentication:login') if not self.verify_state(): return self.get_verify_state_failed_response(redirect_url) dingtalk = DingTalk(appid=settings.DINGTALK_APPKEY, appsecret=settings.DINGTALK_APPSECRET, agentid=settings.DINGTALK_AGENTID) userid = dingtalk.get_userid_by_code(code) if not userid: # 正常流程不会出这个错误,hack 行为 msg = _('Failed to get user from DingTalk') response = self.get_failed_reponse(login_url, title=msg, msg=msg) return response user = get_object_or_none(User, dingtalk_id=userid) if user is None: title = _('DingTalk is not bound') msg = _('Please login with a password and then bind the WeCom') response = self.get_failed_reponse(login_url, title=title, msg=msg) return response try: self.check_oauth2_auth(user, settings.AUTH_BACKEND_DINGTALK) except errors.AuthFailedError as e: self.set_login_failed_mark() msg = e.msg response = self.get_failed_reponse(login_url, title=msg, msg=msg) return response return self.redirect_to_guard_view()
def get(self, request: HttpRequest, user_id): code = request.GET.get('code') redirect_url = request.GET.get('redirect_url') if not self.verify_state(): return self.get_verify_state_failed_response(redirect_url) user = get_object_or_none(User, id=user_id) if user is None: logger.error(f'DingTalkQR bind callback error, user_id invalid: user_id={user_id}') msg = _('Invalid user_id') response = self.get_failed_reponse(redirect_url, msg, msg) return response if user.dingtalk_id: response = self.get_already_bound_response(redirect_url) return response dingtalk = DingTalk( appid=settings.DINGTALK_APPKEY, appsecret=settings.DINGTALK_APPSECRET, agentid=settings.DINGTALK_AGENTID ) userid = dingtalk.get_userid_by_code(code) if not userid: msg = _('DingTalk query user failed') response = self.get_failed_reponse(redirect_url, msg, msg) return response try: user.dingtalk_id = userid user.save() except IntegrityError as e: if e.args[0] == 1062: msg = _('The DingTalk is already bound to another user') response = self.get_failed_reponse(redirect_url, msg, msg) return response raise e msg = _('Binding DingTalk successfully') response = self.get_success_reponse(redirect_url, msg, msg) return response
class DingTalk(BackendBase): account_field = 'dingtalk_id' is_enable_field_in_settings = 'AUTH_DINGTALK' def __init__(self): self.dingtalk = Client(appid=settings.DINGTALK_APPKEY, appsecret=settings.DINGTALK_APPSECRET, agentid=settings.DINGTALK_AGENTID) def send_msg(self, users, message, subject=None): accounts, __, __ = self.get_accounts(users) return self.dingtalk.send_text(accounts, message)
def __init__(self): self.dingtalk = Client(appid=settings.DINGTALK_APPKEY, appsecret=settings.DINGTALK_APPSECRET, agentid=settings.DINGTALK_AGENTID)