202 lines
7.2 KiB
Python
202 lines
7.2 KiB
Python
import os
|
||
import sys
|
||
import json
|
||
import logging
|
||
import mimetypes
|
||
from datetime import datetime
|
||
from flask import Flask, send_from_directory, jsonify
|
||
from flask_cors import CORS
|
||
from flask_apscheduler import APScheduler
|
||
|
||
# ==============================================================================
|
||
# ✅ 1. 核心模块引用
|
||
# ==============================================================================
|
||
try:
|
||
# 数据库实例 (在根目录 extensions.py 中)
|
||
from extensions import db
|
||
|
||
# 数据模型 (在根目录 models.py 中)
|
||
from models import Device, DeviceHistory, MaintenanceLog
|
||
|
||
# 核心业务逻辑 (在 services/core.py 中)
|
||
from services.core import execute_monitor_task
|
||
|
||
# 路由蓝图 (在 routes/api.py 中)
|
||
try:
|
||
from routes.api import api_bp as device_bp
|
||
except ImportError:
|
||
from routes.api import device_bp
|
||
|
||
# 工具函数 (在 routes/api.py 中)
|
||
from routes.api import calculate_offset
|
||
|
||
except ImportError as e:
|
||
print(f"❌ 严重错误: 模块导入失败。请检查文件名和变量名。详细信息: {e}")
|
||
print(f"系统路径: {sys.path}")
|
||
sys.exit(1)
|
||
|
||
|
||
# ==============================================================================
|
||
# 2. 路径计算模块 (兼容 PyInstaller 打包)
|
||
# ==============================================================================
|
||
def get_base_path():
|
||
"""获取运行时基准路径，兼容开发环境和打包环境"""
|
||
if getattr(sys, 'frozen', False):
|
||
if hasattr(sys, '_MEIPASS'):
|
||
return sys._MEIPASS # --onefile 模式
|
||
else:
|
||
return os.path.dirname(os.path.abspath(sys.executable)) # --onedir 模式
|
||
else:
|
||
return os.path.abspath(os.path.dirname(__file__))
|
||
|
||
|
||
BASE_DIR = get_base_path()
|
||
STATIC_FOLDER = os.path.join(BASE_DIR, 'web_dist')
|
||
INSTANCE_FOLDER = os.path.join(BASE_DIR, 'instance')
|
||
DB_PATH = os.path.join(INSTANCE_FOLDER, 'devices.db')
|
||
|
||
# 修复 Windows 下注册表 MIME 类型缺失导致网页白屏的问题
|
||
mimetypes.add_type('application/javascript', '.js')
|
||
mimetypes.add_type('text/css', '.css')
|
||
|
||
print(f"🚀 运行环境: {'Packaged' if getattr(sys, 'frozen', False) else 'Dev'}")
|
||
print(f"📂 基准路径: {BASE_DIR}")
|
||
print(f"💾 数据库路径: {DB_PATH}")
|
||
|
||
|
||
# ==============================================================================
|
||
# 3. 定时任务逻辑
|
||
# ==============================================================================
|
||
def auto_monitor_job(app):
|
||
"""定时任务具体执行逻辑"""
|
||
with app.app_context():
|
||
print(f"⏰ [定时任务] 启动时间: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}")
|
||
|
||
if not execute_monitor_task:
|
||
print("❌ 错误: 无法加载爬虫核心模块 (execute_monitor_task is Missing)")
|
||
return
|
||
|
||
try:
|
||
# 执行爬取
|
||
task_result = execute_monitor_task()
|
||
if not task_result:
|
||
print("⚠️ [定时任务] 爬虫未获取到数据")
|
||
return
|
||
|
||
scraped_list = task_result.get('device_list', [])
|
||
current_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
|
||
|
||
count = 0
|
||
for item in scraped_list:
|
||
d_name = item.get('name')
|
||
if not d_name: continue
|
||
|
||
# 查找或创建设备
|
||
device = Device.query.filter_by(name=d_name).first()
|
||
if not device:
|
||
device = Device(name=d_name, source=item.get('source'), install_site="")
|
||
db.session.add(device)
|
||
db.session.flush()
|
||
|
||
# 更新字段
|
||
device.status = item.get('status')
|
||
device.current_value = item.get('value')
|
||
device.latest_time = item.get('target_time')
|
||
device.check_time = current_time
|
||
device.json_data = json.dumps(item.get('raw_json', {}), ensure_ascii=False)
|
||
device.offset = calculate_offset(item.get('target_time'))
|
||
|
||
# 写入历史
|
||
db.session.add(DeviceHistory(
|
||
device_id=device.id,
|
||
status=device.status,
|
||
result_data=device.current_value,
|
||
data_time=item.get('target_time'),
|
||
json_data=device.json_data
|
||
))
|
||
count += 1
|
||
|
||
db.session.commit()
|
||
print(f"✅ [定时任务] 成功更新 {count} 台设备状态")
|
||
|
||
except Exception as e:
|
||
db.session.rollback()
|
||
print(f"❌ [定时任务] 异常: {str(e)}")
|
||
|
||
|
||
# ==============================================================================
|
||
# 4. Flask 应用工厂
|
||
# ==============================================================================
|
||
def create_app():
|
||
# 🔴 关键修复：移除了 static_url_path=''
|
||
# 这样 Flask 就不会强制拦截所有根路径请求，让下面的 serve_static 有机会处理 /dashboard
|
||
app = Flask(__name__, static_folder=STATIC_FOLDER)
|
||
|
||
CORS(app)
|
||
|
||
# 确保 instance 目录存在
|
||
if not os.path.exists(INSTANCE_FOLDER):
|
||
os.makedirs(INSTANCE_FOLDER, exist_ok=True)
|
||
|
||
app.config['SQLALCHEMY_DATABASE_URI'] = f'sqlite:///{DB_PATH}'
|
||
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
|
||
app.config['SCHEDULER_API_ENABLED'] = True
|
||
|
||
# 初始化数据库
|
||
db.init_app(app)
|
||
|
||
# 初始化定时任务
|
||
scheduler = APScheduler()
|
||
scheduler.init_app(app)
|
||
scheduler.start()
|
||
|
||
# 添加定时任务 (每天 10:00)
|
||
scheduler.add_job(
|
||
id='daily_monitor_task',
|
||
func=auto_monitor_job,
|
||
args=[app],
|
||
trigger='cron',
|
||
hour=12,
|
||
minute=0
|
||
)
|
||
|
||
# 注册蓝图
|
||
app.register_blueprint(device_bp)
|
||
|
||
# -------------------------------------------------
|
||
# 前端路由支持 (Vue History Mode)
|
||
# -------------------------------------------------
|
||
@app.route('/')
|
||
def serve_index():
|
||
if not os.path.exists(os.path.join(app.static_folder, 'index.html')):
|
||
return "❌ 错误: 前端文件丢失 (web_dist/index.html)", 404
|
||
return send_from_directory(app.static_folder, 'index.html')
|
||
|
||
@app.route('/<path:path>')
|
||
def serve_static(path):
|
||
# 1. 优先尝试直接返回实际存在的文件 (js, css, img等)
|
||
file_path = os.path.join(app.static_folder, path)
|
||
if os.path.exists(file_path):
|
||
return send_from_directory(app.static_folder, path)
|
||
|
||
# 2. 如果是 API 请求但没找到对应接口，返回 404 JSON (不返回 HTML)
|
||
if path.startswith('api') or path.startswith('static'):
|
||
return jsonify({'code': 404, 'message': 'Not Found'}), 404
|
||
|
||
# 3. 关键逻辑：
|
||
# 访问 /dashboard 等前端路由时，文件系统中并没有 dashboard 这个文件
|
||
# 所以会走到这里，返回 index.html，让 Vue 及其 Router 接管页面渲染
|
||
return send_from_directory(app.static_folder, 'index.html')
|
||
|
||
with app.app_context():
|
||
db.create_all()
|
||
|
||
return app
|
||
|
||
|
||
if __name__ == '__main__':
|
||
app = create_app()
|
||
# 生产环境/打包环境通常设为 False
|
||
debug_mode = not getattr(sys, 'frozen', False)
|
||
print("🚀 服务启动中...")
|
||
app.run(host='0.0.0.0', port=5000, debug=debug_mode, use_reloader=False) |