HEX check

termspark/helpers/hex.py

@@ -0,0 +1,15 @@
+import re
+from typing import Sequence, Union
+
+
+class HEX:
+    @staticmethod
+    def check(color: Union[str, Sequence[str], None]) -> bool:
+        if color is None:
+            return False
+
+        assert type(color) == str
+        regex = r"^#(?:[0-9a-fA-F]{3}){1,2}$"
+        match = re.match(regex, color)
+
+        return match is not None

tests/hex_test.py

@@ -0,0 +1,20 @@
+from termspark.helpers.hex import HEX
+
+
+class TestHEX:
+    def test_check(self):
+        assert HEX.check("#000000")
+        assert HEX.check("#ffffff")
+        assert HEX.check("#FFFFFF")
+        assert HEX.check("#2472c8")
+        assert HEX.check("#2472C8")
+        assert HEX.check("#000")
+        assert HEX.check("#fff")
+        assert HEX.check("#FFF")
+        assert HEX.check("#ffg") == False
+        assert HEX.check("#FFG") == False
+        assert HEX.check("#FFFFFFF") == False
+        assert HEX.check("#FF") == False
+        assert HEX.check("#") == False
+        assert HEX.check(None) == False
+        assert HEX.check("") == False