Files
OpenBB/stock_market_fundamental_analysis.py

103 lines
4.2 KiB
Python

import FundamentalAnalysis as fa
from alpha_vantage.fundamentaldata import FundamentalData
import config_bot as cfg
import argparse
from stock_market_helper_funcs import *
# ---------------------------------------------------- RATINGS ----------------------------------------------------
def ratings(l_args, s_ticker):
parser = argparse.ArgumentParser(prog='ratings',
description=""" Gives information about the rating of a company which includes
i.a. the company rating and recommendation as well as ratings based
on a variety of ratios.""")
try:
(ns_parser, l_unknown_args) = parser.parse_known_args(l_args)
except SystemExit:
print("")
return
if l_unknown_args:
print(f"The following args couldn't be interpreted: {l_unknown_args}")
try:
df_ratings = fa.rating(s_ticker, cfg.API_KEY_FINANCIALMODELINGPREP)
print(df_ratings)
print("")
except:
print("")
# ---------------------------------------------------- INCOME_STATEMENT ----------------------------------------------------
def income_statement(l_args, s_ticker):
parser = argparse.ArgumentParser(prog='income',
description=""" """)
parser.add_argument('-n', "--num", action="store", dest="n_num", type=check_positive, default=1, help='Number of informations')
parser.add_argument('-q', action="store_true", default=False, dest="b_quarter", help='Quarter fundamental data')
try:
(ns_parser, l_unknown_args) = parser.parse_known_args(l_args)
except SystemExit:
print("")
return
if l_unknown_args:
print(f"The following args couldn't be interpreted: {l_unknown_args}")
try:
fd = FundamentalData(key=cfg.API_KEY_ALPHAVANTAGE, output_format='pandas')
if ns_parser.b_quarter:
df_fd, d_fd_metadata = fd.get_income_statement_quarterly(symbol=s_ticker)
else:
df_fd, d_fd_metadata = fd.get_income_statement_annual(symbol=s_ticker)
df_fd = df_fd.set_index('fiscalDateEnding')
df_fd = df_fd.head(n=ns_parser.n_num).T
df_fd = df_fd.mask(df_fd.astype(object).eq(ns_parser.n_num*['None'])).dropna()
df_fd = df_fd.mask(df_fd.astype(object).eq(ns_parser.n_num*['0'])).dropna()
df_fd = df_fd.applymap(lambda x: long_number_format(x))
df_fd.index = [''.join(' ' + char if char.isupper() else char.strip() for char in idx).strip() for idx in df_fd.index.tolist()]
print(df_fd)
print("")
except:
print("")
return
# ---------------------------------------------------- CASH_FLOW ----------------------------------------------------
def cash_flow(l_args, s_ticker):
parser = argparse.ArgumentParser(prog='cash',
description=""" """)
parser.add_argument('-n', "--num", action="store", dest="n_num", type=check_positive, default=1, help='Number of informations')
parser.add_argument('-q', action="store_true", default=False, dest="b_quarter", help='Quarter fundamental data')
try:
(ns_parser, l_unknown_args) = parser.parse_known_args(l_args)
except SystemExit:
print("")
return
if l_unknown_args:
print(f"The following args couldn't be interpreted: {l_unknown_args}")
try:
fd = FundamentalData(key=cfg.API_KEY_ALPHAVANTAGE, output_format='pandas')
if ns_parser.b_quarter:
df_fd, d_fd_metadata = fd.get_cash_flow_quarterly(symbol=s_ticker)
else:
df_fd, d_fd_metadata = fd.get_cash_flow_annual(symbol=s_ticker)
df_fd = df_fd.set_index('fiscalDateEnding')
df_fd = df_fd.head(n=ns_parser.n_num).T
df_fd = df_fd.mask(df_fd.astype(object).eq(ns_parser.n_num*['None'])).dropna()
df_fd = df_fd.mask(df_fd.astype(object).eq(ns_parser.n_num*['0'])).dropna()
df_fd = df_fd.applymap(lambda x: long_number_format(x))
df_fd.index = [''.join(' ' + char if char.isupper() else char.strip() for char in idx).strip() for idx in df_fd.index.tolist()]
print(df_fd)
print("")
except:
print("")
return