コード例 #1
0
ファイル: setting.py プロジェクト: gaotuan/SQL-Manager
    def put(self, request, args: str = None):

        try:
            if args == '1':  # ldap测试
                ldap = json.loads(request.data['ldap'])
                ldap_test = util.test_auth(
                    username=ldap['user'],
                    password=ldap['password'],
                    host=ldap['host'],
                    type=ldap['type'],
                    sc=ldap['sc'],
                    domain=ldap['domain'])
                if ldap_test:
                    return Response('ldap连接成功!')
                else:
                    return Response('ldap连接失败!')
            elif args == '2':
                ding = request.data['ding']
                util.dingding('SQL审核平台测试', ding)
                return Response('已发送测试消息,请在钉钉中查看')
            elif args == '4':
                # setting = globalpermissions.objects.filter(authorization='global').values('message').first()
                # user = {'mail':dict(setting.get('message')).get('fs_mail_test')}
                user = {'mail':request.data['mail']}
                if fs_send_msg('飞书消息 测试成功!',user):
                    return Response('已发送测试消息,请在飞书中查看')
                else:
                    return Response('飞书测试消息,发送失败!')
            else:
                mail = json.loads(request.data['mail'])
                import smtplib
                from email.utils import parseaddr, formataddr
                from email.mime.text import MIMEText
                from email.header import Header

                def _format_addr(s):
                    name, addr = parseaddr(s)
                    return formataddr((Header(name, 'utf-8').encode(), addr))

                msg = MIMEText('Yearning test Message!', 'plain', 'utf-8')
                msg['From'] = _format_addr('Yearning_Admin <%s>' % mail['user'])
                msg['Subject'] = Header('Yearning 消息推送测试', 'utf-8').encode()
                server = smtplib.SMTP(mail['smtp_host'], mail['smtp_port'])  # SMTP协议默认端口是25
                server.set_debuglevel(1)
                server.login(mail['user'], mail['password'])
                server.sendmail(mail['user'], [mail['to_user']], msg.as_string())
                server.quit()
                return Response('已发送测试邮件,请注意查收!')
        except Exception as e:
            CUSTOM_ERROR.error(f'{e.__class__.__name__}: {e}')
            return HttpResponse(e)
コード例 #2
0
ファイル: setting.py プロジェクト: yanzongzhen/SQLAUDITS
    def put(self, request, args: str = None):

        try:
            if args == '1':  # ldap测试
                ldap = json.loads(request.data['ldap'])
                ldap_test = util.test_auth(url=ldap['url'],
                                           user=ldap['user'],
                                           password=ldap['password'])
                if ldap_test:
                    return Response('ldap连接成功!')
                else:
                    return Response('ldap连接失败!')
            elif args == '2':
                ding = request.data['ding']
                util.dingding('yearning webhook测试', ding)
                return Response('已发送测试消息,请在钉钉中查看')

            else:
                mail = json.loads(request.data['mail'])
                import smtplib
                from email.utils import parseaddr, formataddr
                from email.mime.text import MIMEText
                from email.header import Header

                def _format_addr(s):
                    name, addr = parseaddr(s)
                    return formataddr((Header(name, 'utf-8').encode(), addr))

                msg = MIMEText('SQLAudit Plaform test Message!', 'plain',
                               'utf-8')
                msg['From'] = _format_addr('SQLAudit_Admin <%s>' %
                                           mail['user'])
                msg['Subject'] = Header('SQLAudit 消息推送测试', 'utf-8').encode()
                msg['Cc'] = mail['user']
                if mail['ssl']:
                    server = smtplib.SMTP_SSL(
                        mail['smtp_host'], mail['smtp_port'])  # SMTP协议默认端口是25
                else:
                    server = smtplib.SMTP(mail['smtp_host'],
                                          mail['smtp_port'])  # SMTP协议默认端口是25
                server.set_debuglevel(1)
                server.login(mail['user'], mail['password'])
                server.sendmail(mail['user'], [mail['user'], mail['to_user']],
                                msg.as_string())
                server.quit()
                return Response('已发送测试邮件,请注意查收!')
        except Exception as e:
            CUSTOM_ERROR.error(f'{e.__class__.__name__}: {e}')
            return HttpResponse(e)
コード例 #3
0
ファイル: setting.py プロジェクト: JesseBaomihua/Yearning
    def put(self, request, args: str = None):

        try:
            if args == '1':  # ldap测试
                ldap = json.loads(request.data['ldap'])
                ldap_test = util.test_auth(
                    username=ldap['user'],
                    password=ldap['password'],
                    host=ldap['host'],
                    type=ldap['type'],
                    sc=ldap['sc'],
                    domain=ldap['domain'],
                    ou=ldap['ou'])
                if ldap_test:
                    return Response('ldap连接成功!')
                else:
                    return Response('ldap连接失败!')
            elif args == '2':
                ding = request.data['ding']
                util.dingding('yearning webhook测试', ding)
                return Response('已发送测试消息,请在钉钉中查看')

            else:
                mail = json.loads(request.data['mail'])
                import smtplib
                from email.utils import parseaddr, formataddr
                from email.mime.text import MIMEText
                from email.header import Header

                def _format_addr(s):
                    name, addr = parseaddr(s)
                    return formataddr((Header(name, 'utf-8').encode(), addr))
                msg = MIMEText('Yearning test Message!', 'plain', 'utf-8')
                msg['From'] = _format_addr('Yearning_Admin <%s>' % mail['user'])
                msg['Subject'] = Header('Yearning 消息推送测试', 'utf-8').encode()
                if mail['ssl']:
                    server = smtplib.SMTP_SSL(mail['smtp_host'], mail['smtp_port'])  # SMTP协议默认端口是25
                else:
                    server = smtplib.SMTP(mail['smtp_host'], mail['smtp_port'])  # SMTP协议默认端口是25
                server.set_debuglevel(1)
                server.login(mail['user'], mail['password'])
                server.sendmail(mail['user'], [mail['to_user']], msg.as_string())
                server.quit()
                return Response('已发送测试邮件,请注意查收!')
        except Exception as e:
            CUSTOM_ERROR.error(f'{e.__class__.__name__}: {e}')
            return HttpResponse(e)