hxf/backend/th_agenter/models/table_metadata.py

"""表元数据模型"""

from sqlalchemy import Integer, String, Text, DateTime, Boolean, JSON
from sqlalchemy.orm import Mapped, mapped_column
from ..db.base import BaseModel

class TableMetadata(BaseModel):
    """表元数据表"""
    __tablename__ = "table_metadata"
    
    id: Mapped[int] = mapped_column(Integer, primary_key=True, index=True)
    # database_config_id = Column(Integer, ForeignKey('database_configs.id'), nullable=False)
    table_name: Mapped[str] = mapped_column(String(100), nullable=False, index=True)
    table_schema: Mapped[str] = mapped_column(String(50), default='public')
    table_type: Mapped[str] = mapped_column(String(20), default='BASE TABLE')
    table_comment: Mapped[str | None] = mapped_column(Text, nullable=True)  # 表描述
    database_config_id: Mapped[int | None] = mapped_column(Integer, nullable=True) #数据库配置ID
    # 表结构信息
    columns_info: Mapped[dict] = mapped_column(JSON, nullable=False)  # 列信息：名称、类型、注释等
    primary_keys: Mapped[dict | None] = mapped_column(JSON, nullable=True)   # 主键列表
    foreign_keys: Mapped[dict | None] = mapped_column(JSON, nullable=True)   # 外键信息
    indexes: Mapped[dict | None] = mapped_column(JSON, nullable=True)        # 索引信息
    
    # 示例数据
    sample_data: Mapped[dict | None] = mapped_column(JSON, nullable=True)    # 前5条示例数据
    row_count: Mapped[int] = mapped_column(Integer, default=0)       # 总行数
    
    # 问答相关
    is_enabled_for_qa: Mapped[bool] = mapped_column(Boolean, default=True)  # 是否启用问答
    qa_description: Mapped[str | None] = mapped_column(Text, nullable=True)       # 问答描述
    business_context: Mapped[str | None] = mapped_column(Text, nullable=True)     # 业务上下文

    last_synced_at: Mapped[DateTime | None] = mapped_column(DateTime(timezone=True), nullable=True)  # 最后同步时间
    
    # 关系
    # database_config = relationship("DatabaseConfig", back_populates="table_metadata")
    
    def to_dict(self):
        return {
            "id": self.id,
            "created_by": self.created_by,  # 改为created_by
            "database_config_id": self.database_config_id,
            "table_name": self.table_name,
            "table_schema": self.table_schema,
            "table_type": self.table_type,
            "table_comment": self.table_comment,
            "columns_info": self.columns_info,
            "primary_keys": self.primary_keys,
            # "foreign_keys": self.foreign_keys,
            "indexes": self.indexes,
            "sample_data": self.sample_data,
            "row_count": self.row_count,
            "is_enabled_for_qa": self.is_enabled_for_qa,
            "qa_description": self.qa_description,
            "business_context": self.business_context,
            "created_at": self.created_at.isoformat() if self.created_at else None,
            "updated_at": self.updated_at.isoformat() if self.updated_at else None,
            "last_synced_at": self.last_synced_at.isoformat() if self.last_synced_at else None
        }