pyLanguageTool/file_loader_worker.py at main · knochenhans/pyLanguageTool · 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
from PySide6.QtCore import QThread, Signal
from PySide6.QtWidgets import QInputDialog


class FileLoaderWorker(QThread):
    fileLoaded = Signal(str)

    def __init__(self, text_editor, file_name: str):
        super().__init__()
        self.text_editor = text_editor
        self.file_name = file_name

        from file_handler import FileHandler

        self.file_handler = FileHandler(text_editor)

    def run(self):
        self.text_editor.statusBar().showMessage(f"Loading {self.file_name}...")
        text = self.file_handler.load_file(self.file_name)
        self.fileLoaded.emit(text)

    def select_column(self, column_names) -> int:
        column_name, ok = QInputDialog.getItem(
            self.text_editor,
            "Select Column",
            "Select the column to check:",
            column_names,
            0,
            False,
        )

        column_index = column_names.index(column_name)
        return column_index

    def select_table(self, table_names, tables) -> int:
        if len(tables) > 1:
            table_name, ok = QInputDialog.getItem(
                self.text_editor,
                "Select Table",
                "Select the table to check:",
                table_names,
                0,
                False,
            )
            if not ok:
                table_name = table_names[0]
        else:
            table_name = table_names[0]

        table_index = table_names.index(table_name) + 1
        return table_index