from extensions import db
from datetime import datetime

# Modelo para TimeSchedule (Horários dentro de um turno)
class TimeSchedule(db.Model):
    __tablename__ = 'time_schedules'
    
    id = db.Column(db.Integer, primary_key=True)
    shift_id = db.Column(db.Integer, db.ForeignKey('shift.id'), nullable=False)  # Relaciona com Shift
    start_time = db.Column(db.Time, nullable=False)  # Hora de início do time slot
    end_time = db.Column(db.Time, nullable=False)    # Hora de término do time slot
    description = db.Column(db.String(255))           # Descrição do time schedule (ex: 'Almoço')
    records = db.relationship("Record", back_populates="schedule")

    def __repr__(self):
        return f'<TimeSchedule {self.description} ({self.start_time} - {self.end_time})>'