332 lines
15 KiB
Python
332 lines
15 KiB
Python
from django.shortcuts import render, redirect
|
||
from django.http import JsonResponse, HttpResponse
|
||
from django.core.mail import send_mail, EmailMultiAlternatives
|
||
from django.template.loader import render_to_string
|
||
from django.utils.html import strip_tags
|
||
from django.views.decorators.csrf import csrf_exempt
|
||
from django.views.decorators.http import require_http_methods
|
||
from django.contrib import messages
|
||
from django import forms
|
||
from hertz_studio_django_captcha.captcha_generator import HertzCaptchaGenerator
|
||
import json
|
||
from django.conf import settings
|
||
import random
|
||
import string
|
||
|
||
class HertzCaptchaForm(forms.Form):
|
||
"""Hertz验证码表单"""
|
||
captcha_input = forms.CharField(
|
||
max_length=10,
|
||
widget=forms.TextInput(attrs={
|
||
'placeholder': '请输入验证码',
|
||
'class': 'form-control',
|
||
'autocomplete': 'off'
|
||
}),
|
||
label='验证码'
|
||
)
|
||
captcha_id = forms.CharField(widget=forms.HiddenInput(), required=False)
|
||
|
||
def captcha_demo(request):
|
||
"""
|
||
验证码演示页面
|
||
展示多种验证码功能的使用方法
|
||
"""
|
||
# 获取请求的验证码类型
|
||
captcha_type = request.GET.get('type', 'random_char')
|
||
|
||
# 初始化验证码生成器
|
||
captcha_generator = HertzCaptchaGenerator()
|
||
|
||
if request.method == 'POST':
|
||
# 检查是否是Ajax请求
|
||
if request.headers.get('X-Requested-With') == 'XMLHttpRequest':
|
||
try:
|
||
data = json.loads(request.body)
|
||
action = data.get('action')
|
||
|
||
if action == 'refresh':
|
||
# 刷新验证码
|
||
captcha_data = captcha_generator.generate_captcha()
|
||
return JsonResponse({
|
||
'success': True,
|
||
'data': captcha_data
|
||
})
|
||
elif action == 'verify':
|
||
# 验证验证码
|
||
captcha_id = data.get('captcha_id', '')
|
||
user_input = data.get('user_input', '')
|
||
|
||
is_valid = captcha_generator.verify_captcha(captcha_id, user_input)
|
||
|
||
if is_valid:
|
||
return JsonResponse({
|
||
'success': True,
|
||
'valid': True,
|
||
'message': f'验证成功！验证码类型: {captcha_type}'
|
||
})
|
||
else:
|
||
return JsonResponse({
|
||
'success': True,
|
||
'valid': False,
|
||
'message': '验证码错误，请重新输入'
|
||
})
|
||
except json.JSONDecodeError:
|
||
return JsonResponse({
|
||
'success': False,
|
||
'error': '请求数据格式错误'
|
||
})
|
||
else:
|
||
# 普通表单提交处理
|
||
form = HertzCaptchaForm(request.POST)
|
||
username = request.POST.get('username', '')
|
||
captcha_id = request.POST.get('captcha_id', '')
|
||
captcha_input = request.POST.get('captcha_input', '')
|
||
|
||
# 验证验证码
|
||
is_valid = captcha_generator.verify_captcha(captcha_id, captcha_input)
|
||
|
||
if is_valid and username:
|
||
# 生成新的验证码用于显示
|
||
initial_captcha = captcha_generator.generate_captcha()
|
||
return render(request, 'captcha_demo.html', {
|
||
'form': HertzCaptchaForm(),
|
||
'success_message': f'验证成功！用户名: {username}，验证码类型: {captcha_type}',
|
||
'captcha_unavailable': False,
|
||
'current_type': captcha_type,
|
||
'initial_captcha': initial_captcha,
|
||
'captcha_types': {
|
||
'random_char': '随机字符验证码',
|
||
'math': '数学运算验证码',
|
||
'word': '单词验证码'
|
||
}
|
||
})
|
||
|
||
# GET请求或表单验证失败时，生成初始验证码
|
||
form = HertzCaptchaForm()
|
||
initial_captcha = captcha_generator.generate_captcha()
|
||
|
||
return render(request, 'captcha_demo.html', {
|
||
'form': form,
|
||
'captcha_unavailable': False,
|
||
'current_type': captcha_type,
|
||
'initial_captcha': initial_captcha,
|
||
'captcha_types': {
|
||
'random_char': '随机字符验证码',
|
||
'math': '数学运算验证码',
|
||
'word': '单词验证码'
|
||
}
|
||
})
|
||
|
||
def websocket_demo(request):
|
||
"""WebSocket演示页面"""
|
||
return render(request, 'websocket_demo.html')
|
||
|
||
def websocket_test(request):
|
||
"""
|
||
WebSocket简单测试页面
|
||
"""
|
||
return render(request, 'websocket_test.html')
|
||
|
||
# 测试热重启功能 - 添加注释触发文件变化
|
||
|
||
def email_demo(request):
|
||
"""邮件系统演示页面"""
|
||
if request.method == 'GET':
|
||
return render(request, 'email_demo.html')
|
||
|
||
elif request.method == 'POST':
|
||
try:
|
||
# 获取表单数据
|
||
email_type = request.POST.get('email_type', 'welcome')
|
||
recipient_email = request.POST.get('recipient_email')
|
||
recipient_name = request.POST.get('recipient_name', '用户')
|
||
custom_subject = request.POST.get('subject', '')
|
||
custom_message = request.POST.get('message', '')
|
||
|
||
if not recipient_email:
|
||
return JsonResponse({
|
||
'success': False,
|
||
'message': '请输入收件人邮箱地址'
|
||
})
|
||
|
||
# 根据邮件类型生成内容
|
||
email_content = generate_email_content(email_type, recipient_name, custom_subject, custom_message)
|
||
|
||
# 发送邮件
|
||
success = send_demo_email(
|
||
recipient_email=recipient_email,
|
||
subject=email_content['subject'],
|
||
html_content=email_content['html_content'],
|
||
text_content=email_content['text_content']
|
||
)
|
||
|
||
if success:
|
||
return JsonResponse({
|
||
'success': True,
|
||
'message': f'邮件已成功发送到 {recipient_email}'
|
||
})
|
||
else:
|
||
return JsonResponse({
|
||
'success': False,
|
||
'message': '邮件发送失败，请检查邮件配置'
|
||
})
|
||
|
||
except Exception as e:
|
||
return JsonResponse({
|
||
'success': False,
|
||
'message': f'发送失败：{str(e)}'
|
||
})
|
||
|
||
def generate_email_content(email_type, recipient_name, custom_subject='', custom_message=''):
|
||
"""根据邮件类型生成邮件内容"""
|
||
|
||
email_templates = {
|
||
'welcome': {
|
||
'subject': '🎉 欢迎加入 Hertz Server Django！',
|
||
'html_template': f'''
|
||
<html>
|
||
<body style="font-family: Arial, sans-serif; line-height: 1.6; color: #333;">
|
||
<div style="max-width: 600px; margin: 0 auto; padding: 20px;">
|
||
<div style="background: linear-gradient(135deg, #667eea 0%, #764ba2 100%); color: white; padding: 30px; text-align: center; border-radius: 10px 10px 0 0;">
|
||
<h1 style="margin: 0; font-size: 28px;">🎉 欢迎加入我们！</h1>
|
||
</div>
|
||
<div style="background: white; padding: 30px; border: 1px solid #e1e5e9; border-radius: 0 0 10px 10px;">
|
||
<p style="font-size: 16px;">亲爱的 <strong>{recipient_name}</strong>，</p>
|
||
<p>欢迎您注册成为我们的用户！我们很高兴您能加入我们的大家庭。</p>
|
||
<p>在这里，您可以享受到：</p>
|
||
<ul style="color: #666;">
|
||
<li>🔐 安全的验证码系统</li>
|
||
<li>🌐 实时WebSocket通信</li>
|
||
<li>📧 完善的邮件服务</li>
|
||
<li>📚 详细的API文档</li>
|
||
</ul>
|
||
<p>如果您有任何问题，请随时联系我们。</p>
|
||
<div style="text-align: center; margin: 30px 0;">
|
||
<a href="http://127.0.0.1:8000/" style="background: #667eea; color: white; padding: 12px 30px; text-decoration: none; border-radius: 5px; display: inline-block;">开始使用</a>
|
||
</div>
|
||
<p>祝您使用愉快！</p>
|
||
<hr style="border: none; border-top: 1px solid #eee; margin: 20px 0;">
|
||
<p style="color: #666; font-size: 14px;">此致<br><strong>Hertz Server Django 团队</strong></p>
|
||
</div>
|
||
</div>
|
||
</body>
|
||
</html>
|
||
'''
|
||
},
|
||
'notification': {
|
||
'subject': '🔔 系统通知 - Hertz Server Django',
|
||
'html_template': f'''
|
||
<html>
|
||
<body style="font-family: Arial, sans-serif; line-height: 1.6; color: #333;">
|
||
<div style="max-width: 600px; margin: 0 auto; padding: 20px;">
|
||
<div style="background: #007bff; color: white; padding: 20px; text-align: center; border-radius: 10px 10px 0 0;">
|
||
<h1 style="margin: 0; font-size: 24px;">🔔 系统通知</h1>
|
||
</div>
|
||
<div style="background: white; padding: 30px; border: 1px solid #e1e5e9; border-radius: 0 0 10px 10px;">
|
||
<p style="font-size: 16px;">亲爱的 <strong>{recipient_name}</strong>，</p>
|
||
<p>您有一条新的系统通知：</p>
|
||
<div style="background: #f8f9fa; padding: 20px; border-left: 4px solid #007bff; margin: 20px 0;">
|
||
<p style="margin: 0; font-weight: 500;">您的账户设置已更新，如果这不是您的操作，请立即联系我们。</p>
|
||
</div>
|
||
<p>系统会持续为您提供安全保障，如有疑问请联系客服。</p>
|
||
<div style="text-align: center; margin: 30px 0;">
|
||
<a href="http://127.0.0.1:8000/" style="background: #007bff; color: white; padding: 12px 30px; text-decoration: none; border-radius: 5px; display: inline-block;">查看详情</a>
|
||
</div>
|
||
<hr style="border: none; border-top: 1px solid #eee; margin: 20px 0;">
|
||
<p style="color: #666; font-size: 14px;">此致<br><strong>Hertz Server Django 团队</strong></p>
|
||
</div>
|
||
</div>
|
||
</body>
|
||
</html>
|
||
'''
|
||
},
|
||
'verification': {
|
||
'subject': '🔐 邮箱验证 - Hertz Server Django',
|
||
'html_template': f'''
|
||
<html>
|
||
<body style="font-family: Arial, sans-serif; line-height: 1.6; color: #333;">
|
||
<div style="max-width: 600px; margin: 0 auto; padding: 20px;">
|
||
<div style="background: #28a745; color: white; padding: 20px; text-align: center; border-radius: 10px 10px 0 0;">
|
||
<h1 style="margin: 0; font-size: 24px;">🔐 邮箱验证</h1>
|
||
</div>
|
||
<div style="background: white; padding: 30px; border: 1px solid #e1e5e9; border-radius: 0 0 10px 10px;">
|
||
<p style="font-size: 16px;">亲爱的 <strong>{recipient_name}</strong>，</p>
|
||
<p>感谢您注册 Hertz Server Django！请点击下面的按钮验证您的邮箱地址：</p>
|
||
<div style="text-align: center; margin: 40px 0;">
|
||
<a href="http://127.0.0.1:8000/verify?token=demo_token" style="background: #28a745; color: white; padding: 15px 40px; text-decoration: none; border-radius: 5px; display: inline-block; font-size: 16px; font-weight: 500;">验证邮箱地址</a>
|
||
</div>
|
||
<p style="color: #666; font-size: 14px;">如果按钮无法点击，请复制以下链接到浏览器：<br>
|
||
<code style="background: #f8f9fa; padding: 5px; border-radius: 3px;">http://127.0.0.1:8000/verify?token=demo_token</code></p>
|
||
<p style="color: #666;">如果您没有注册账户，请忽略此邮件。此验证链接将在24小时后失效。</p>
|
||
<hr style="border: none; border-top: 1px solid #eee; margin: 20px 0;">
|
||
<p style="color: #666; font-size: 14px;">此致<br><strong>Hertz Server Django 团队</strong></p>
|
||
</div>
|
||
</div>
|
||
</body>
|
||
</html>
|
||
'''
|
||
},
|
||
'custom': {
|
||
'subject': custom_subject or '自定义邮件 - Hertz Server Django',
|
||
'html_template': f'''
|
||
<html>
|
||
<body style="font-family: Arial, sans-serif; line-height: 1.6; color: #333;">
|
||
<div style="max-width: 600px; margin: 0 auto; padding: 20px;">
|
||
<div style="background: linear-gradient(135deg, #667eea 0%, #764ba2 100%); color: white; padding: 20px; text-align: center; border-radius: 10px 10px 0 0;">
|
||
<h1 style="margin: 0; font-size: 24px;">{custom_subject or '自定义邮件'}</h1>
|
||
</div>
|
||
<div style="background: white; padding: 30px; border: 1px solid #e1e5e9; border-radius: 0 0 10px 10px;">
|
||
<p style="font-size: 16px;">亲爱的 <strong>{recipient_name}</strong>，</p>
|
||
<div style="margin: 20px 0; font-size: 16px;">
|
||
{custom_message.replace(chr(10), '<br>') if custom_message else '这是一封自定义邮件。'}
|
||
</div>
|
||
<hr style="border: none; border-top: 1px solid #eee; margin: 20px 0;">
|
||
<p style="color: #666; font-size: 14px;">此致<br><strong>Hertz Server Django 团队</strong></p>
|
||
</div>
|
||
</div>
|
||
</body>
|
||
</html>
|
||
'''
|
||
}
|
||
}
|
||
|
||
template = email_templates.get(email_type, email_templates['welcome'])
|
||
html_content = template['html_template']
|
||
text_content = strip_tags(html_content)
|
||
|
||
return {
|
||
'subject': template['subject'],
|
||
'html_content': html_content,
|
||
'text_content': text_content
|
||
}
|
||
|
||
def send_demo_email(recipient_email, subject, html_content, text_content):
|
||
"""发送演示邮件"""
|
||
try:
|
||
# 检查邮件配置
|
||
if not settings.EMAIL_HOST_USER or not settings.EMAIL_HOST_PASSWORD:
|
||
print("邮件配置不完整，使用控制台输出模式")
|
||
print(f"收件人: {recipient_email}")
|
||
print(f"主题: {subject}")
|
||
print(f"内容: {text_content[:200]}...")
|
||
return True
|
||
|
||
# 创建邮件
|
||
email = EmailMultiAlternatives(
|
||
subject=subject,
|
||
body=text_content,
|
||
from_email=settings.DEFAULT_FROM_EMAIL,
|
||
to=[recipient_email]
|
||
)
|
||
|
||
# 添加HTML内容
|
||
email.attach_alternative(html_content, "text/html")
|
||
|
||
# 发送邮件
|
||
email.send()
|
||
return True
|
||
|
||
except Exception as e:
|
||
print(f"邮件发送失败: {str(e)}")
|
||
return False
|