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",
"spacs",
"orders",
"spachero",
"high_short",
"low_float",
"up_earnings",
"simply_wallst",
"uwhales",
"mill",
],
)
print_discovery()
# Loop forever and ever
while True:
# Get input command from user
as_input = input(f"{get_flair()} (disc)> ")
# Parse fundamental analysis command of the list of possible commands
try:
(ns_known_args,
l_args) = disc_parser.parse_known_args(as_input.split())
except SystemExit:
print("The command selected doesn't exist\n")
continue
if ns_known_args.cmd == "help":
print_discovery()
elif ns_known_args.cmd == "q":
# Just leave the DISC menu
return False
elif ns_known_args.cmd == "quit":
# Abandon the program
return True
elif ns_known_args.cmd == "map":
finviz_api.map_sp500(l_args)
elif ns_known_args.cmd == "sectors":
alpha_vantage_api.sectors(l_args)
elif ns_known_args.cmd == "gainers":
yahoo_finance_api.gainers(l_args)
elif ns_known_args.cmd == "spachero":
spachero_api.spachero(l_args)
elif ns_known_args.cmd == "uwhales":
unusual_whales_api.unusual_whales(l_args)
elif ns_known_args.cmd == "orders":
fidelity_api.orders(l_args)
elif ns_known_args.cmd == "simply_wallst":
simply_wallst_api.simply_wallst(l_args)
elif ns_known_args.cmd == "up_earnings":
seeking_alpha_api.earnings_release_dates(l_args)
elif ns_known_args.cmd == "high_short":
short_interest_api.high_short_interest(l_args)
elif ns_known_args.cmd == "low_float":
short_interest_api.low_float(l_args)
else:
print("Command not recognized!")
def test_orders(self):
orders([])