from datetime import datetime
from .. import db


class Message(db.Model):
    __tablename__ = 'Messages'

    id = db.Column(db.Integer, primary_key=True)
    content = db.Column(db.String(500), nullable=True)
    remark = db.Column(db.Text, nullable=True)
    role = db.Column(db.String(50), nullable=False)

    # # 外键字段（使用驼峰命名）
    sessionId = db.Column(db.Integer, db.ForeignKey(
        'Sessions.id'), nullable=False)  # 外键关联 Session 表

    # 通用字段
    createdAt = db.Column(db.DateTime, default=datetime.now)  # 创建时间
    updatedAt = db.Column(db.DateTime, default=datetime.now,
                          onupdate=datetime.now)  # 更新时间
    deleteAt = db.Column(db.DateTime, default=None)  # 软删除标记

    session = db.relationship('Session', backref='messages')

    def to_dict(self):
        return {
            'id': self.id,
            'content': self.content,
            'remark': self.remark,
            'role': self.role,
            # 'session': {
            #     "id": self.session.id,
            #     "name": self.session.name,
            # }
        }

    def __repr__(self):
        return f'<Message {self.content}>'