59 lines
1.5 KiB
Python
59 lines
1.5 KiB
Python
import json
|
|
import sys
|
|
import time
|
|
from PyQt5.QtWidgets import QApplication
|
|
from client_socket import SocketRobotArm
|
|
from gui.init import MainContentComponent
|
|
|
|
class MyApp:
|
|
with open("./robots.json", "r") as file:
|
|
robots = json.load(file)
|
|
|
|
def __init__(self):
|
|
self.startRobot()
|
|
self.startGui()
|
|
|
|
def get_status(self):
|
|
if self.robot_app:
|
|
return self.robot_app.get_status()
|
|
else:
|
|
return self.def_robot_status()
|
|
|
|
def def_robot_status(self):
|
|
return SocketRobotArm.Status[1]
|
|
|
|
def startGui(self):
|
|
app = QApplication(sys.argv)
|
|
mainWindow = MainContentComponent(
|
|
robotPanel={
|
|
"robots": self.robots,
|
|
"updateRobot": self.updateRobot,
|
|
"status": self.get_status,
|
|
},
|
|
statusPanel={
|
|
"status": self.get_status,
|
|
}
|
|
)
|
|
mainWindow.setWindowTitle("ROBOT GUI")
|
|
mainWindow.show()
|
|
|
|
sys.exit(app.exec_())
|
|
|
|
def startRobot(self):
|
|
self.robot_app = SocketRobotArm()
|
|
|
|
def updateRobot(self, robot):
|
|
if robot in self.robots:
|
|
selected_robot = robot
|
|
selected_robot = {"name": "test", "host": "127.0.0.1", "slave_id": 11}
|
|
|
|
if self.robot_app.status == "connected":
|
|
self.robot_app.close()
|
|
time.sleep(0.3)
|
|
else:
|
|
self.robot_app.connect(selected_robot["host"], selected_robot["slave_id"])
|
|
|
|
|
|
if __name__ == "__main__":
|
|
MyApp()
|