from datetime import datetime
import json  # 用于在模型内部处理序列化（可选，主要在业务逻辑用）
from extensions import db


# -------------------------------------------------------------------------
# 1. 设备主表 (快照表)
#    作用：永远存储所有出现过的设备。
#    逻辑：如果网页上设备消失了，这里的记录不会删，只是不会更新时间，
#          这样你就能知道它“失联”了，但数据还在。
# -------------------------------------------------------------------------
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 数据字符串 🔥🔥🔥
    # 这样无论爬虫爬到什么奇怪字段，都可以在这里找到
    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)

    history = db.relationship('DeviceHistory', backref='device', lazy='dynamic', cascade='all, delete-orphan')

    def to_dict(self):
        """转字典，供前端 API 使用"""
        api_status = 'offline' if self.status in ['离线', '异常', '已离线'] else 'online'

        # 尝试解析 JSON 字符串返回给前端对象，如果解析失败则返回原字符串
        raw_obj = None
        if self.json_data:
            try:
                raw_obj = json.loads(self.json_data)
            except:
                raw_obj = self.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,
            'raw_json': raw_obj,  # 🔥 前端可以看到原始数据
            'reason': self.reason,
            'install_site': self.install_site,
            'is_maintaining': self.is_maintaining,
            'is_hidden': self.is_hidden,
            'offset': self.offset
        }


# -------------------------------------------------------------------------
# 2. 历史记录表 (流水账/日志表)
#    作用：无限追加，记录每一次抓取的数据。
# -------------------------------------------------------------------------
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 包 🔥🔥🔥
    json_data = db.Column(db.Text)

    file_path = db.Column(db.String(255), default="")  # 关联的归档文件路径
    recorded_at = db.Column(db.DateTime, default=datetime.now)


# -------------------------------------------------------------------------
# 3. 维修日志表 (无变化)
# -------------------------------------------------------------------------
class MaintenanceLog(db.Model):
    __tablename__ = 'maintenance_log'
    id = db.Column(db.Integer, primary_key=True)
    device_name = 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,
            'content': self.content,
            'timestamp': self.timestamp.strftime('%Y-%m-%d %H:%M:%S') if self.timestamp else ""
        }