from sqlalchemy import Column, Integer, ForeignKey, VARCHAR
from sqlalchemy.orm import relationship

from app.infrastructure.database.models import Base


class Step(Base):
    __tablename__ = 'steps'

    id = Column(Integer, primary_key=True, autoincrement=True)
    title = Column(VARCHAR(200))

    lesson_id = Column(Integer, ForeignKey('lessons.id'), nullable=False)
    type_id = Column(Integer, ForeignKey('step_types.id'), nullable=False)

    lesson = relationship('Lesson', back_populates='steps')
    type = relationship('StepType', back_populates='steps')

    lectures = relationship('Lectures', back_populates='step')
    tasks = relationship('StepTask', back_populates='step')