neopixel: Simplify color_order parsing

Signed-off-by: Kevin O'Connor <kevin@koconnor.net>

1 files changed, 2 insertions, 5 deletions

diff --git a/klippy/extras/neopixel.py b/klippy/extras/neopixel.py
index ef728fc2..b6daeb4d 100644
--- a/klippy/extras/neopixel.py
+++ b/klippy/extras/neopixel.py
@@ -33,12 +33,9 @@ class PrinterNeoPixel:
             raise config.error("color_order does not match chain_count")
         color_indexes = []
         for lidx, co in enumerate(color_order):
-            rgb = "RGB"
-            if 'W' in co:
-                rgb = "RGBW"
-            if sorted(co) != sorted(rgb):
+            if sorted(co) not in (sorted("RGB"), sorted("RGBW")):
                 raise config.error("Invalid color_order '%s'" % (co,))
-            color_indexes.extend([(lidx, rgb.index(c)) for c in co])
+            color_indexes.extend([(lidx, "RGBW".index(c)) for c in co])
         self.color_map = list(enumerate(color_indexes))
         if len(self.color_map) > MAX_MCU_SIZE:
             raise config.error("neopixel chain too long")