from app import db
from datetime import datetime

class ShortCourse(db.Model):
    __tablename__ = 'short_courses'
    id = db.Column(db.Integer, primary_key=True)
    title = db.Column(db.String(100), nullable=False)
    description = db.Column(db.Text, nullable=True)
    reading_content = db.Column(db.Text, nullable=True)
    speaking_prompt = db.Column(db.Text, nullable=True)
    comprehension_questions = db.Column(db.JSON, nullable=True)  # List of questions
    created_at = db.Column(db.DateTime, default=datetime.utcnow)
    teacher_id = db.Column(db.Integer, db.ForeignKey('users.id'), nullable=True)
    
    # Relationship with the teacher (User model)
    teacher = db.relationship('User', backref=db.backref('short_courses', lazy=True))

    def __repr__(self):
        return f'<ShortCourse {self.title}>'

class LearningCourse(db.Model):
    __tablename__ = 'learning_courses'
    id = db.Column(db.Integer, primary_key=True)
    title = db.Column(db.String(100), nullable=False)
    description = db.Column(db.Text, nullable=True)
    reading_content = db.Column(db.Text, nullable=True)
    speaking_prompt = db.Column(db.Text, nullable=True)
    comprehension_questions = db.Column(db.JSON, nullable=True)  # List of questions
    created_at = db.Column(db.DateTime, default=datetime.utcnow)
    teacher_id = db.Column(db.Integer, db.ForeignKey('users.id'), nullable=True)
    
    # Relationship with the teacher (User model)
    teacher = db.relationship('User', backref=db.backref('learning_courses', lazy=True))

    def __repr__(self):
        return f'<LearningCourse {self.title}>'