|
@@ -14,6 +14,7 @@ Requirements: 3.1, 3.4, 4.1, 4.2, 4.3, 4.4, 4.5, 4.6
|
|
|
import os
|
|
import os
|
|
|
import json
|
|
import json
|
|
|
import uuid
|
|
import uuid
|
|
|
|
|
+import redis
|
|
|
from datetime import datetime
|
|
from datetime import datetime
|
|
|
from flask import jsonify, request, current_app
|
|
from flask import jsonify, request, current_app
|
|
|
from werkzeug.utils import secure_filename
|
|
from werkzeug.utils import secure_filename
|
|
@@ -257,7 +258,8 @@ def create_task():
|
|
|
# 先测试Redis连接
|
|
# 先测试Redis连接
|
|
|
import redis
|
|
import redis
|
|
|
print(f"🔍 测试Redis连接...")
|
|
print(f"🔍 测试Redis连接...")
|
|
|
- r = redis.Redis(host='localhost', port=6379, db=0)
|
|
|
|
|
|
|
+ broker_url = current_app.config.get('CELERY_BROKER_URL', 'redis://localhost:6379/0')
|
|
|
|
|
+ r = redis.from_url(broker_url)
|
|
|
r.ping()
|
|
r.ping()
|
|
|
print(f"✅ Redis连接成功")
|
|
print(f"✅ Redis连接成功")
|
|
|
|
|
|
|
@@ -511,7 +513,8 @@ def upload_scan():
|
|
|
|
|
|
|
|
try:
|
|
try:
|
|
|
# Test Redis connection
|
|
# Test Redis connection
|
|
|
- r = redis.Redis(host='localhost', port=6379, db=0)
|
|
|
|
|
|
|
+ broker_url = current_app.config.get('CELERY_BROKER_URL', 'redis://localhost:6379/0')
|
|
|
|
|
+ r = redis.from_url(broker_url)
|
|
|
r.ping()
|
|
r.ping()
|
|
|
|
|
|
|
|
# Initialize Celery
|
|
# Initialize Celery
|