eka/WaaaghLampGUI
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
main
WaaaghLampGUI/WaaaghLampGUI.py

337 lines
9.8 KiB
Python
Raw Permalink Blame History

#!/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, QMenu, QSystemTrayIcon
from PyQt6.QtGui import QPixmap, QIcon, QAction
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(250, 450)
# Установка иконки в трей
self.tray_icon = QSystemTrayIcon(self)
self.tray_icon.setIcon(QIcon("res/waaagh_icon.png"))
self.tray_menu = QMenu()
self.tray_icon.activated.connect(self.showMaximized)
# Создание действий для меню трея
minimize_action = QAction("Свернуть", self)
minimize_action.triggered.connect(self.showMinimized)
self.tray_menu.addAction(minimize_action)
exit_action = QAction("Выход", self)
exit_action.triggered.connect(self.close)
self.tray_menu.addAction(exit_action)
# Привязка меню к иконке трея
self.tray_icon.setContextMenu(self.tray_menu)
self.tray_icon.show()
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.close)
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):
def print_message(message):
print(message)
self.status_line.setText(message)
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_message(f"ERROR: {e}")
data = ""
try:
data = s.recv(1024)
except Exception as e:
print_message(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_message(f"Mode {mode} with color {color}")
else:
print_message(f"ERROR: Wrong answer from lamp")
except Exception as e:
print_message(f"ERROR: {e}")
s.close()
def close(self, event):
print("Bye!")
QApplication.quit()
if __name__ == "__main__":
app = QApplication(sys.argv)
mainWindow = WaaaghLampGUI()
mainWindow.show()
sys.exit(app.exec())
Reference in New Issue View Git Blame Copy Permalink