models.py

import sqlite3
from datetime import datetime

DATABASE = 'lms.db'

def get_db():
    """Kết nối database"""
    conn = sqlite3.connect(DATABASE)
    conn.row_factory = sqlite3.Row
    return conn

def init_db():
    """Khởi tạo database với các bảng cần thiết"""
    conn = get_db()
    cursor = conn.cursor()
    
    # Bảng users
    cursor.execute('''
        CREATE TABLE IF NOT EXISTS users (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            username TEXT UNIQUE NOT NULL,
            password TEXT NOT NULL,
            full_name TEXT NOT NULL,
            role TEXT NOT NULL CHECK(role IN ('admin', 'teacher', 'student')),
            created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
        )
    ''')
    
    # Bảng courses
    cursor.execute('''
        CREATE TABLE IF NOT EXISTS courses (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            code TEXT UNIQUE NOT NULL,
            name TEXT NOT NULL,
            description TEXT,
            teacher_id INTEGER,
            color TEXT DEFAULT '#ec4899',
            created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
            FOREIGN KEY (teacher_id) REFERENCES users(id)
        )
    ''')
    
    # Bảng course_enrollments (sinh viên tham gia khóa học)
    cursor.execute('''
        CREATE TABLE IF NOT EXISTS course_enrollments (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            course_id INTEGER NOT NULL,
            student_id INTEGER NOT NULL,
            enrolled_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
            FOREIGN KEY (course_id) REFERENCES courses(id),
            FOREIGN KEY (student_id) REFERENCES users(id),
            UNIQUE(course_id, student_id)
        )
    ''')
    
    # Bảng materials (học liệu)
    cursor.execute('''
        CREATE TABLE IF NOT EXISTS materials (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            course_id INTEGER NOT NULL,
            title TEXT NOT NULL,
            description TEXT,
            file_path TEXT,
            file_name TEXT,
            uploaded_by INTEGER NOT NULL,
            created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
            FOREIGN KEY (course_id) REFERENCES courses(id),
            FOREIGN KEY (uploaded_by) REFERENCES users(id)
        )
    ''')
    
    # Bảng announcements (thông báo)
    cursor.execute('''
        CREATE TABLE IF NOT EXISTS announcements (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            title TEXT NOT NULL,
            content TEXT NOT NULL,
            author_id INTEGER NOT NULL,
            created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
            FOREIGN KEY (author_id) REFERENCES users(id)
        )
    ''')
    
    # Bảng forum_posts (diễn đàn)
    cursor.execute('''
        CREATE TABLE IF NOT EXISTS forum_posts (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            title TEXT NOT NULL,
            content TEXT NOT NULL,
            author_id INTEGER NOT NULL,
            likes INTEGER DEFAULT 0,
            created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
            FOREIGN KEY (author_id) REFERENCES users(id)
        )
    ''')
    
    # Bảng forum_likes (lượt thích bài viết)
    cursor.execute('''
        CREATE TABLE IF NOT EXISTS forum_likes (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            post_id INTEGER NOT NULL,
            user_id INTEGER NOT NULL,
            created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
            FOREIGN KEY (post_id) REFERENCES forum_posts(id),
            FOREIGN KEY (user_id) REFERENCES users(id),
            UNIQUE(post_id, user_id)
        )
    ''')
    
    # Bảng quizzes (bài kiểm tra)
    cursor.execute('''
        CREATE TABLE IF NOT EXISTS quizzes (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            course_id INTEGER NOT NULL,
            title TEXT NOT NULL,
            description TEXT,
            duration INTEGER DEFAULT 30,
            max_score INTEGER DEFAULT 100,
            show_results INTEGER DEFAULT 1,
            show_correct_answers INTEGER DEFAULT 0,
            results_published INTEGER DEFAULT 0,
            created_by INTEGER NOT NULL,
            created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
            FOREIGN KEY (course_id) REFERENCES courses(id),
            FOREIGN KEY (created_by) REFERENCES users(id)
        )
    ''')
    
    # Thêm cột results_published nếu bảng đã tồn tại
    try:
        cursor.execute('ALTER TABLE quizzes ADD COLUMN results_published INTEGER DEFAULT 0')
    except:
        pass  # Cột đã tồn tại
    
    # Thêm cột show_correct_answers nếu bảng đã tồn tại
    try:
        cursor.execute('ALTER TABLE quizzes ADD COLUMN show_correct_answers INTEGER DEFAULT 0')
    except:
        pass  # Cột đã tồn tại
    
    # Thêm cột is_visible nếu bảng đã tồn tại
    try:
        cursor.execute('ALTER TABLE quizzes ADD COLUMN is_visible INTEGER DEFAULT 1')
    except:
        pass  # Cột đã tồn tại
    
    # Bảng quiz_questions (câu hỏi)
    cursor.execute('''
        CREATE TABLE IF NOT EXISTS quiz_questions (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            quiz_id INTEGER NOT NULL,
            question_text TEXT NOT NULL,
            question_type TEXT NOT NULL CHECK(question_type IN ('multiple_choice', 'essay')),
            points INTEGER DEFAULT 10,
            order_num INTEGER DEFAULT 0,
            created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
            FOREIGN KEY (quiz_id) REFERENCES quizzes(id)
        )
    ''')
    
    # Bảng quiz_choices (đáp án trắc nghiệm)
    cursor.execute('''
        CREATE TABLE IF NOT EXISTS quiz_choices (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            question_id INTEGER NOT NULL,
            choice_text TEXT NOT NULL,
            is_correct INTEGER DEFAULT 0,
            FOREIGN KEY (question_id) REFERENCES quiz_questions(id)
        )
    ''')
    
    # Bảng quiz_submissions (bài nộp)
    cursor.execute('''
        CREATE TABLE IF NOT EXISTS quiz_submissions (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            quiz_id INTEGER NOT NULL,
            student_id INTEGER NOT NULL,
            score REAL DEFAULT 0,
            status TEXT DEFAULT 'pending' CHECK(status IN ('pending', 'graded')),
            submitted_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
            FOREIGN KEY (quiz_id) REFERENCES quizzes(id),
            FOREIGN KEY (student_id) REFERENCES users(id),
            UNIQUE(quiz_id, student_id)
        )
    ''')
    
    # Bảng quiz_answers (câu trả lời)
    cursor.execute('''
        CREATE TABLE IF NOT EXISTS quiz_answers (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            submission_id INTEGER NOT NULL,
            question_id INTEGER NOT NULL,
            choice_id INTEGER,
            essay_answer TEXT,
            points_earned REAL DEFAULT 0,
            FOREIGN KEY (submission_id) REFERENCES quiz_submissions(id),
            FOREIGN KEY (question_id) REFERENCES quiz_questions(id),
            FOREIGN KEY (choice_id) REFERENCES quiz_choices(id)
        )
    ''')
    
    conn.commit()
    
    # Thêm dữ liệu mẫu
    cursor.execute("SELECT COUNT(*) FROM users")
    if cursor.fetchone()[0] == 0:
        # Tạo tài khoản mẫu
        users_data = [
            ('admin', 'admin123', 'Quản trị viên', 'admin'),
            ('teacher1', 'teacher123', 'Nguyễn Văn A', 'teacher'),
            ('teacher2', 'teacher123', 'Trần Thị B', 'teacher'),
            ('student1', 'student123', 'Bùi Minh Sơn', 'student'),
            ('student2', 'student123', 'Nguyễn Thái Dương', 'student'),
        ]
        cursor.executemany(
            'INSERT INTO users (username, password, full_name, role) VALUES (?, ?, ?, ?)',
            users_data
        )
        
        # Tạo khóa học mẫu
        courses_data = [
            ('HIS1001_40', 'Lịch sử Đảng Cộng sản Việt Nam', 'Khóa học về lịch sử Đảng CSVN', 2, '#ec4899'),
            ('INT2210_44', 'Cấu trúc dữ liệu và giải thuật', 'Khóa học về CTDL&GT', 2, '#14b8a6'),
            ('INT3310_40', 'Quản trị mạng', 'Khóa học về quản trị mạng máy tính', 2, '#ec4899'),
            ('POL1001_42', 'Tư tưởng Hồ Chí Minh', 'Khóa học về tư tưởng HCM', 2, '#14b8a6'),
            ('INT3318E_40', 'Các thiết bị mạng và môi trường truyền', 'Khóa học về thiết bị mạng', 2, '#ec4899'),
            ('SPO1001_40', 'Golf', 'Khóa học Golf', 2, '#14b8a6'),
        ]
        cursor.executemany(
            'INSERT INTO courses (code, name, description, teacher_id, color) VALUES (?, ?, ?, ?, ?)',
            courses_data
        )
        
        # Đăng ký sinh viên vào khóa học
        enrollments_data = [
            (1, 4), (2, 4), (3, 4), (4, 4), (5, 4), (6, 4),
            (1, 5), (2, 5), (3, 5),
        ]
        cursor.executemany(
            'INSERT INTO course_enrollments (course_id, student_id) VALUES (?, ?)',
            enrollments_data
        )
        
        # Tạo học liệu mẫu
        materials_data = [
            (5, 'Bài thực hành trên lab 404', 'Ngày hôm nay thầy có việc bận nên các em lên phòng lab 404 để thực hành', '6.2.4_Packet_Tracer_-_Configure_EtherChannel.pka', '6.2.4_Packet_Tracer_-_Configure_EtherChannel.pka', 2),
            (5, '6.4.1 Packet Tracer - Implement Etherchannel', 'Hôm nay lớp sẽ làm bài thực hành 6.4.1 trên phòng máy', 'CCNAv7_SRWE_Skills_Assessment.pka', 'CCNAv7_SRWE_Skills_Assessment.pka', 2),
            (5, 'Test', 'test', 'Automobile_data.csv', 'Automobile_data.csv', 2),
        ]
        cursor.executemany(
            'INSERT INTO materials (course_id, title, description, file_name, file_path, uploaded_by) VALUES (?, ?, ?, ?, ?, ?)',
            materials_data
        )
        
        # Tạo thông báo mẫu
        announcements_data = [
            ('Hôm nay nghỉ nhé các bé', 'Hôm nay nghỉ nhé các bé', 4),
            ('TALKSHOW EXPLORE TECHNOLOGY WITH VIETCOMBANK NGÀY 06/12/2022', 'Thời gian chương trình: 8h00 - 11h00 ngày 06/12/2022 Địa điểm: Tầng 1, Hội trường SunWah, 144 Xuân Thủy, Cầu Giấy, Hà Nội', 4),
        ]
        cursor.executemany(
            'INSERT INTO announcements (title, content, author_id) VALUES (?, ?, ?)',
            announcements_data
        )
        
        # Tạo bài viết diễn đàn mẫu
        forum_data = [
            ('Câu hỏi về diễn đàn', 'Diễn đàn này được sử dụng cho các sinh viên trên toàn khóa ?', 4, 0),
            ('TestPost2', 'TestPost2', 4, 0),
            ('TestPost', 'TestPost', 5, 0),
        ]
        cursor.executemany(
            'INSERT INTO forum_posts (title, content, author_id, likes) VALUES (?, ?, ?, ?)',
            forum_data
        )
        
        conn.commit()
    
    conn.close()

if __name__ == '__main__':
    init_db()
    print("Database initialized successfully!")