From c16f42ad0edfc4b4e2e27dd9ac5362349909b97c Mon Sep 17 00:00:00 2001
From: didier <dro.lopes@campus.fct.unl.pt>
Date: Mon, 8 Feb 2021 21:56:40 +0000
Subject: [PATCH] Add spachero website to discovery menu

---
 discovery/disc_menu.py    |  9 ++++++++-
 discovery/spachero_api.py | 19 +++++++++++++++++++
 due_diligence/dd_menu.py  |  2 +-
 3 files changed, 28 insertions(+), 2 deletions(-)
 create mode 100644 discovery/spachero_api.py

diff --git a/discovery/disc_menu.py b/discovery/disc_menu.py
index 0b9080dd24f..881d01d471e 100644
--- a/discovery/disc_menu.py
+++ b/discovery/disc_menu.py
@@ -7,6 +7,7 @@ from discovery import short_interest_api
 from discovery import seeking_alpha_api
 from discovery import fidelity_api
 from discovery import simply_wallst_api
+from discovery import spachero_api
 
 
 # -----------------------------------------------------------------------------------------------------------------------
@@ -21,6 +22,7 @@ def print_discovery():
     print("   map           S&P500 index stocks map [Finviz]")
     print("   sectors       show sectors performance [Alpha Vantage]")
     print("   sw            Simply Wall St. research data [Simply Wall St.]")
+    print("   spachero      Best website for SPACs research [SpacHero]")
     print("   gainers       show latest top gainers [Yahoo Finance]")
     print("   orders        Orders by Fidelity Customers [Fidelity]")
     print("   earnings      next earnings release dates [Seeking Alpha]")
@@ -36,7 +38,8 @@ def disc_menu():
     # Add list of arguments that the discovery parser accepts
     disc_parser = argparse.ArgumentParser(prog='discovery', add_help=False)
     disc_parser.add_argument('cmd', choices=['help', 'q', 'quit',
-                                             'map', 'sectors', 'gainers', 'orders', 'high_short', 'low_float', 'earnings', 'sw'])
+                                             'map', 'sectors', 'gainers', 'spacs', 'orders', 'spachero',
+                                             'high_short', 'low_float', 'earnings', 'sw'])
 
     print_discovery()
 
@@ -76,6 +79,10 @@ def disc_menu():
         elif ns_known_args.cmd == 'gainers':
             yahoo_finance_api.gainers(l_args)
 
+        # -------------------------------------------------- SPACHERO --------------------------------------------------
+        elif ns_known_args.cmd == 'spachero':
+            spachero_api.spachero(l_args)
+
         # --------------------------------------------------- FIDELITY ---------------------------------------------------
         elif ns_known_args.cmd == 'orders':
             fidelity_api.orders(l_args)
diff --git a/discovery/spachero_api.py b/discovery/spachero_api.py
new file mode 100644
index 00000000000..61d341c5c3c
--- /dev/null
+++ b/discovery/spachero_api.py
@@ -0,0 +1,19 @@
+import argparse
+import webbrowser
+
+# ---------------------------------------------------- SPACHERO ----------------------------------------------------
+def spachero(l_args):
+    parser = argparse.ArgumentParser(prog='spachero', description='''Great website for SPACs research''')
+
+    try:
+        (ns_parser, l_unknown_args) = parser.parse_known_args(l_args)
+    
+        if l_unknown_args:
+            print(f"The following args couldn't be interpreted: {l_unknown_args}")
+
+        webbrowser.open(f"https://www.spachero.com")
+        print("")
+
+    except SystemExit:
+        print("")
+        return
\ No newline at end of file
diff --git a/due_diligence/dd_menu.py b/due_diligence/dd_menu.py
index d908b674ac4..2bb5152bfc6 100644
--- a/due_diligence/dd_menu.py
+++ b/due_diligence/dd_menu.py
@@ -4,7 +4,7 @@ import config_bot as cfg
 import argparse
 import datetime
 from datetime import datetime
-from stock_market_helper_funcs import *
+from helper_funcs import *
 import pandas as pd
 import json
 import requests