V1

.gitignore

@@ -0,0 +1,2 @@
+venv
+.idea

WaaaghLampGUI.py

@@ -0,0 +1,314 @@
+#!/usr/bin/env python3
+
+import sys
+import socket
+import json
+
+# from PyQt6.QtCore import QLine
+from PyQt6.QtWidgets import QApplication, QMainWindow, QPushButton, QLabel, QFrame, \
+    QVBoxLayout, QWidget, QButtonGroup, QStatusBar, QComboBox
+from PyQt6.QtGui import QPixmap, QIcon
+
+WAL_HOST = '10.0.1.11'
+WAL_PORT = 9000
+
+MODES = {
+    "off": 0,
+    "red": 1,
+    "green": 2,
+    "blue": 3,
+    "white": 4,
+    "orange": 5,
+}
+
+SPECIAL_MODES = {
+    "cyclone": 11,
+    "police": 12,
+}
+
+CUSTOM_MODES = {
+    "solid": 21,
+    "twinkle": 22,
+    "fade": 23,
+}
+
+PALLETE = {
+    "black": 0x000000,
+    "aliceblue": 0xF0F8FF,
+    "amethyst": 0x9966CC,
+    "antiquewhite": 0xFAEBD7,
+    "aqua": 0x00FFFF,
+    "aquamarine": 0x7FFFD4,
+    "azure": 0xF0FFFF,
+    "beige": 0xF5F5DC,
+    "bisque": 0xFFE4C4,
+    "blanchedalmond": 0xFFEBCD,
+    "blue": 0x0000FF,
+    "blueviolet": 0x8A2BE2,
+    "brown": 0xA52A2A,
+    "burlywood": 0xDEB887,
+    "cadetblue": 0x5F9EA0,
+    "chartreuse": 0x7FFF00,
+    "chocolate": 0xD2691E,
+    "coral": 0xFF7F50,
+    "cornflowerblue": 0x6495ED,
+    "cornsilk": 0xFFF8DC,
+    "crimson": 0xDC143C,
+    "cyan": 0x00FFFF,
+    "darkblue": 0x00008B,
+    "darkcyan": 0x008B8B,
+    "darkgoldenrod": 0xB8860B,
+    "darkgray": 0xA9A9A9,
+    "darkgreen": 0x006400,
+    "darkkhaki": 0xBDB76B,
+    "darkmagenta": 0x8B008B,
+    "darkolivegreen": 0x556B2F,
+    "darkorange": 0xFF8C00,
+    "darkorchid": 0x9932CC,
+    "darkred": 0x8B0000,
+    "darksalmon": 0xE9967A,
+    "darkseagreen": 0x8FBC8F,
+    "darkslateblue": 0x483D8B,
+    "darkslategray": 0x2F4F4F,
+    "darkturquoise": 0x00CED1,
+    "darkviolet": 0x9400D3,
+    "deeppink": 0xFF1493,
+    "deepskyblue": 0x00BFFF,
+    "dimgray": 0x696969,
+    "dodgerblue": 0x1E90FF,
+    "firebrick": 0xB22222,
+    "floralwhite": 0xFFFAF0,
+    "forestgreen": 0x228B22,
+    "fuchsia": 0xFF00FF,
+    "gainsboro": 0xDCDCDC,
+    "ghostwhite": 0xF8F8FF,
+    "gold": 0xFFD700,
+    "goldenrod": 0xDAA520,
+    "gray": 0x808080,
+    "green": 0x008000,
+    "greenyellow": 0xADFF2F,
+    "honeydew": 0xF0FFF0,
+    "hotpink": 0xFF69B4,
+    "indianred": 0xCD5C5C,
+    "indigo": 0x4B0082,
+    "ivory": 0xFFFFF0,
+    "khaki": 0xF0E68C,
+    "lavender": 0xE6E6FA,
+    "lavenderblush": 0xFFF0F5,
+    "lawngreen": 0x7CFC00,
+    "lemonchiffon": 0xFFFACD,
+    "lightblue": 0xADD8E6,
+    "lightcoral": 0xF08080,
+    "lightcyan": 0xE0FFFF,
+    "lightgoldenrodyellow": 0xFAFAD2,
+    "lightgreen": 0x90EE90,
+    "lightgrey": 0xD3D3D3,
+    "lightpink": 0xFFB6C1,
+    "lightsalmon": 0xFFA07A,
+    "lightseagreen": 0x20B2AA,
+    "lightskyblue": 0x87CEFA,
+    "lightslategray": 0x778899,
+    "lightsteelblue": 0xB0C4DE,
+    "lightyellow": 0xFFFFE0,
+    "lime": 0x00FF00,
+    "limegreen": 0x32CD32,
+    "linen": 0xFAF0E6,
+    "magenta": 0xFF00FF,
+    "maroon": 0x800000,
+    "mediumaquamarine": 0x66CDAA,
+    "mediumblue": 0x0000CD,
+    "mediumorchid": 0xBA55D3,
+    "mediumpurple": 0x9370DB,
+    "mediumseagreen": 0x3CB371,
+    "mediumslateblue": 0x7B68EE,
+    "mediumspringgreen": 0x00FA9A,
+    "mediumturquoise": 0x48D1CC,
+    "mediumvioletred": 0xC71585,
+    "midnightblue": 0x191970,
+    "mintcream": 0xF5FFFA,
+    "mistyrose": 0xFFE4E1,
+    "moccasin": 0xFFE4B5,
+    "navajowhite": 0xFFDEAD,
+    "navy": 0x000080,
+    "oldlace": 0xFDF5E6,
+    "olive": 0x808000,
+    "olivedrab": 0x6B8E23,
+    "orange": 0xFFA500,
+    "orangered": 0xFF4500,
+    "orchid": 0xDA70D6,
+    "palegoldenrod": 0xEEE8AA,
+    "palegreen": 0x98FB98,
+    "paleturquoise": 0xAFEEEE,
+    "paleVioletred": 0xDB7093,
+    "papayawhip": 0xFFEFD5,
+    "peachpuff": 0xFFDAB9,
+    "peru": 0xCD853F,
+    "pink": 0xFFC0CB,
+    "plaid": 0xCC5533,
+    "plum": 0xDDA0DD,
+    "powderblue": 0xB0E0E6,
+    "purple": 0x800080,
+    "red": 0xFF0000,
+    "rosybrown": 0xBC8F8F,
+    "royalblue": 0x4169E1,
+    "saddlebrown": 0x8B4513,
+    "salmon": 0xFA8072,
+    "sandybrown": 0xF4A460,
+    "seagreen": 0x2E8B57,
+    "seashell": 0xFFF5EE,
+    "sienna": 0xA0522D,
+    "silver": 0xC0C0C0,
+    "skyblue": 0x87CEEB,
+    "slateblue": 0x6A5ACD,
+    "slategray": 0x708090,
+    "snow": 0xFFFAFA,
+    "springgreen": 0x00FF7F,
+    "steelblue": 0x4682B4,
+    "tan": 0xD2B48C,
+    "teal": 0x008080,
+    "thistle": 0xD8BFD8,
+    "tomato": 0xFF6347,
+    "turquoise": 0x40E0D0,
+    "violet": 0xEE82EE,
+    "wheat": 0xF5DEB3,
+    "white": 0xFFFFFF,
+    "whitesmoke": 0xF5F5F5,
+    "yellow": 0xFFFF00,
+    "yellowgreen": 0x9ACD32,
+}
+
+
+class WaaaghLampGUI(QMainWindow):
+    def __init__(self):
+        # super().__init__()
+        super(WaaaghLampGUI, self).__init__()
+
+        self.setWindowIcon(QIcon("res/waaagh_icon.png"))
+
+        self.state = {
+            "mode": 0,
+            "color": 0,
+        }
+
+        self.setWindowTitle("WaaaghLamp GUI")
+        # self.setFixedSize(200, 100)
+
+        self.mode_button_group = QButtonGroup()
+        self.status_bar = QStatusBar()
+        self.status_line = QLabel()
+        self.status_bar.addWidget(self.status_line)
+
+        self.main_layout = QVBoxLayout()
+        self.add_mode_buttons(self.main_layout, MODES, self.set_mode)
+        splitter1 = QFrame()
+        splitter1.setFrameShape(QFrame.Shape.HLine)
+        splitter1.setFrameShadow(QFrame.Shadow.Sunken)
+        self.main_layout.addWidget(splitter1)
+
+        self.add_mode_buttons(self.main_layout, SPECIAL_MODES, self.set_mode)
+        splitter2 = QFrame()
+        splitter2.setFrameShape(QFrame.Shape.HLine)
+        splitter2.setFrameShadow(QFrame.Shadow.Sunken)
+        self.main_layout.addWidget(splitter2)
+
+        self.pallete_list = QComboBox()
+        self.populate_pallete_list()
+        self.pallete_list.currentTextChanged.connect(self.set_custom_mode)
+        self.main_layout.addWidget(self.pallete_list)
+
+        self.custom_modes_layout = QVBoxLayout()
+        self.custom_modes_layout.setDirection(QVBoxLayout.Direction.LeftToRight)
+        self.add_mode_buttons(self.custom_modes_layout, CUSTOM_MODES, self.set_custom_mode)
+        self.main_layout.addLayout(self.custom_modes_layout)
+        splitter3 = QFrame()
+        splitter3.setFrameShape(QFrame.Shape.HLine)
+        splitter3.setFrameShadow(QFrame.Shadow.Sunken)
+        self.main_layout.addWidget(splitter3)
+
+        self.exit_button = QPushButton("Exit")
+        self.exit_button.clicked.connect(self.exit)
+        self.main_layout.addWidget(self.exit_button)
+
+        container = QWidget()
+        container.setLayout(self.main_layout)
+
+        self.setCentralWidget(container)
+        self.setStatusBar(self.status_bar)
+        # self.setMinimumSize()
+        # self.setFixedWidth(self.width())
+        # self.setFixedHeight(self.height())
+
+    def populate_pallete_list(self):
+        for color in PALLETE.keys():
+            pixmap = QPixmap(16, 16)
+            pixmap.fill(PALLETE[color])
+            icon = QIcon(pixmap)
+            self.pallete_list.addItem(icon, color)
+
+    def add_mode_buttons(self, layout, modes, function=None):
+        for mode in modes.keys():
+            new_button = QPushButton(mode.capitalize())
+            new_button.setCheckable(True)
+            if new_button.text().lower() == "off":
+                new_button.setChecked(True)
+            new_button.clicked.connect(function)
+            self.mode_button_group.addButton(new_button, modes[mode])
+            layout.addWidget(new_button)
+
+    def set_mode(self):
+        print(self.mode_button_group.checkedId())
+        self.status_line.text = self.mode_button_group.checkedId()
+        self.send_message(self.mode_button_group.checkedId())
+
+    def parse_state(self):
+        pass
+
+    def set_custom_mode(self):
+        print(self.mode_button_group.checkedId())
+        self.status_line.text = self.mode_button_group.checkedId()
+        self.send_message(self.mode_button_group.checkedId(), PALLETE[self.pallete_list.currentText()])
+
+    def send_message(self, mode, color=0x000000):
+        message = {
+            "mode": mode,
+            "color": color,
+        }
+
+        try:
+            s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+            s.connect((WAL_HOST, WAL_PORT))
+            s.settimeout(5)
+            s.send(json.dumps(message).encode('utf-8'))
+        except Exception as e:
+            print(f"ERROR: {e}")
+
+        data = ""
+        try:
+            data = s.recv(1024)
+        except Exception as e:
+            print(f"ERROR: {e}")
+
+        if data != "":
+            try:
+                self.state = json.loads(data.decode())
+                recieved_mode = self.state["mode"]
+                recieved_color = self.state["color"]
+                if recieved_mode == mode and recieved_color == color:
+                    print(f"Mode {mode} with color {color} set successfully")
+                else:
+                    print(f"ERROR: Wrong answer from lamp")
+            except Exception as e:
+                print(f"ERROR: {e}")
+
+        s.close()
+
+    def exit(self):
+        print("Bye!")
+        exit(0)
+
+
+if __name__ == "__main__":
+    app = QApplication(sys.argv)
+    window = WaaaghLampGUI()
+    window.show()
+    app.exec()

requirements.txt

@@ -0,0 +1 @@
+PyQt6