Further simplify for this example

This commit is contained in:
Danny Staple 2022-11-03 22:55:02 +00:00
parent 8e8783d9b2
commit 5a951d3b64

View File

@ -11,25 +11,22 @@ class RobotDisplay:
self.ble_connection = BleConnection(self.handle_data) self.ble_connection = BleConnection(self.handle_data)
self.line = "" self.line = ""
self.arena = None
self.display_closed = False self.display_closed = False
def handle_close(self, _): def handle_close(self, _):
self.display_closed = True self.display_closed = True
def handle_data(self, data): def handle_data(self, data):
line_part = data.decode("utf-8") self.line += data.decode("utf-8")
self.line += line_part
if not self.line.endswith("\n"): if not self.line.endswith("\n"):
return return
print(f"Received data: {self.line}") print(f"Received data: {self.line}")
data = json.loads(self.line) message = json.loads(self.line)
self.line = "" self.line = ""
if "arena" in data: if "arena" in message:
self.update(data) self.update_arena(message)
def update(self, arena): def update_arena(self, arena):
self.arena = arena
plt.gca().clear() plt.gca().clear()
for line in arena["arena"]: for line in arena["arena"]:
plt.gca().plot( plt.gca().plot(