325 lines
10 KiB
Python
325 lines
10 KiB
Python
from flask import Blueprint, jsonify, current_app, g
|
||
import sqlite3
|
||
import datetime
|
||
|
||
|
||
bp = Blueprint('device_warning', __name__, url_prefix='/api/warning')
|
||
|
||
|
||
def get_db():
|
||
"""获取数据库连接"""
|
||
if 'db' not in g:
|
||
g.db = sqlite3.connect(
|
||
current_app.config['DATABASE'],
|
||
detect_types=sqlite3.PARSE_DECLTYPES
|
||
)
|
||
g.db.row_factory = sqlite3.Row
|
||
return g.db
|
||
|
||
|
||
def close_db(e=None):
|
||
"""关闭数据库连接"""
|
||
db = g.pop('db', None)
|
||
if db is not None:
|
||
db.close()
|
||
|
||
|
||
def is_data_constant(data_points, threshold=0.5, time_threshold_hours=2):
|
||
"""
|
||
检测数据是否长时间保持不变
|
||
data_points: 数据点列表，每个元素为(timestamp, value)
|
||
threshold: 数值变化阈值，小于此值视为不变
|
||
time_threshold_hours: 时间阈值，单位小时（默认2小时）
|
||
"""
|
||
if len(data_points) < 2:
|
||
return False, None
|
||
|
||
# 检查时间跨度是否达到阈值
|
||
first_time = datetime.datetime.strptime(data_points[0][0], '%Y-%m-%d %H:%M:%S')
|
||
last_time = datetime.datetime.strptime(data_points[-1][0], '%Y-%m-%d %H:%M:%S')
|
||
time_diff = last_time - first_time
|
||
|
||
if time_diff.total_seconds() < time_threshold_hours * 3600:
|
||
return False, None
|
||
|
||
# 检查数值变化是否小于阈值
|
||
first_value = data_points[0][1]
|
||
for time, value in data_points[1:]:
|
||
if abs(value - first_value) > threshold:
|
||
return False, None
|
||
|
||
return True, first_value
|
||
|
||
|
||
def check_device_warnings():
|
||
"""检查所有设备是否有数据长时间不变的情况"""
|
||
db = get_db()
|
||
cursor = db.cursor()
|
||
|
||
# 获取所有设备
|
||
cursor.execute("SELECT id, device_name, device_code FROM device")
|
||
devices = cursor.fetchall()
|
||
|
||
warning_updates = []
|
||
|
||
for device in devices:
|
||
device_id = device['id']
|
||
device_name = device['device_name']
|
||
device_code = device['device_code']
|
||
|
||
# 检查设备当前状态
|
||
cursor.execute("SELECT status FROM device WHERE id = ?", (device_id,))
|
||
device_status = cursor.fetchone()
|
||
|
||
# 如果设备已经是故障状态，则跳过检测
|
||
if device_status['status'] == 'Faulty':
|
||
continue
|
||
|
||
warning_type = None
|
||
warning_value = None
|
||
|
||
# 检查温度数据（使用2小时阈值）
|
||
cursor.execute(
|
||
"""SELECT timestamp, temperature FROM temperature_data
|
||
WHERE device_id = ?
|
||
ORDER BY timestamp DESC
|
||
LIMIT 24""", # 取最近24条数据
|
||
(device_id,)
|
||
)
|
||
temp_data = cursor.fetchall()
|
||
|
||
if temp_data:
|
||
is_constant, value = is_data_constant(
|
||
[(item['timestamp'], item['temperature']) for item in temp_data]
|
||
)
|
||
if is_constant:
|
||
warning_type = 'temperature_constant'
|
||
warning_value = value
|
||
|
||
# 如果温度没有问题，检查湿度数据（使用2小时阈值）
|
||
if not warning_type:
|
||
cursor.execute(
|
||
"""SELECT timestamp, humidity FROM temperature_data
|
||
WHERE device_id = ?
|
||
ORDER BY timestamp DESC
|
||
LIMIT 24""",
|
||
(device_id,)
|
||
)
|
||
humidity_data = cursor.fetchall()
|
||
|
||
if humidity_data:
|
||
is_constant, value = is_data_constant(
|
||
[(item['timestamp'], item['humidity']) for item in humidity_data]
|
||
)
|
||
if is_constant:
|
||
warning_type = 'humidity_constant'
|
||
warning_value = value
|
||
|
||
# 如果温度和湿度都没有问题，检查pH值数据（使用1小时阈值）
|
||
if not warning_type:
|
||
cursor.execute(
|
||
"""SELECT timestamp, ph FROM temperature_data
|
||
WHERE device_id = ?
|
||
ORDER BY timestamp DESC
|
||
LIMIT 24""",
|
||
(device_id,)
|
||
)
|
||
ph_data = cursor.fetchall()
|
||
|
||
if ph_data:
|
||
is_constant, value = is_data_constant(
|
||
[(item['timestamp'], item['ph']) for item in ph_data],
|
||
threshold=0.2, # pH变化阈值更小
|
||
time_threshold_hours=1 # pH检查时间阈值更短
|
||
)
|
||
if is_constant:
|
||
warning_type = 'ph_constant'
|
||
warning_value = value
|
||
|
||
# 更新设备状态
|
||
if warning_type:
|
||
# 如果设备之前不是警告状态，则更新
|
||
if device_status['status'] != 'Faulty':
|
||
warning_updates.append({
|
||
'device_id': device_id,
|
||
'status': 'warning',
|
||
'warning_type': warning_type,
|
||
'warning_value': warning_value,
|
||
'warning_time': datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')
|
||
})
|
||
else:
|
||
# 如果设备之前是警告状态，但现在不是了，则恢复为正常状态
|
||
if device_status['status'] == 'warning':
|
||
warning_updates.append({
|
||
'device_id': device_id,
|
||
'status': 'normal',
|
||
'warning_type': None,
|
||
'warning_value': None,
|
||
'warning_time': None
|
||
})
|
||
|
||
# 批量更新设备状态
|
||
for update in warning_updates:
|
||
cursor.execute(
|
||
"""
|
||
UPDATE device
|
||
SET status = ?,
|
||
warning_type = ?,
|
||
warning_value = ?,
|
||
warning_time = ?
|
||
WHERE id = ?
|
||
""",
|
||
(
|
||
update['status'],
|
||
update['warning_type'],
|
||
update['warning_value'],
|
||
update['warning_time'],
|
||
update['device_id']
|
||
)
|
||
)
|
||
|
||
db.commit()
|
||
return warning_updates
|
||
|
||
|
||
@bp.route('/check', methods=['GET'])
|
||
def check_warnings():
|
||
"""检查并更新设备警告状态"""
|
||
try:
|
||
updates = check_device_warnings()
|
||
return jsonify({
|
||
'status': 'success',
|
||
'data': updates,
|
||
'message': f'更新了{len(updates)}个设备状态'
|
||
})
|
||
except Exception as e:
|
||
return jsonify({
|
||
'status': 'error',
|
||
'message': f'检查设备警告失败: {str(e)}'
|
||
}), 500
|
||
|
||
|
||
@bp.route('/list', methods=['GET'])
|
||
def get_warning_list():
|
||
"""获取警告设备列表（优先使用存储的故障描述）"""
|
||
try:
|
||
db = get_db()
|
||
cursor = db.cursor()
|
||
|
||
# 尝试获取完整警告信息（包括fault_description字段）
|
||
try:
|
||
cursor.execute(
|
||
"""
|
||
SELECT
|
||
id,
|
||
device_name,
|
||
device_code,
|
||
status,
|
||
warning_type,
|
||
warning_value,
|
||
warning_time,
|
||
fault_description # 显式查询存储的故障描述
|
||
FROM device
|
||
WHERE status = 'Faulty' OR status = 'warning'
|
||
ORDER BY warning_time DESC
|
||
"""
|
||
)
|
||
warnings = cursor.fetchall()
|
||
|
||
warning_list = []
|
||
for warning in warnings:
|
||
warning_dict = dict(warning)
|
||
|
||
# 优先使用数据库中存储的fault_description
|
||
stored_description = warning_dict.get('fault_description')
|
||
|
||
if stored_description:
|
||
warning_dict['fault_description'] = stored_description
|
||
else:
|
||
# 如果没有存储描述，则根据warning_type生成默认描述
|
||
warning_type = warning_dict.get('warning_type')
|
||
if warning_type == 'temperature_constant':
|
||
warning_dict['fault_description'] = '温度持续异常'
|
||
elif warning_type == 'humidity_constant':
|
||
warning_dict['fault_description'] = '湿度持续异常'
|
||
elif warning_type == 'ph_constant':
|
||
warning_dict['fault_description'] = 'pH值持续异常'
|
||
else:
|
||
warning_dict['fault_description'] = '环境数据异常'
|
||
|
||
warning_list.append(warning_dict)
|
||
|
||
return jsonify({
|
||
'status': 'success',
|
||
'data': warning_list,
|
||
'message': f'获取到{len(warning_list)}个警告设备'
|
||
})
|
||
|
||
except sqlite3.OperationalError as e:
|
||
# 如果表结构不完整（缺少fault_description等字段），回退到简单查询
|
||
cursor.execute(
|
||
"""
|
||
SELECT
|
||
id,
|
||
device_name,
|
||
device_code,
|
||
status
|
||
FROM device
|
||
WHERE status = 'Faulty' OR status = 'warning'
|
||
ORDER BY created_at DESC
|
||
"""
|
||
)
|
||
warnings = cursor.fetchall()
|
||
|
||
# 简单查询结果只能提供默认描述
|
||
warning_list = [dict(warning, fault_description='环境数据异常')
|
||
for warning in warnings]
|
||
|
||
return jsonify({
|
||
'status': 'success',
|
||
'data': warning_list,
|
||
'message': f'获取到{len(warning_list)}个警告设备（简化模式）'
|
||
})
|
||
|
||
except Exception as e:
|
||
return jsonify({
|
||
'status': 'error',
|
||
'message': f'获取警告设备列表失败: {str(e)}'
|
||
}), 500
|
||
|
||
@bp.route('/resolve/<int:device_id>', methods=['POST'])
|
||
def resolve_warning(device_id):
|
||
"""解决设备警告"""
|
||
try:
|
||
db = get_db()
|
||
cursor = db.cursor()
|
||
|
||
cursor.execute(
|
||
"""
|
||
UPDATE device
|
||
SET status = 'Online',
|
||
warning_type = NULL,
|
||
warning_value = NULL,
|
||
warning_time = NULL
|
||
WHERE id = ?
|
||
""",
|
||
(device_id,)
|
||
)
|
||
db.commit()
|
||
|
||
return jsonify({
|
||
'status': 'success',
|
||
'message': '设备警告已解除'
|
||
})
|
||
except Exception as e:
|
||
return jsonify({
|
||
'status': 'error',
|
||
'message': f'解除设备警告失败: {str(e)}'
|
||
}), 500
|
||
|
||
|
||
# 初始化数据库表结构（如需在代码中初始化）
|
||
def init_db():
|
||
db = get_db()
|
||
with current_app.open_resource('schema.sql') as f:
|
||
db.executescript(f.read().decode('utf8')) |