PyCuts/KeyboardDialog.py at main · DatBogie/PyCuts · 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
134
135
136
import sys
from PyQt6.QtWidgets import QDialog, QLabel, QPushButton, QVBoxLayout, QHBoxLayout, QWidget, QDialogButtonBox
from PyQt6.QtCore import Qt
from PyQt6.QtGui import QKeyEvent

MAP = {
    "Control": ["Meta","whitelist",["darwin"]],
    "Meta": ["Control","whitelist",["darwin"]]
}

def getTextFromQKeyEvent(x:QKeyEvent|Qt.Key):
    k = x.key() if type(x) == QKeyEvent else x
    key = "Tab" if k == Qt.Key.Key_Backtab else Qt.Key(k).name[4:]
    if key in MAP.keys():
        if MAP[key][1] == "blacklist":
            if sys.platform in MAP[key][2]: return key
        elif MAP[key][1] == "whitelist":
            if not sys.platform in MAP[key][2]: return key
        return MAP[key][0]
    return key

class btnIgnoreKeys(QPushButton):
    def __init__(self, text:str="", parent:QWidget=None):
        super().__init__(text,parent)
    def keyPressEvent(self, a0):
        a0.ignore()

class KeyboardDialog(QDialog):
    def __init__(self, parent=None, mode:int=0):
        super().__init__(parent)
        # self.setFixedSize(250,150)

        self.awaitingInput = False
        self.mode = mode
        self.data = {} if mode == 1 else None
        self.btn = btnIgnoreKeys()
        self.btn.clicked.connect(self.startListening)
        self.label = QLabel()

        main_lay = QVBoxLayout()
        main_lay.addWidget(self.label)
        main_lay.addWidget(self.btn)

        # cancel = QPushButton("Cancel")
        # ok = QPushButton("OK")

        buttons = QDialogButtonBox(QDialogButtonBox.StandardButton.Ok|QDialogButtonBox.StandardButton.Cancel)
        buttons.setFocusPolicy(Qt.FocusPolicy.NoFocus)

        # cancel.setFocusPolicy(Qt.FocusPolicy.NoFocus)
        # ok.setFocusPolicy(Qt.FocusPolicy.NoFocus)

        # cancel.clicked.connect(self.reject)
        # ok.clicked.connect(self.accept)

        buttons.rejected.connect(self.reject)
        buttons.accepted.connect(self.accept)

        # btn_lay = QHBoxLayout()
        # btn_lay.addStretch()

        # if sys.platform == "darwin":
        #     btn_lay.addWidget(cancel)
        #     btn_lay.addWidget(ok)
        # else:
        #     btn_lay.addWidget(ok)
        #     btn_lay.addWidget(cancel)

        # main_lay.addLayout(btn_lay)
        main_lay.addWidget(buttons)

        self.setLayout(main_lay)
        self.installEventFilter(self)

    class DialogMode():
        SINGLE_KEY = 0
        SHORTCUT = 1

    def keyPressEvent(self, a0):
        if not self.awaitingInput: return super().keyPressEvent(a0)
        if self.mode == 0:
            self.data = getTextFromQKeyEvent(a0)
            self.stopListening()
        else:
            self.data[getTextFromQKeyEvent(a0)] = getTextFromQKeyEvent(a0)
            # if not getTextFromQKeyEvent(a0).upper() in Qt.Modifier._member_names_:
                # self.stopListening()
        self.btn.setText(f"({self.updateText()})")
    def keyReleaseEvent(self, a0):
        if not self.awaitingInput: return super().keyReleaseEvent(a0)
        if self.mode == 0: return
        return
        if getTextFromQKeyEvent(a0) in self.data:
            self.data.pop(getTextFromQKeyEvent(a0))
        self.btn.setText(f"({self.updateText()})")
    def eventFilter(self, a0, a1):
        if type(a1) == QKeyEvent:
            if a1.key() == Qt.Key.Key_Tab:
                if a1.type() == QKeyEvent.Type.KeyPress:
                    self.keyPressEvent(a1)
                else:
                    self.keyReleaseEvent(a1)
                return False
        return super().eventFilter(a0, a1)

    def updateText(self) -> str:
        if self.mode == 0:
            txt = getTextFromQKeyEvent(self.data)
        else:
            txt = " + ".join(self.data.keys())
        self.btn.setText(txt)
        return txt
    def startListening(self):
        if self.awaitingInput:
            return self.stopListening()
        if self.mode == 1: self.data.clear()
        self.btn.setText("(...)")
        self.awaitingInput = True
    def stopListening(self):
        self.awaitingInput = False
        self.updateText()

    @staticmethod
    def getKey(parent:QWidget=None,title:str="",label:str="") -> tuple[QKeyEvent|None, bool]:
        dia = KeyboardDialog(parent,KeyboardDialog.DialogMode.SINGLE_KEY)
        dia.setWindowTitle(title)
        dia.label.setText(label)
        result = dia.exec()
        return dia.data, result == QDialog.DialogCode.Accepted
    @staticmethod
    def getShortcut(parent:QWidget=None,title:str="",label:str="") -> tuple[dict[str,QKeyEvent], bool]:
        dia = KeyboardDialog(parent,mode=KeyboardDialog.DialogMode.SHORTCUT)
        dia.setWindowTitle(title)
        dia.label.setText(label)
        result = dia.exec()
        return dia.data, result == QDialog.DialogCode.Accepted