TimeTableParser/python-scripts/classes/controller/gui_controller.py at 678cb4143fd56ad317d15af22348e6df8d478472 · FachschaftIuE/TimeTableParser · 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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
from tkinter import messagebox

from classes.controller.timetable_controller import TimetableController
from classes.file_management.loader import Loader
from classes.models.timetable import Timetable
from classes.view import gui
from helper.data_handler import get_module_names_from_data_items
from helper.data_item_from_json import data_items_from_json
from helper.data_output import create_ics, create_csv, create_json_from_data_item
from helper.folder_manager import path_leaf, get_cached_file_for
from helper.get_pdf_pages import count_pdf_pages


class GuiController:

    def __init__(self):
        self.input_files = []
        self.selectable_modules = []
        self._parsed_data_items = []

    def clear_inputs(self):
        self.input_files.clear()
        self.selectable_modules.clear()
        self._parsed_data_items.clear()

    def create_calendar(self, agreed_to_tos, export_as_ics, selected_module_indices):
        if not agreed_to_tos:
            messagebox.showerror(gui.title, "You need to agree to the Terms Of Service to do that.")
            return

        if len(self._parsed_data_items) == 0:
            messagebox.showinfo(gui.title, "There are no modules to select.\nPlease parse a timetable.")
            return

        if len(selected_module_indices) == 0:
            messagebox.showinfo(gui.title, "There are no modules selected.\n"
                                           "Please select modules to fill the calendar with.")
            return

        selected_module_names = []
        for index in selected_module_indices:
            selected_module_names.append(self.selectable_modules[index])

        selected_modules = []
        for data_item in self._parsed_data_items:
            if selected_module_names.__contains__(data_item.module):
                selected_modules.append(data_item)

        if export_as_ics:
            create_ics(selected_modules)
        else:
            create_csv(selected_modules)

        messagebox.showinfo(gui.title, "Created the calendar. Check the /data/output folder! :)")

    def parse_input_files(self, use_cache, agreed_to_tos):
        if not agreed_to_tos:
            messagebox.showerror(gui.title, "You need to agree to the Terms Of Service to do that.")
            return

        if len(self.input_files) == 0:
            messagebox.showinfo(gui.title, "There are no timetables to parse. "
                                           "To parse a timetable add it via the 'Add Timetable'-Button.")
            return

        # reset from previous parse
        self.selectable_modules = []
        self._parsed_data_items = []

        messagebox.showinfo(gui.title, "This process may take a while. Grab a something to drink! :)")

        for file in self.input_files:
            try:
                cached_file = get_cached_file_for(path_leaf(file))
                file_is_parsed_already = cached_file.is_file()

                data_items_in_file = get_data_items_from_file(cached_file) if file_is_parsed_already and use_cache \
                    else get_data_items_from_file(file)

                for data_item in data_items_in_file:
                    self._parsed_data_items.append(data_item)

                modules_in_data_items = get_module_names_from_data_items(data_items_in_file)
                for module in modules_in_data_items:
                    self.selectable_modules.append(module)

                if not (file_is_parsed_already and use_cache):
                    create_json_from_data_item(data_items_in_file, path_leaf(cached_file).replace('.json', ''))
                print(path_leaf(file) + " was successfully parsed.")

            except Exception as e:
                messagebox.showerror(gui.title,
                                     "Couldn't parse the timetable '" + path_leaf(file) + "' because:\n" + str(e))

        messagebox.showinfo(gui.title, "Parsing finished! :)")


def get_data_items_from_file(file):
    """
    Summary
    -------
    Retrieves the data items from the given file.

    Parameter
    ---------
    file : path                 # Absolute path of the file that's going to be parsed.
                                  Supports *.json and *.pdf files.
    timetable : timetable       # The timetable to retrieve the info with.
    """

    data_items = []

    if str(file).endswith('.pdf'):
        # get data items from *.pdf file
        timetable = Timetable()

        page_count = count_pdf_pages(file)
        for page_index in range(page_count):
            timetable_controller = TimetableController(Loader(file, path_leaf(file), page_index), timetable)
            timetable_controller.create_timetable_information()
            timetable_controller.send_data_to_timetable()
            timetable.search_modules()
            timetable.get_weeks()
            timetable.find_modules(data_items)

    elif str(file).endswith('.json'):
        # get data items from *.json file
        data_items_from_json(data_items, file)

    else:
        throw: Exception("File type not supported. Please provide only *.pdf or *.json files.")

    return data_items