mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-10-27 04:31:13 +00:00
78f110b389
* Add lookup commands * main ETF commands * Add technical analysis to ETF * add prediction techniques to ETF * add comparison analysis to etf * allow to select nr of holdings on load * add shold and summary commands to ETF * minor improvs * screener on ETF menu * add etf/scr wsj commands * readd compare command * add sbc to etf/screener * rename chold and shold * fix rogue etf and reset * minor corrections * minor menu print improv * legend of weights and title of plot * refactor screener to screener and disc on ETF * add ETF scripts
125 lines
3.1 KiB
Python
125 lines
3.1 KiB
Python
"""StockAnalysis.com view functions"""
|
|
__docformat__ = "numpy"
|
|
|
|
import os
|
|
from typing import List
|
|
|
|
from tabulate import tabulate
|
|
from gamestonk_terminal import feature_flags as gtff
|
|
from gamestonk_terminal.etf import stockanalysis_model
|
|
from gamestonk_terminal.helper_funcs import export_data
|
|
|
|
|
|
def view_overview(symbol: str, export: str):
|
|
"""Print etf overview information
|
|
|
|
Parameters
|
|
----------
|
|
symbol:str
|
|
ETF symbols to display overview for
|
|
export:str
|
|
Format to export data
|
|
"""
|
|
|
|
if symbol.upper() not in stockanalysis_model.get_all_names_symbols()[0]:
|
|
print(f"{symbol.upper()} not found in ETFs\n")
|
|
return
|
|
|
|
data = stockanalysis_model.get_etf_overview(symbol)
|
|
|
|
print(tabulate(data, headers=data.columns, tablefmt="fancy_grid"), "\n")
|
|
|
|
export_data(export, os.path.dirname(os.path.abspath(__file__)), "overview", data)
|
|
|
|
|
|
def view_holdings(symbol: str, num_to_show: int, export: str):
|
|
"""
|
|
|
|
Parameters
|
|
----------
|
|
symbol: str
|
|
ETF symbol to show holdings for
|
|
num_to_show: int
|
|
Number of holdings to show
|
|
export: str
|
|
Format to export data
|
|
"""
|
|
|
|
data = stockanalysis_model.get_etf_holdings(symbol)
|
|
if gtff.USE_TABULATE_DF:
|
|
print(
|
|
tabulate(
|
|
data[:num_to_show],
|
|
headers=data.columns,
|
|
tablefmt="fancy_grid",
|
|
),
|
|
"\n",
|
|
)
|
|
else:
|
|
print(data.head(num_to_show).to_string(), "\n")
|
|
|
|
export_data(export, os.path.dirname(os.path.abspath(__file__)), "holdings", data)
|
|
|
|
|
|
def view_comparisons(symbols: List[str], export: str):
|
|
"""Show ETF comparisons
|
|
|
|
Parameters
|
|
----------
|
|
symbols: List[str]
|
|
List of ETF symbols
|
|
export: str
|
|
Format to export data
|
|
"""
|
|
|
|
etf_list = stockanalysis_model.get_all_names_symbols()[0]
|
|
for etf in symbols:
|
|
if etf not in etf_list:
|
|
print(f"{etf} not a known symbol.\n")
|
|
etf_list.remove(etf)
|
|
|
|
data = stockanalysis_model.compare_etfs(symbols)
|
|
if data.empty:
|
|
print("No data found for given ETFs\n")
|
|
return
|
|
if gtff.USE_TABULATE_DF:
|
|
print(tabulate(data, headers=data.columns, tablefmt="fancy_grid"), "\n")
|
|
else:
|
|
print(data.to_string())
|
|
export_data(export, os.path.dirname(os.path.abspath(__file__)), "overview", data)
|
|
|
|
|
|
def display_etf_by_name(name: str, limit: int, export: str):
|
|
"""Display ETFs matching search string. [Source: StockAnalysis]
|
|
|
|
Parameters
|
|
----------
|
|
name: str
|
|
String being matched
|
|
limit: int
|
|
Limit of ETFs to display
|
|
export: str
|
|
Export to given file type
|
|
|
|
"""
|
|
matching_etfs = stockanalysis_model.get_etfs_by_name(name)
|
|
|
|
if gtff.USE_TABULATE_DF:
|
|
print(
|
|
tabulate(
|
|
matching_etfs.head(limit),
|
|
showindex=False,
|
|
tablefmt="fancy_grid",
|
|
),
|
|
"\n",
|
|
)
|
|
else:
|
|
print(matching_etfs.head(limit).to_string(), "\n")
|
|
|
|
export_data(
|
|
export,
|
|
os.path.dirname(os.path.abspath(__file__)),
|
|
"ln_sa",
|
|
matching_etfs,
|
|
)
|