GeneticAlgorithmUniversityClassScheduler/components/ScheduleParser.py at 382dcd755e80c0f987f4359dbbd509bcbf949a08 · edceliz/GeneticAlgorithmUniversityClassScheduler · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
from PyQt6 import QtCore, QtWidgets, QtGui
from components import Settings, TableModel, Utilities
import json


class ScheduleParser:
    # Section / Room View
    # Subject Name + Instructor
    # Instructor View
    # Subject Name + Instructor + Section

    # table = QTableView, data = []
    def __init__(self, table, data):
        self.table = table
        header = [["Monday", "Tuesday", "Wednesday", "Thursday", "Friday", "Saturday"]]
        with open("timeslots.json") as json_file:
            self.timeslots = timeslots = json.load(json_file)["timeslots"]
        self.settings = settings = Settings.getSettings()
        header.append(
            timeslots[settings["starting_time"] : settings["ending_time"] + 1]
        )
        temporaryData = []
        for i in range(settings["ending_time"] + 1 - settings["starting_time"]):
            temporaryData.append(["", "", "", "", "", ""])
        self.model = ScheduleParserModel(header, temporaryData)
        table.setModel(self.model)
        table.setFocusPolicy(QtCore.Qt.FocusPolicy.NoFocus)
        table.setSelectionMode(QtWidgets.QAbstractItemView.SelectionMode.NoSelection)
        # table.horizontalHeader().setSectionResizeMode(QtWidgets.QHeaderView.ResizeMode.Fixed)
        table.verticalHeader().setSectionResizeMode(
            QtWidgets.QHeaderView.ResizeMode.Fixed
        )
        self.parseData(data)

    # data = [{'color': None, 'text': '', 'instances': [[day, startingTS, endingTS]]}]
    def parseData(self, data):
        view = self.table
        model = self.model
        for entry in data:
            entry["color"] = Utilities.colorGenerator()
            for instance in entry["instances"]:
                index = model.index(instance[1], instance[0])
                view.setSpan(instance[1], instance[0], instance[2] - instance[1], 1)
                item = QtGui.QStandardItem(entry["text"])
                item.setBackground(QtGui.QBrush(QtGui.QColor(*entry["color"])))
                item.setForeground(
                    QtGui.QBrush(QtGui.QColor(*Utilities.textColor(entry["color"])))
                )
                model.setData(index, item)

    def subjectGenerator(self):
        print(self.settings["starting_time"])


class ScheduleParserModel(TableModel.TableModel):
    def __init__(self, header, data):
        super().__init__(header, data)

    def setData(self, index, value, role=None):
        if not index.isValid():
            return False
        elif role is None:
            self.data[index.row()][index.column()] = value
        self.dataChanged.emit(index, index)
        return True

    def data(self, index, role):
        if not index.isValid() or not self.data[index.row()][index.column()]:
            return QtCore.QVariant()
        elif role == QtCore.Qt.ItemDataRole.TextAlignmentRole:
            return QtCore.Qt.AlignmentFlag.AlignCenter
        elif role == QtCore.Qt.ItemDataRole.BackgroundRole:
            return self.data[index.row()][index.column()].background()
        elif role == QtCore.Qt.ItemDataRole.ForegroundRole:
            return self.data[index.row()][index.column()].foreground()
        elif role != QtCore.Qt.ItemDataRole.DisplayRole:
            return QtCore.QVariant()
        return self.data[index.row()][index.column()].text()