feat: 添加PDF报告邮件发送功能

- 在core/tasks.py中添加celery_send_pdf_report_email任务,用于生成PDF报告并发送邮件
- 在test_redis_celery.py中添加对应的测试用例test_celery_redis_pdf_email
- 更新main函数中的测试计数和结果输出逻辑
This commit is contained in:
2026-01-19 21:36:23 +08:00
parent e2f389a325
commit 83bbcd8ff7
2 changed files with 273 additions and 10 deletions

View File

@@ -353,6 +353,65 @@ def test_celery_redis_email():
return False
def test_celery_redis_pdf_email():
"""测试通过Celery和Redis发送包含PDF附件的邮件功能"""
logger.info("开始测试通过Celery和Redis发送包含PDF附件的邮件...")
try:
from core.tasks import celery_send_pdf_report_email
# 发送PDF报告邮件任务
logger.info("发送Celery PDF报告邮件测试任务到Redis队列...")
task = celery_send_pdf_report_email.delay()
logger.info(f"PDF报告邮件测试任务已发送任务ID: {task.id}")
# 等待任务完成
logger.info("等待任务执行最多60秒...")
result = task.get(timeout=60)
logger.info(f"PDF报告邮件任务执行结果: {result}")
# 检查结果状态
if result.get('status') == 'success':
logger.success("通过Celery和Redis发送包含PDF附件的邮件测试成功")
return True
else:
error_msg = result.get('error', '未知错误')
logger.error(f"包含PDF附件的邮件发送失败: {error_msg}")
# 提供具体的解决方案
if "WeasyPrint" in error_msg:
logger.error("📄 WeasyPrint相关错误可能的原因:")
logger.error("1. WeasyPrint库未安装")
logger.error("2. WeasyPrint依赖项缺失")
logger.error("3. 权限问题导致无法生成PDF")
logger.error("解决方案: 运行 'pip install weasyprint' 安装WeasyPrint库")
return False
except Exception as e:
error_msg = str(e)
logger.error(f"通过Celery和Redis发送包含PDF附件的邮件测试失败: {error_msg}")
# 提供具体的解决方案
if "No such file or directory" in error_msg:
logger.error("📁 文件路径错误,可能的原因:")
logger.error("1. 报告目录不存在")
logger.error("2. 权限问题导致无法访问目录")
elif "Connection refused" in error_msg:
logger.error("🔌 连接被拒绝,可能的原因:")
logger.error("1. Celery worker未运行")
logger.error("2. Redis服务未运行")
logger.error("3. 防火墙阻止连接")
else:
logger.error("请检查:")
logger.error("1. Celery worker是否运行: sudo supervisorctl status celery_worker")
logger.error("2. Redis服务状态: sudo systemctl status redis-server")
logger.error("3. 系统配置中的邮件设置是否正确")
logger.error("4. SMTP服务器配置是否正确")
logger.error("5. WeasyPrint库是否已正确安装")
return False
def check_logs_config():
"""检查Gunicorn、Celery、Redis的日志是否写入同一个文件"""
logger.info("开始检查日志配置...")
@@ -430,7 +489,7 @@ def main():
logger.info("=" * 50)
tests_passed = 0
total_tests = 5
total_tests = 6
# 测试1: Redis连接
logger.info("\n[测试1] Redis连接测试")
@@ -454,8 +513,13 @@ def main():
if test_celery_redis_email():
tests_passed += 1
# 测试5: 检查日志配置
logger.info("\n[测试5] 检查日志配置")
# 测试5: 通过Celery和Redis发送包含PDF附件的邮件
logger.info("\n[测试5] 通过Celery和Redis发送包含PDF附件的邮件测试")
if test_celery_redis_pdf_email():
tests_passed += 1
# 测试6: 检查日志配置
logger.info("\n[测试6] 检查日志配置")
if check_logs_config():
tests_passed += 1
@@ -471,19 +535,21 @@ def main():
logger.info("2. ✅ Celery可以连接到Redis")
logger.info("3. ✅ Redis性能满足要求")
logger.info("4. ✅ 可以通过Celery和Redis发送邮件")
logger.info("5. ✅ 日志配置检查完成")
logger.info("6. 建议配置Redis持久化和备份")
logger.info("7. 建议监控Redis内存使用情况")
logger.info("8. 确保Gunicorn、Celery、Redis日志写入同一个文件")
logger.info("5. ✅ 可以通过Celery和Redis发送包含PDF附件的邮件")
logger.info("6. ✅ 日志配置检查完成")
logger.info("7. 建议配置Redis持久化和备份")
logger.info("8. 建议监控Redis内存使用情况")
logger.info("9. 确保Gunicorn、Celery、Redis日志写入同一个文件")
return 0
elif tests_passed >= 3:
elif tests_passed >= 4:
logger.warning("部分测试通过,生产环境基本可用。")
logger.info("\n需要检查:")
logger.info("1. 确保Redis服务正常运行")
logger.info("2. 检查Celery worker配置")
logger.info("3. 参考README中的故障排除指南")
logger.info("4. 检查邮件配置是否正确")
logger.info("5. 检查日志配置是否符合要求")
logger.info("5. 检查WeasyPrint库是否已正确安装")
logger.info("6. 检查日志配置是否符合要求")
return 1
else:
logger.error("多数测试失败,生产环境可能无法正常工作。")
@@ -493,7 +559,8 @@ def main():
logger.info("3. ❌ 检查Django settings.py中的Celery配置")
logger.info("4. ❌ 检查Celery worker是否运行")
logger.info("5. ❌ 检查系统配置中的邮件设置")
logger.info("6. 参考README中的Redis部署章节重新配置")
logger.info("6. ❌ 检查WeasyPrint库是否已正确安装")
logger.info("7. ❌ 参考README中的Redis部署章节重新配置")
return 1
if __name__ == "__main__":