mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-10-27 12:40:35 +00:00
137 lines
5.6 KiB
Python
137 lines
5.6 KiB
Python
import FundamentalAnalysis as fa
|
|
from alpha_vantage.fundamentaldata import FundamentalData
|
|
import config_bot as cfg
|
|
import argparse
|
|
import datetime
|
|
from datetime import datetime
|
|
from stock_market_helper_funcs import *
|
|
import pandas as pd
|
|
import json
|
|
import requests
|
|
from pandas.io.json import json_normalize
|
|
|
|
from technical_analysis import overlap as ta_overlap
|
|
from technical_analysis import momentum as ta_momentum
|
|
from technical_analysis import trend as ta_trend
|
|
from technical_analysis import volatility as ta_volatility
|
|
from technical_analysis import volume as ta_volume
|
|
|
|
# -----------------------------------------------------------------------------------------------------------------------
|
|
def print_technical_analysis(s_ticker, s_start, s_interval):
|
|
""" Print help """
|
|
|
|
s_intraday = (f'Intraday {s_interval}', 'Daily')[s_interval == "1440min"]
|
|
|
|
if s_start:
|
|
print(f"\n{s_intraday} Stock: {s_ticker} (from {s_start.strftime('%Y-%m-%d')})")
|
|
else:
|
|
print(f"\n{s_intraday} Stock: {s_ticker}")
|
|
|
|
print("\nTechnical Analysis:") # https://github.com/twopirllc/pandas-ta
|
|
print(" help show this technical analysis menu again")
|
|
print(" q quit this menu, and shows back to main menu")
|
|
print(" quit quit to abandon program")
|
|
print("overlap:")
|
|
print(" ema exponential moving average")
|
|
print(" sma simple moving average")
|
|
if s_interval != "1440min":
|
|
print(" vwap volume weighted average price")
|
|
print("momentum:")
|
|
print(" cci commodity channel index")
|
|
print(" macd moving average convergence/divergence")
|
|
print(" rsi relative strength index")
|
|
print(" stoch stochastic oscillator")
|
|
print("trend:")
|
|
print(" adx average directional movement index")
|
|
print(" aroon aroon indicator")
|
|
print("volatility:")
|
|
print(" bbands bollinger bands")
|
|
print("volume:")
|
|
print(" ad chaikin accumulation/distribution line values")
|
|
print(" obv on balance volume")
|
|
print("")
|
|
|
|
|
|
# ---------------------------------------------------- MENU ----------------------------------------------------
|
|
def ta_menu(df_stock, s_ticker, s_start, s_interval):
|
|
|
|
# Add list of arguments that the technical analysis parser accepts
|
|
ta_parser = argparse.ArgumentParser(prog='technical_analysis', add_help=False)
|
|
ta_parser.add_argument('cmd', choices=['help', 'q', 'quit',
|
|
'ema', 'sma', 'vwap', # overlap
|
|
'cci', 'macd', 'rsi', 'stoch', # momentum
|
|
'adx', 'aroon', # trend
|
|
'bbands', # volatility
|
|
'ad', 'obv']) # volume
|
|
|
|
print_technical_analysis(s_ticker, s_start, s_interval)
|
|
|
|
# Loop forever and ever
|
|
while True:
|
|
# Get input command from user
|
|
as_input = input('> ')
|
|
|
|
# Parse fundamental analysis command of the list of possible commands
|
|
try:
|
|
(ns_known_args, l_args) = ta_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_technical_analysis(s_ticker, s_start, s_interval)
|
|
|
|
elif ns_known_args.cmd == 'q':
|
|
# Just leave the FA menu
|
|
return False
|
|
|
|
elif ns_known_args.cmd == 'quit':
|
|
# Abandon the program
|
|
return True
|
|
|
|
# -------------------------------------------------- OVERLAP --------------------------------------------------
|
|
elif ns_known_args.cmd == 'ema':
|
|
ta_overlap.ema(l_args, s_ticker, s_interval, df_stock)
|
|
|
|
elif ns_known_args.cmd == 'sma':
|
|
ta_overlap.sma(l_args, s_ticker, s_interval, df_stock)
|
|
|
|
elif ns_known_args.cmd == 'vwap':
|
|
ta_overlap.vwap(l_args, s_ticker, s_interval, df_stock)
|
|
|
|
# --------------------------------------------------- MOMENTUM ---------------------------------------------------
|
|
elif ns_known_args.cmd == 'cci':
|
|
ta_momentum.cci(l_args, s_ticker, s_interval, df_stock)
|
|
|
|
elif ns_known_args.cmd == 'macd':
|
|
ta_momentum.macd(l_args, s_ticker, s_interval, df_stock)
|
|
|
|
elif ns_known_args.cmd == 'rsi':
|
|
ta_momentum.rsi(l_args, s_ticker, s_interval, df_stock)
|
|
|
|
elif ns_known_args.cmd == 'stoch':
|
|
ta_momentum.stoch(l_args, s_ticker, s_interval, df_stock)
|
|
|
|
# ---------------------------------------------------- TREND ----------------------------------------------------
|
|
elif ns_known_args.cmd == 'adx':
|
|
ta_trend.adx(l_args, s_ticker, s_interval, df_stock)
|
|
|
|
elif ns_known_args.cmd == 'aroon':
|
|
ta_trend.aroon(l_args, s_ticker, s_interval, df_stock)
|
|
|
|
# -------------------------------------------------- VOLATILITY --------------------------------------------------
|
|
elif ns_known_args.cmd == 'bbands':
|
|
ta_volatility.bbands(l_args, s_ticker, s_interval, df_stock)
|
|
|
|
# ---------------------------------------------------- VOLUME ----------------------------------------------------
|
|
elif ns_known_args.cmd == 'ad':
|
|
ta_volume.ad(l_args, s_ticker, s_interval, df_stock)
|
|
|
|
elif ns_known_args.cmd == 'obv':
|
|
ta_volume.obv(l_args, s_ticker, s_interval, df_stock)
|
|
|
|
# ------------------------------------------------------------------------------------------------------------
|
|
else:
|
|
print("Command not recognized!")
|