From 4b873be3192a44bd6eda5c4885b34825d9644c8c Mon Sep 17 00:00:00 2001
From: Artem Veremey <artem@veremey.net>
Date: Sun, 28 Mar 2021 13:40:35 -0700
Subject: [PATCH] Updating ARK view

---
 gamestonk_terminal/discovery/ark_view.py      | 25 +++++++++++++++++--
 .../discovery/disc_controller.py              |  4 +--
 2 files changed, 25 insertions(+), 4 deletions(-)

diff --git a/gamestonk_terminal/discovery/ark_view.py b/gamestonk_terminal/discovery/ark_view.py
index 2244258c5a8..afa17263e36 100644
--- a/gamestonk_terminal/discovery/ark_view.py
+++ b/gamestonk_terminal/discovery/ark_view.py
@@ -2,6 +2,7 @@
 __docformat__ = "numpy"
 
 import argparse
+from typing import List
 from colorama import Fore, Style
 import pandas as pd
 
@@ -16,6 +17,19 @@ from gamestonk_terminal.discovery import ark_model
 
 
 def direction_color_red_green(val: str) -> str:
+    """Adds color tags to the Direction information: Buy -> Green, Sell -> Red
+
+    Parameters
+    ----------
+    val : str
+        Direction string - either Buy or Sell
+
+    Returns
+    -------
+    str
+        Direction string with color tags added
+    """
+
     if val == "Buy":
         ret = Fore.GREEN + val + Style.RESET_ALL
     elif val == "Sell":
@@ -26,7 +40,14 @@ def direction_color_red_green(val: str) -> str:
     return ret
 
 
-def ark_orders(l_args):
+def ark_orders_view(other_args: List[str]):
+    """Prints a table of the last N ARK Orders
+
+    Parameters
+    ----------
+    other_args : List[str]
+        argparse other args - ["-n", "10"]
+    """
     parser = argparse.ArgumentParser(
         add_help=False,
         prog="ARK Orders",
@@ -45,7 +66,7 @@ def ark_orders(l_args):
         help="Last N orders.",
     )
 
-    ns_parser = parse_known_args_and_warn(parser, l_args)
+    ns_parser = parse_known_args_and_warn(parser, other_args)
     if not ns_parser:
         return
 
diff --git a/gamestonk_terminal/discovery/disc_controller.py b/gamestonk_terminal/discovery/disc_controller.py
index fb754300b51..4cafd4bff4a 100644
--- a/gamestonk_terminal/discovery/disc_controller.py
+++ b/gamestonk_terminal/discovery/disc_controller.py
@@ -1,5 +1,6 @@
 import argparse
 from typing import List
+from prompt_toolkit.completion import NestedCompleter
 
 from gamestonk_terminal import feature_flags as gtff
 from gamestonk_terminal.helper_funcs import get_flair
@@ -16,7 +17,6 @@ from gamestonk_terminal.discovery import (
     unusual_whales_view,
     yahoo_finance_view,
 )
-from prompt_toolkit.completion import NestedCompleter
 
 
 class DiscoveryController:
@@ -126,7 +126,7 @@ class DiscoveryController:
 
     def call_ark_orders(self, other_args: List[str]):
         """Process ark_orders command"""
-        ark_view.ark_orders(other_args)
+        ark_view.ark_orders_view(other_args)
 
     def call_up_earnings(self, other_args: List[str]):
         """Process up_earnings command"""