From 2a6f43c6d47a417de3adf238bfabc6da6f1ff4d9 Mon Sep 17 00:00:00 2001
From: Pieterv24 <9167905+Pieterv24@users.noreply.github.com>
Date: Mon, 19 Apr 2021 00:53:07 +0200
Subject: [PATCH] added keylogger

---
 requirements.txt               |  4 ++--
 src/main/python/matrix_test.py | 15 +++++++++++++--
 2 files changed, 15 insertions(+), 4 deletions(-)

diff --git a/requirements.txt b/requirements.txt
index d022793..e6ab8ba 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -6,6 +6,6 @@ keyboard==0.13.5
 macholib==1.14
 pefile==2019.4.18
 PyInstaller==3.4
-PyQt5==5.9.2
+PyQt5==5.15.2
 https://github.com/danthedeckie/simpleeval/archive/41c99b8e224a7a0ae0ac59c773598fe79a4470db.zip
-sip==4.19.8
+PyQt5-sip==12.8.1
\ No newline at end of file
diff --git a/src/main/python/matrix_test.py b/src/main/python/matrix_test.py
index 33131c4..c97ea59 100644
--- a/src/main/python/matrix_test.py
+++ b/src/main/python/matrix_test.py
@@ -15,6 +15,8 @@ class MatrixTest(BasicEditor):
         self.layout_editor = layout_editor
 
         self.keyboardWidget = KeyboardWidget(layout_editor)
+        self.keyboardWidget.set_enabled(False)
+
         self.startButtonWidget = QPushButton("Start testing")
 
         layout = QVBoxLayout()
@@ -43,8 +45,6 @@ class MatrixTest(BasicEditor):
         return isinstance(self.device, VialKeyboard)
 
     def matrix_poller(self):
-        # print(f"Rows: {self.keyboard.rows}")
-        # print(f"Cols: {self.keyboard.cols}")
         rows = self.keyboard.rows
         cols = self.keyboard.cols
         matrix = [ [None for y in range(cols)] for x in range(rows) ]
@@ -63,6 +63,17 @@ class MatrixTest(BasicEditor):
                 state = (row_data[col_byte] >> col) & 1
                 matrix[row][col] = state
 
+        for w in self.keyboardWidget.widgets:
+            row = w.desc.row
+            col = w.desc.col
+
+            if row < len(matrix) and col < len(matrix[row]):
+                w.setActive(matrix[row][col])
+        
+        self.keyboardWidget.update_layout()
+        self.keyboardWidget.update()
+        self.keyboardWidget.updateGeometry()
+
     def start_poller(self):
         if not self.polling:
             Unlocker.unlock(self.keyboard)