gradehelper.py

import os
from bisect import bisect_left

from PyQt5 import QtWidgets
from PyQt5.QtGui import QDoubleValidator
from PyQt5.QtWidgets import QLabel, QLineEdit, QHBoxLayout, QWidget, QVBoxLayout, QPushButton, QMessageBox

import constants
from grade_helper_ops.compile_grade_report import compile_grade_report
from grade_helper_ops.load_next_student import load_next_student
from grade_helper_ops.submit_grades import submit_grades

# create working folder to copy files that I'm viewing to
if not os.path.exists(constants.WORKING_DIRECTORY):
    os.makedirs(constants.WORKING_DIRECTORY, exist_ok=True)


def create_message_pop_up_box(text):
    msg = QMessageBox()
    msg.setWindowTitle("Error")
    msg.setText(text)
    msg.setIcon(QMessageBox.Critical)
    msg.exec_()


# pyqt window
def create_layout_with_widget(label):
    widget = QLabel(label)
    layout_text_set_box = QLineEdit()
    layout_text_set_box.setText("")

    layout = QHBoxLayout()
    layout.addWidget(widget)
    layout.addWidget(layout_text_set_box)

    return layout


def layout_edit_widget(layout):
    return layout.itemAt(1).widget()


class Window(QtWidgets.QMainWindow):

    def __init__(self):
        super().__init__()
        self._student_directories = sorted(os.listdir(constants.LAB_DIRECTORY))
        self._current_student_graded_index = 0

        self._current_student_grades_submitted = False
        self._program_started = True  # just a flag when we start program

        self._columns = ["First name", "Surname", "ID number", "Feedback"]
        for i in range(constants.TASKS):
            self._columns.append(f"Task {i}")

        self._student_name_layout = create_layout_with_widget("Student Name: ")
        self._username_layout = create_layout_with_widget("Student Username: ")
        self._id_layout = create_layout_with_widget("Student ID: ")
        self._feedback_layout = create_layout_with_widget("Feedback: ")

        self._grade_layouts = []
        self._grade_layout_set_boxes = []
        for i in range(constants.TASKS):
            grade_layout_text = QLabel(f"Task {i}: ")
            grade_layout_text_set_box = QLineEdit(self)
            grade_layout_text_set_box.setValidator(QDoubleValidator(0, 10, 1))
            grade_layout_text_set_box.setText("0")

            grade_layout = QHBoxLayout()
            grade_layout.addWidget(grade_layout_text)
            grade_layout.addWidget(grade_layout_text_set_box)

            self._grade_layouts.append(grade_layout)

        submitButton = QPushButton('Submit Grades')
        nextButton = QPushButton('Next Student')
        compileButton = QPushButton('Compile Report')

        verticalLayout = QVBoxLayout()
        verticalLayout.addLayout(self._student_name_layout)
        verticalLayout.addLayout(self._username_layout)
        verticalLayout.addLayout(self._id_layout)
        verticalLayout.addLayout(self._feedback_layout)
        for layout in self._grade_layouts:
            verticalLayout.addLayout(layout)

        verticalLayout.addWidget(submitButton)
        verticalLayout.addWidget(nextButton)
        verticalLayout.addWidget(compileButton)

        submitButton.clicked.connect(self.submit_grades)
        nextButton.clicked.connect(self.load_next_student)
        compileButton.clicked.connect(self.compile_report)

        widget = QWidget()
        self.setCentralWidget(widget)
        widget.setLayout(verticalLayout)

        self.load_next_student()

    # this function should write out grades to each folder in a csv file per student
    def submit_grades(self):
        student_grade_path = os.path.join(constants.LAB_DIRECTORY,
                                          self._student_directories[self._current_student_graded_index],
                                          constants.CSV_FILE_NAME)

        first_name, last_name = "", ""
        if layout_edit_widget(self._student_name_layout).text():
            try:
                first_name, last_name = layout_edit_widget(self._student_name_layout).text().split(" ", 1)
            except ValueError:
                create_message_pop_up_box(
                    "Could not find either first or last name, please check student files or input random last "
                    "name (e.g John -> John Smith)"
                )
                return
        grades = self.get_grades()

        row_values = [
            first_name,
            last_name,
            layout_edit_widget(self._id_layout).text(),
            layout_edit_widget(self._feedback_layout).text()
        ]
        for grade in grades:
            row_values.append(grade)

        try:
            submit_grades(student_grade_path, self._columns, row_values)
        except ValueError as e:
            create_message_pop_up_box(e)

        self._current_student_grades_submitted = True
        self.clear_fields()

    def get_grades(self):
        grades = []
        for grade_layout in self._grade_layouts:
            grades.append(int(float(layout_edit_widget(grade_layout).text()) * constants.SCALE_FACTOR))
        return grades

    # Gets information for next student in the list that doesn't have a CSV already stored in the folder and replaces
    # working directory with next student's files
    def load_next_student(self):
        def binary_search(students, student_name):
            i = bisect_left(students, student_name)
            if i != len(students) and students[i] == student_name:
                return i
            else:
                return -1

        if self._current_student_grades_submitted or self._program_started:
            try:
                student = load_next_student(self._student_directories)
                if not student["error_message"]:
                    self._current_student_graded_index = binary_search(self._student_directories, student["username"])
                    layout_edit_widget(self._username_layout).setText(student["username"])
                    layout_edit_widget(self._student_name_layout).setText(student["name"])
                    layout_edit_widget(self._id_layout).setText(student["id"])
                else:
                    create_message_pop_up_box("\n".join(student["error_message"]))
            except ValueError as e:
                create_message_pop_up_box(f"{e}")

        else:
            create_message_pop_up_box("Grades have not been submitted yet")
        self._program_started = False
        self._current_student_grades_submitted = False

    # function to clear fields in the textboxes
    def clear_fields(self):
        layout_edit_widget(self._feedback_layout).setText("")
        for grade_layout in self._grade_layouts:
            layout_edit_widget(grade_layout).setText("0")

    # function will run through all directories in self.studentDirectories load the csvs and output a final csv in
    # the eclass format
    def compile_report(self):
        response = compile_grade_report(self._student_directories, constants.LAB_DIRECTORY,
                                        constants.FINAL_GRADE_REPORT_PATH, self._columns)
        if response["error"]:
            create_message_pop_up_box(response["error"])


app = QtWidgets.QApplication([])
w = Window()
w.show()
app.exec_()