From 014c06f8a20dd931fcd41e0068150dadafc7f74f Mon Sep 17 00:00:00 2001 From: Arksine Date: Tue, 23 Jun 2020 07:56:50 -0400 Subject: klippy: return message category in get_state_message() Signed-off-by: Eric Callahan --- klippy/klippy.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) (limited to 'klippy/klippy.py') diff --git a/klippy/klippy.py b/klippy/klippy.py index 1da0a028..cd22ace9 100644 --- a/klippy/klippy.py +++ b/klippy/klippy.py @@ -65,7 +65,13 @@ class Printer: def get_reactor(self): return self.reactor def get_state_message(self): - return self.state_message + if self.state_message == message_ready: + category = "ready" + elif self.state_message == message_startup: + category = "startup" + else: + category = "error" + return self.state_message, category def is_shutdown(self): return self.in_shutdown_state def _set_state(self, msg): -- cgit v1.2.3-70-g09d2