215 lines
7.4 KiB
Python
215 lines
7.4 KiB
Python
# routes/api.py
|
|
import os
|
|
import shutil
|
|
import json
|
|
from flask import Blueprint, jsonify, request
|
|
from datetime import datetime
|
|
from extensions import db # ✅ 从 extensions 导入 db 用于提交事务
|
|
from models import Device, DeviceHistory, MaintenanceLog
|
|
|
|
# 尝试导入爬虫模块，如果没有则跳过（防止报错）
|
|
try:
|
|
from services.core import execute_monitor_task
|
|
except ImportError:
|
|
execute_monitor_task = None
|
|
|
|
# ✅ 定义蓝图，设置统一前缀 /api
|
|
api_bp = Blueprint('api', __name__, url_prefix='/api')
|
|
|
|
|
|
def calculate_offset(latest_time_str):
|
|
"""计算时间差的辅助函数"""
|
|
if not latest_time_str or latest_time_str == "N/A": return "从未同步"
|
|
try:
|
|
clean = str(latest_time_str).split()[0].replace('_', '-')
|
|
if len(clean) < 8: return latest_time_str
|
|
target = datetime.strptime(clean, "%Y-%m-%d").date()
|
|
diff = (datetime.now().date() - target).days
|
|
return "当天已同步" if diff == 0 else f"滞后 {diff} 天"
|
|
except:
|
|
return "时间解析失败"
|
|
|
|
|
|
# 👇👇👇 修复核心：补全 devices_overview 接口 👇👇👇
|
|
@api_bp.route('/devices_overview', methods=['GET'])
|
|
def devices_overview():
|
|
try:
|
|
# 获取所有设备
|
|
devices = Device.query.all()
|
|
# 转换为字典列表
|
|
data_list = [d.to_dict() for d in devices]
|
|
return jsonify({'code': 200, 'data': data_list})
|
|
except Exception as e:
|
|
print(f"Error in devices_overview: {e}")
|
|
return jsonify({'code': 500, 'message': str(e)})
|
|
|
|
|
|
@api_bp.route('/run_monitor', methods=['POST'])
|
|
def run_monitor():
|
|
try:
|
|
if not execute_monitor_task:
|
|
return jsonify({'code': 500, 'message': 'Core module missing'})
|
|
|
|
# 1. 准备归档目录
|
|
base_dir = os.path.abspath(os.path.dirname(os.path.dirname(__file__)))
|
|
binary_dir = os.path.join(base_dir, 'instance', 'binary')
|
|
if not os.path.exists(binary_dir): os.makedirs(binary_dir)
|
|
|
|
# 2. 获取数据 (列表)
|
|
task_result = execute_monitor_task()
|
|
if not task_result: return jsonify({'code': 200, 'message': '任务跳过(正在运行)'})
|
|
|
|
scraped_list = task_result.get('device_list', [])
|
|
current_check_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
|
|
|
|
# 3. 处理每一条数据
|
|
for item in scraped_list:
|
|
d_name = item.get('name')
|
|
d_source = item.get('source')
|
|
d_status = item.get('status')
|
|
d_value = item.get('value')
|
|
d_time = item.get('target_time')
|
|
d_raw_json = item.get('raw_json', {})
|
|
d_temp_file = item.get('temp_file')
|
|
|
|
# --- 文件归档 ---
|
|
final_db_rel_path = ""
|
|
if d_temp_file and os.path.exists(d_temp_file):
|
|
# 移动文件: instance/temp -> instance/binary
|
|
fname = os.path.basename(d_temp_file)
|
|
dest = os.path.join(binary_dir, fname)
|
|
try:
|
|
shutil.move(d_temp_file, dest)
|
|
final_db_rel_path = f"binary/{fname}"
|
|
except Exception as e:
|
|
print(f"File move error: {e}")
|
|
|
|
# --- 序列化 Raw JSON ---
|
|
json_str = json.dumps(d_raw_json, ensure_ascii=False)
|
|
|
|
# --- A. 更新 Device (快照) ---
|
|
device = Device.query.filter_by(name=d_name).first()
|
|
if not device:
|
|
device = Device(name=d_name, source=d_source)
|
|
db.session.add(device)
|
|
db.session.flush() # 获取 ID
|
|
|
|
device.status = d_status
|
|
device.current_value = d_value
|
|
device.latest_time = d_time
|
|
device.check_time = current_check_time
|
|
device.json_data = json_str
|
|
device.offset = calculate_offset(d_time)
|
|
|
|
# --- B. 插入 History (日志) ---
|
|
new_history = DeviceHistory(
|
|
device_id=device.id,
|
|
status=d_status,
|
|
result_data=d_value,
|
|
data_time=d_time,
|
|
json_data=json_str,
|
|
file_path=final_db_rel_path
|
|
)
|
|
db.session.add(new_history)
|
|
|
|
db.session.commit()
|
|
return jsonify({'code': 200, 'message': f'更新 {len(scraped_list)} 台设备'})
|
|
|
|
except Exception as e:
|
|
db.session.rollback()
|
|
return jsonify({'code': 500, 'message': str(e)})
|
|
|
|
|
|
@api_bp.route('/device_history', methods=['GET'])
|
|
def get_device_history():
|
|
try:
|
|
name = request.args.get('name')
|
|
device = Device.query.filter_by(name=name).first()
|
|
if not device: return jsonify({'error': 'Not found'}), 404
|
|
|
|
history = DeviceHistory.query.filter_by(device_id=device.id) \
|
|
.order_by(DeviceHistory.recorded_at.desc()).limit(100).all()
|
|
|
|
data = []
|
|
for h in history:
|
|
raw = None
|
|
if h.json_data:
|
|
try:
|
|
raw = json.loads(h.json_data)
|
|
except:
|
|
raw = h.json_data
|
|
|
|
data.append({
|
|
'recorded_at': h.recorded_at.strftime('%Y-%m-%d %H:%M:%S'),
|
|
'data_time': h.data_time,
|
|
'status': h.status,
|
|
'value': h.result_data,
|
|
'file_path': h.file_path,
|
|
'raw_data': raw
|
|
})
|
|
return jsonify({'device': device.name, 'history': data})
|
|
except Exception as e:
|
|
return jsonify({'error': str(e)}), 500
|
|
|
|
|
|
# 👇👇👇 以下是 Vue 前端用到的其他接口，之前缺失的 👇👇👇
|
|
|
|
@api_bp.route('/update_site', methods=['POST'])
|
|
def update_site():
|
|
"""更新安装地点"""
|
|
data = request.get_json()
|
|
name = data.get('name')
|
|
site = data.get('site')
|
|
|
|
device = Device.query.filter_by(name=name).first()
|
|
if device:
|
|
device.install_site = site
|
|
db.session.commit()
|
|
return jsonify({'code': 200, 'message': 'Updated'})
|
|
return jsonify({'code': 404, 'message': 'Device not found'}), 404
|
|
|
|
|
|
@api_bp.route('/toggle_maintenance', methods=['POST'])
|
|
def toggle_maintenance():
|
|
"""切换维修状态"""
|
|
data = request.get_json()
|
|
name = data.get('name')
|
|
is_maintaining = data.get('is_maintaining')
|
|
|
|
device = Device.query.filter_by(name=name).first()
|
|
if device:
|
|
device.is_maintaining = is_maintaining
|
|
db.session.commit()
|
|
return jsonify({'code': 200, 'message': 'Updated'})
|
|
return jsonify({'code': 404, 'message': 'Device not found'}), 404
|
|
|
|
|
|
@api_bp.route('/toggle_hidden', methods=['POST'])
|
|
def toggle_hidden():
|
|
"""切换隐藏/回收站状态"""
|
|
data = request.get_json()
|
|
name = data.get('name')
|
|
is_hidden = data.get('is_hidden')
|
|
|
|
device = Device.query.filter_by(name=name).first()
|
|
if device:
|
|
device.is_hidden = is_hidden
|
|
db.session.commit()
|
|
return jsonify({'code': 200, 'message': 'Updated'})
|
|
return jsonify({'code': 404, 'message': 'Device not found'}), 404
|
|
|
|
|
|
@api_bp.route('/logs/add', methods=['POST'])
|
|
def add_log():
|
|
"""添加维修日志"""
|
|
data = request.get_json()
|
|
device_name = data.get('device_name')
|
|
content = data.get('content')
|
|
|
|
if not device_name or not content:
|
|
return jsonify({'code': 400, 'message': 'Missing data'}), 400
|
|
|
|
new_log = MaintenanceLog(device_name=device_name, content=content)
|
|
db.session.add(new_log)
|
|
db.session.commit()
|
|
return jsonify({'code': 200, 'message': 'Log added'}) |