ScreenShotMake/launch_window.py at master · YevGenerator/ScreenShotMake · 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
import PySide2
from PySide2.QtCore import Slot, Qt
from PySide2.QtWidgets import QFileDialog

from ui_launcher import Ui_Form
from defining_area import DefineAreaWindow, QWidget, QApplication
from command_reader import HotKey, KeyMapper
from hotkeys_window import HotKeyWindow
import sys

from hotkey_processor import HotKeyProcessor
from default_struct import DefaultStruct
from settings_window import SettingsWindow


class LaunchWindow(QWidget):
    def __init__(self, parent=None):
        super().__init__(parent)
        self.ui = Ui_Form()
        self.ui.setupUi(self)
        self.set_connections()
        self.defaults = DefaultStruct.from_file("defaults.json")
        self.load_default_value(self.defaults)
        self.keys = KeyMapper.from_file("keys.json")
        self.win: DefineAreaWindow = None
        self.hotkey_win: HotKeyWindow = None
        self.processor: HotKeyProcessor = None
        self.setting_window: SettingsWindow = None

    def load_default_value(self, struct: DefaultStruct):
        self.ui.path_input.setText(struct.default_path)
        self.ui.filename_input.setText(struct.default_image_name)
        self.ui.extension_input.setText(struct.default_image_ext)
        self.ui.backend_input.setCurrentText(struct.default_backend)

    def get_struct(self):
        return DefaultStruct(
            default_path=self.ui.path_input.text(),
            default_backend=self.ui.backend_input.currentText(),
            default_image_name=self.ui.filename_input.text(),
            default_image_ext=self.ui.extension_input.text(),
            default_is_paused=self.defaults.default_is_paused,
            default_after_screen_defining=self.defaults.default_after_screen_defining
        )

    def set_connections(self):
        self.ui.launch_button.clicked.connect(self.launch_button_click)
        self.ui.hotkey_button.clicked.connect(self.hotkey_button_click)
        self.ui.browse_button.clicked.connect(self.browse_button_click)
        self.ui.settings_button.clicked.connect(self.settings_button_click)

    def launch_button_click(self):
        self.processor = HotKeyProcessor(self,
                                         self.keys.active_keys(),
                                         self.get_struct())
        self.processor.setWindowFlags(self.processor.windowFlags() | Qt.WindowModal)
        self.processor.show()
        self.ui.launch_button.setEnabled(False)
        self.processor.closed.connect(self.unblock_launch)
        self.processor.start()

    def browse_button_click(self):
        path = QFileDialog.getExistingDirectory()
        if path:
            self.ui.path_input.setText(path)

    @Slot()
    def unblock_launch(self):
        self.ui.launch_button.setEnabled(True)

    @Slot(DefaultStruct)
    def set_new_defaults(self, defaults: DefaultStruct):
        self.defaults = defaults
        self.load_default_value(self.defaults)
        self.defaults.save("defaults.json")

    def hotkey_button_click(self):
        self.hotkey_win = HotKeyWindow(self, self.keys)
        self.hotkey_win.setWindowFlags(self.hotkey_win.windowFlags() | Qt.WindowModal)
        self.hotkey_win.show()

    def settings_button_click(self):
        self.setting_window = SettingsWindow(parent=self, defaults=self.defaults)
        self.setting_window.setWindowFlags(self.setting_window.windowFlags() | Qt.WindowModal)
        self.setting_window.closed.connect(self.set_new_defaults)
        self.setting_window.show()

    def closeEvent(self, event: PySide2.QtGui.QCloseEvent) -> None:
        self.keys.write("keys.json")


if __name__ == "__main__":
    app = QApplication(sys.argv)

    window = LaunchWindow()
    window.show()

    sys.exit(app.exec_())