"""
聊天消息模型
"""
from datetime import datetime
from typing import Optional
from sqlmodel import SQLModel, Field, Relationship
from .user import User


class ChatMessage(SQLModel, table=True):
    """聊天消息表"""
    __tablename__ = "chat_messages"
    
    id: Optional[int] = Field(default=None, primary_key=True)
    content: str
    sent_at: datetime = Field(default_factory=datetime.now)
    user_id: int = Field(foreign_key="users.id", index=True)
    room: str = Field(max_length=50, default="main", index=True)
    
    # 关系
    user: Optional[User] = Relationship()