Moved error and markdown to specific files

Author: lainq

src/arguments/error.py

@@ -0,0 +1,20 @@
+from termcolor import colored
+
+
+class SearchError():
+    def __init__(self, error_statement, suggestion="Try again"):
+        # the error statement
+        self.error_statement = error_statement
+
+        # the suggestion statement
+        self.suggestion = suggestion
+
+        self.evoke_search_error(self.error_statement)
+
+    def evoke_search_error(self, error_statement):
+        print_text = [
+            colored(error_statement, 'red'),
+            colored(self.suggestion, 'green')
+        ]
+        for text_to_print in print_text:
+            print(text_to_print)

src/arguments/markdown.py

@@ -0,0 +1,26 @@
+from rich.markdown import Markdown
+from rich.console import Console
+import html as html
+
+class MarkdownRenderer(object):
+    def __init__(self, markdown_text, console_print=True):
+        assert isinstance(markdown_text, str), "Expected a string"
+
+        markdown_text = html.unescape(markdown_text)
+        self.markdown_text = markdown_text
+        self.do_console_print = bool(console_print)
+
+        self.console = Console()  # rich console
+
+        self.render = self.print_mark_down_text()
+
+    def print_mark_down_text(self):
+        rendered_markdown = Markdown(self.markdown_text)
+
+        if self.do_console_print:
+            self.console.print(rendered_markdown)
+
+        return rendered_markdown
+
+    def __repr__(self):
+        return str(self.render)

src/arguments/search.py

@@ -2,9 +2,11 @@
 import argparse
 import webbrowser
 from termcolor import colored
-from .utility import Utility, SearchError
 import sys as sys
 
+from .error import SearchError
+from .utility import Utility
+
 
 class Prompt():
     def __init__(self, message):

src/arguments/utility.py

@@ -1,61 +1,12 @@
-import requests
-import html
 from termcolor import colored
+import requests
 from rich.console import Console
-from rich.markdown import Markdown
 import sys as sys
 
-# the rich console
-console = Console()
-
-# render markdown text in the terminal
-# def print_markdown(markdown):
-#     md = Markdown(markdown)
-#     console.print(md)
-
-
-class MarkdownRenderer(object):
-    def __init__(self, markdown_text, console_print=True):
-        assert isinstance(markdown_text, str), "Expected a string"
-
-        markdown_text = html.unescape(markdown_text)
-        self.markdown_text = markdown_text
-        self.do_console_print = bool(console_print)
-
-        self.console = Console()  # rich console
-
-        self.render = self.print_mark_down_text()
-
-    def print_mark_down_text(self):
-        rendered_markdown = Markdown(self.markdown_text)
-
-        if self.do_console_print:
-            self.console.print(rendered_markdown)
-
-        return rendered_markdown
-
-    def __repr__(self):
-        return str(self.render)
-
-
-class SearchError():
-    def __init__(self, error_statement, suggestion="Try again"):
-        # the error statement
-        self.error_statement = error_statement
-
-        # the suggestion statement
-        self.suggestion = suggestion
-
-        self.evoke_search_error(self.error_statement)
-
-    def evoke_search_error(self, error_statement):
-        print_text = [
-            colored(error_statement, 'red'),
-            colored(self.suggestion, 'green')
-        ]
-        for text_to_print in print_text:
-            print(text_to_print)
+from .error import SearchError
+from .markdown import MarkdownRenderer
 
+console = Console()
 
 class Utility():
     def __init__(self):