From 9d9e5c7edc6d34ce25636d6174579fa28ee5e367 Mon Sep 17 00:00:00 2001
From: Kevin O'Connor <kevin@koconnor.net>
Date: Sun, 16 Aug 2020 14:08:38 -0400
Subject: [PATCH] klippy: Report "shutdown" category from get_state_message()

Reported by @Arksine.

Signed-off-by: Kevin O'Connor <kevin@koconnor.net>
---
 klippy/klippy.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/klippy/klippy.py b/klippy/klippy.py
index 838a43c0..6cac8099 100644
--- a/klippy/klippy.py
+++ b/klippy/klippy.py
@@ -70,6 +70,8 @@ class Printer:
             category = "ready"
         elif self.state_message == message_startup:
             category = "startup"
+        elif self.in_shutdown_state:
+            category = "shutdown"
         else:
             category = "error"
         return self.state_message, category