From 4d26519cb11b9e0cc7afaa76e9a7c76cf46358d1 Mon Sep 17 00:00:00 2001 From: Laurier Loiselle Date: Mon, 30 Jan 2023 16:10:17 -0500 Subject: [PATCH] wv_colors: manage SingleColor case better --- src/wireviz/wv_colors.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/wireviz/wv_colors.py b/src/wireviz/wv_colors.py index be36ae2..162c158 100644 --- a/src/wireviz/wv_colors.py +++ b/src/wireviz/wv_colors.py @@ -101,6 +101,9 @@ class SingleColor: def html_padded(self): return ":".join([self.html] * padding_amount) + def __len__(self): + return 1 + def __bool__(self): return self._code_en is not None @@ -166,7 +169,7 @@ class MultiColor: def __getitem__(self, idx): if not self: - return None + return SingleColor(None) else: return self.colors[idx % len(self)]