refactor: Move ansi class into new style module

The Ansi class has moved from its own module into the remit of the style
module.

Author: nebbles

gitcommit/ansi.py

@@ -29,7 +29,7 @@
 from prompt_toolkit.history import FileHistory
 from prompt_toolkit.application.current import get_app
 from prompt_toolkit.styles import Style
-from gitcommit.ansi import ANSI as Ansi
+from gitcommit.style import Ansi
 from gitcommit.validators import (
     DescriptionValidator,
     TypeValidator,

gitcommit/gitcommit.py

@@ -3,3 +3,93 @@
 style = Style.from_dict(
     {"green": "#a0d762 bold", "red": "#e67061 bold", "command": "#f78ae0 bold"}
 )
+
+
+class Ansi:
+    """
+    This class contains the ANSI escape codes for printing to the console as
+    well as commonly used combinations of these codes.
+    """
+
+    reset = "\033[0m"
+    bold = "\033[1m"
+    faint = "\033[2m"
+    italic = "\033[3m"
+    underline = "\033[4m"
+    blink = "\033[5m"
+    inverse = "\033[7m"
+
+    class fg:
+        black = "\033[30m"
+        red = "\033[31m"
+        green = "\033[32m"
+        yellow = "\033[33m"
+        blue = "\033[34m"
+        magenta = "\033[35m"
+        cyan = "\033[36m"
+        white = "\033[37m"
+
+        grey = "\033[90m"
+        bright_red = "\033[91m"
+        bright_green = "\033[92m"
+        bright_yellow = "\033[93m"
+        bright_blue = "\033[94m"
+        bright_magenta = "\033[95m"
+        bright_cyan = "\033[96m"
+        bright_white = "\033[97m"
+
+    class bg:
+        black = "\033[40m"
+        red = "\033[41m"
+        green = "\033[42m"
+        yellow = "\033[43m"
+        blue = "\033[44m"
+        magenta = "\033[45m"
+        cyan = "\033[46m"
+        white = "\033[47m"
+
+        grey = "\033[100m"
+        bright_red = "\033[101m"
+        bright_green = "\033[102m"
+        bright_yellow = "\033[103m"
+        bright_blue = "\033[104m"
+        bright_magenta = "\033[105m"
+        bright_cyan = "\033[106m"
+        bright_white = "\033[107m"
+
+    @classmethod
+    def colour(cls, *args):
+        """Combines all args into string. Should pass colour modifiers in first followed by the string. Does not need reset sequence at the end."""
+        return "".join(args) + cls.reset
+
+    @classmethod
+    def b_green(cls, content):
+        return cls.fg.bright_green + cls.bold + str(content) + cls.reset
+
+    @classmethod
+    def b_red(cls, content):
+        return cls.fg.bright_red + cls.bold + str(content) + cls.reset
+
+    @classmethod
+    def b_yellow(cls, content):
+        return cls.fg.bright_yellow + cls.bold + str(content) + cls.reset
+
+    @classmethod
+    def b_blue(cls, content):
+        return cls.fg.bright_blue + cls.bold + str(content) + cls.reset
+
+    @classmethod
+    def print_ok(cls, content, end="\n"):
+        print(cls.b_green(content), end=end)
+
+    @classmethod
+    def print_error(cls, content, end="\n"):
+        print(cls.b_red(content), end=end)
+
+    @classmethod
+    def print_warning(cls, content, end="\n"):
+        print(cls.b_yellow(content), end=end)
+
+    @classmethod
+    def print_info(cls, content, end="\n"):
+        print(cls.b_blue(content), end=end)

gitcommit/style.py

@@ -4,10 +4,9 @@
 from prompt_toolkit import print_formatted_text
 from prompt_toolkit.formatted_text import FormattedText
 from prompt_toolkit.styles import Style
-from gitcommit.ansi import ANSI as Ansi
 import pyperclip  # pylint: disable=import-error
 from packaging import version
-from gitcommit.style import style
+from gitcommit.style import style, Ansi
 
 
 def get_github_tags():
@@ -51,9 +50,6 @@ def check_for_update():
         upgrade_command = "pip install --upgrade conventional-commit"
         pyperclip.copy(upgrade_command)
 
-        style = Style.from_dict(
-            {"green": "#a0d762 bold", "red": "#e67061 bold", "command": "#f78ae0 bold"}
-        )
         text = FormattedText(
             [
                 ("", "Version "),