89 lines
3.0 KiB
Python
89 lines
3.0 KiB
Python
# models.py
|
||
from datetime import datetime
|
||
from extensions import db
|
||
|
||
|
||
class Device(db.Model):
|
||
__tablename__ = 'devices'
|
||
|
||
id = db.Column(db.Integer, primary_key=True)
|
||
name = db.Column(db.String(100), unique=True, index=True)
|
||
source = db.Column(db.String(50))
|
||
|
||
# 快照字段（爬虫更新）
|
||
status = db.Column(db.String(50))
|
||
current_value = db.Column(db.String(200))
|
||
latest_time = db.Column(db.String(50))
|
||
json_data = db.Column(db.Text)
|
||
check_time = db.Column(db.String(50))
|
||
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'
|
||
return {
|
||
'id': self.id,
|
||
'name': self.name,
|
||
'source': self.source,
|
||
'latest_time': self.latest_time,
|
||
'status': api_status,
|
||
'status_text': self.status,
|
||
'value': self.current_value,
|
||
'reason': self.reason,
|
||
'install_site': self.install_site or '',
|
||
'is_maintaining': self.is_maintaining,
|
||
'is_hidden': self.is_hidden,
|
||
'is_whitelist': self.is_whitelist,
|
||
'offset': self.offset,
|
||
'file_count': self.file_count # ✅ 返回给前端
|
||
}
|
||
|
||
|
||
class DeviceHistory(db.Model):
|
||
__tablename__ = 'device_history'
|
||
|
||
id = db.Column(db.Integer, primary_key=True)
|
||
device_id = db.Column(db.Integer, db.ForeignKey('devices.id'))
|
||
|
||
data_time = db.Column(db.String(50))
|
||
status = db.Column(db.String(50))
|
||
result_data = db.Column(db.String(200), default="")
|
||
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)
|
||
device_name = db.Column(db.String(100), nullable=False)
|
||
engineer = db.Column(db.String(50))
|
||
location = db.Column(db.String(100))
|
||
content = db.Column(db.Text)
|
||
timestamp = db.Column(db.DateTime, default=datetime.now)
|
||
|
||
def to_dict(self):
|
||
return {
|
||
'id': self.id,
|
||
'device_name': self.device_name,
|
||
'engineer': self.engineer or '',
|
||
'location': self.location or '',
|
||
'content': self.content,
|
||
'timestamp': self.timestamp.strftime('%Y-%m-%d %H:%M:%S')
|
||
} |