# 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'})