# coding=utf-8
"""特征模板表（总服务器权威；SQLite 可缓存镜像）。

一条记录 = 某道菜的一张参考图特征向量，用于识别匹配。
"""
from datetime import datetime

from sqlalchemy import Column, DateTime, Float, ForeignKey, Index, Integer, String, Text
from sqlalchemy.orm import relationship
from sqlalchemy.sql import func

from dao.db import Base


class GlobalTemplate(Base):
    __tablename__ = 'global_templates'

    id = Column(Integer, primary_key=True, autoincrement=True, comment='主键ID')
    category_id = Column(
        Integer,
        ForeignKey('dish_categories.id', ondelete='RESTRICT'),
        nullable=False,
        comment='关联菜品ID',
    )
    # Base64(float32 特征向量)；跨 MySQL/SQLite 兼容，读写见 dao.feature_codec
    feature_blob = Column(Text, nullable=False, comment='特征向量 Base64(float32)')
    # 总服务器优先存可访问 URL；本地缓存可为文件路径
    image_url = Column(String(512), nullable=True, comment='模板图片 URL 或路径')
    thumbnail_url = Column(String(512), nullable=True, comment='缩略图 URL 或路径')
    source_base_id = Column(
        Integer,
        ForeignKey('bases.id', ondelete='SET NULL'),
        nullable=True,
        comment='上传该模板的基地ID',
    )
    confidence = Column(Float, nullable=False, default=1.0, server_default='1', comment='质量分/置信度')
    is_active = Column(Integer, nullable=False, default=1, server_default='1', comment='是否生效：1启用 0废弃')
    create_time = Column(
        DateTime,
        nullable=False,
        default=datetime.now,
        server_default=func.now(),
        comment='创建时间',
    )
    update_time = Column(
        DateTime,
        nullable=False,
        default=datetime.now,
        server_default=func.now(),
        onupdate=datetime.now,
        comment='更新时间（增量同步关键）',
    )

    category = relationship('DishCategory', back_populates='templates')
    source_base = relationship('BaseInfo', back_populates='templates', foreign_keys=[source_base_id])

    __table_args__ = (
        Index('idx_gt_category', 'category_id'),
        Index('idx_gt_update_time', 'update_time'),
        Index('idx_gt_active', 'is_active'),
        Index('idx_gt_active_update', 'is_active', 'update_time'),
        Index('idx_gt_source_base', 'source_base_id'),
    )

    # 兼容旧代码属性名
    @property
    def image_path(self):
        return self.image_url

    @image_path.setter
    def image_path(self, value):
        self.image_url = value

    @property
    def thumbnail_path(self):
        return self.thumbnail_url

    @thumbnail_path.setter
    def thumbnail_path(self, value):
        self.thumbnail_url = value

    def to_dict(self):
        return {
            'id': self.id,
            'category_id': self.category_id,
            'feature_blob': self.feature_blob,
            'image_url': self.image_url,
            'thumbnail_url': self.thumbnail_url,
            'source_base_id': self.source_base_id,
            'confidence': self.confidence,
            'is_active': self.is_active,
            'create_time': self.create_time.strftime('%Y-%m-%d %H:%M:%S') if self.create_time else None,
            'update_time': self.update_time.strftime('%Y-%m-%d %H:%M:%S') if self.update_time else None,
        }