refactor ModeManager

termspark/painter/mode_manager.py

@@ -1,22 +1,19 @@
+from importlib import import_module
 from typing import Union
 
-from .modes.hex import HEX
-from .modes.name import Name
-from .modes.rgb import RGB
+from termspark.painter.modes.mode import Mode
 
 
 class ModeManager:
-    __mode: Union[RGB, HEX, Name]
-
     def __init__(self, color: Union[str, tuple]):
         self._color = color
 
-        if RGB.check(color):
-            self.__mode = RGB(color)
-        elif HEX.check(color):
-            self.__mode = HEX(color)
-        else:
-            self.__mode = Name(color)
+        for mode in Mode.__subclasses__():
+            _module = import_module(f"termspark.painter.modes.{mode.__name__.lower()}")
+            _class = getattr(_module, mode.__name__)
+            if _class.check(color):
+                self.__mode = _class(color)
+                break
 
     def format(self):
         return self.__mode.format()

termspark/painter/modes/mode.py

@@ -3,6 +3,11 @@
 
 
 class Mode(ABC):
+    @staticmethod
+    @abstractmethod
+    def check(color: Union[str, tuple, None]) -> bool:
+        pass
+
     @abstractmethod
     def format(self) -> Union[str, bool]:
         pass

termspark/painter/modes/name.py

@@ -10,5 +10,9 @@ def __init__(self, color: Union[str, tuple]):
         assert type(color) == str
         self.__color = getattr(Color, color.upper(), color)
 
+    @staticmethod
+    def check(color: Union[str, tuple, None]) -> bool:
+        return True
+
     def format(self) -> Union[str, bool]:
         return RGB(self.__color).format()