2.3权限管理

2_3banben/app.py

@@ -0,0 +1,202 @@
+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=10,
+        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)