Minor improvements

dahlia-lib · Dec 1, 2022 · b4f9db7 · b4f9db7
1 parent ebb7d41
commit b4f9db7
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/dahlia/dahlia.py b/src/dahlia/dahlia.py
@@ -59,7 +59,10 @@ def __hash__(self) -> int:
         return hash((self.depth, self.no_reset, self.marker))
 
     def __repr__(self) -> str:
-        return f"Dahlia(depth={self.depth}, no_reset={self.no_reset}, marker={self.marker!r})"
+        return (
+            f"Dahlia(depth={self.depth}, "
+            f"no_reset={self.no_reset}, marker={self.marker!r})"
+        )
 
     @property
     def depth(self) -> int:
@@ -116,7 +119,7 @@ def convert(self, string: str) -> str:
         if not (string.endswith(self.__reset) or self.no_reset):
             string += self.__reset
         for code, bg, color in _find_codes(string, self.__patterns):
-            string = string.replace(code, self.__get_ansi(color, bg))
+            string = string.replace(code, self.__get_ansi(color, bg=bg))
         return string
 
     def input(self, prompt: str) -> str:
@@ -193,7 +196,7 @@ def test(self) -> None:
             + "&r&ll&r&mm&r&nn&r&oo".replace("&", self.marker)
         )
 
-    def __get_ansi(self, code: str, bg: bool) -> str:
+    def __get_ansi(self, code: str, *, bg: bool) -> str:
         formats = BG_FORMAT_TEMPLATES if bg else FORMAT_TEMPLATES
         if len(code) == 6:
             r, g, b = (int(code[i : i + 2], 16) for i in range(0, 6, 2))