修改新增加文件数量的查询功能

2_1banben/models.py

@@ -1,7 +1,8 @@
+# models.py
 from datetime import datetime
-import json
 from extensions import db
 
+
 class Device(db.Model):
     __tablename__ = 'devices'
 
@@ -18,11 +19,17 @@ class Device(db.Model):
     reason = db.Column(db.String(255))
     offset = db.Column(db.String(50))
 
+    # ✅ 新增字段：文件数量
+    file_count = db.Column(db.Integer, default=0)
+
     # 手动录入字段（受保护，run_monitor 不主动覆盖）
     install_site = db.Column(db.String(100), default="")
     is_maintaining = db.Column(db.Boolean, default=False)
     is_hidden = db.Column(db.Boolean, default=False)
 
+    # 白名单字段 (根据上下文可能存在，补全以防万一)
+    is_whitelist = db.Column(db.Boolean, default=False)
+
     def to_dict(self):
         # 统一状态映射逻辑
         api_status = 'offline' if self.status in ['离线', '异常', '已离线'] else 'online'
@@ -38,9 +45,12 @@ class Device(db.Model):
             'install_site': self.install_site or '',
             'is_maintaining': self.is_maintaining,
             'is_hidden': self.is_hidden,
-            'offset': self.offset
+            'is_whitelist': self.is_whitelist,
+            'offset': self.offset,
+            'file_count': self.file_count  # ✅ 返回给前端
         }
 
+
 class DeviceHistory(db.Model):
     __tablename__ = 'device_history'
 
@@ -53,8 +63,12 @@ class DeviceHistory(db.Model):
     json_data = db.Column(db.Text)
     file_path = db.Column(db.String(255))
 
+    # ✅ 新增字段：历史记录文件数量
+    file_count = db.Column(db.Integer, default=0)
+
     recorded_at = db.Column(db.DateTime, default=datetime.now)
 
+
 class MaintenanceLog(db.Model):
     __tablename__ = 'maintenance_logs'
     id = db.Column(db.Integer, primary_key=True)