from src.extensions import db

class AgenteTreino(db.Model):
    __tablename__ = 'agente_treinos'
    id = db.Column(db.Integer, primary_key=True)
    agente_id = db.Column(db.Integer, db.ForeignKey('agente.id', ondelete='CASCADE'), nullable=False)
    arquivo = db.Column(db.String(255), nullable=False)
    created_at = db.Column(db.DateTime, server_default=db.func.now())

    agente = db.relationship('Agente', back_populates='treinos')

    def __repr__(self):
        return f'<AgenteTreino {self.id}>'