# models.py
from datetime import datetime
import json
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))      # 滞后描述

    install_site = db.Column(db.String(100), default="")
    is_maintaining = db.Column(db.Boolean, default=False)
    is_hidden = db.Column(db.Boolean, default=False)

    def to_dict(self):
        # 兼容处理 API 返回
        api_status = 'offline' if self.status in ['离线', '异常', '已离线'] else 'online'
        # 注意：列表接口不返回 json_data 以提高性能
        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,
            'is_maintaining': self.is_maintaining,
            'is_hidden': self.is_hidden,
            'offset': self.offset
        }

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))

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