mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-10-29 13:32:08 +00:00
b71abcfbf4
* quick econ fix * black * keys and feature flags * terminal name 👀 * some more replacements * some more replacements * edit pyproject * gst -> openbb * add example portfolios back to git * Update api from gst * sorry. skipping some tests * another round of names * another round of test edits * Missed some .gst refs and update timezone * water mark stuff * Fixing Names in terminal.spec and name of GTFF_DEFAULTS to OBBFF_DEFAULTS * fix more GST to OpenBB Terminal * Logging : merge conflicts with main * Revert wrong files Co-authored-by: Andrew <andrew.kenreich@gmail.com> Co-authored-by: DidierRLopes <dro.lopes@campus.fct.unl.pt> Co-authored-by: Chavithra PARANA <chavithra@gmail.com>
119 lines
3.4 KiB
Python
Executable File
119 lines
3.4 KiB
Python
Executable File
import logging
|
|
|
|
import disnake
|
|
|
|
from bots import imps
|
|
from openbb_terminal.decorators import log_start_end
|
|
from openbb_terminal.etf import stockanalysis_model
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
@log_start_end(log=logger)
|
|
def holdings_command(etf="", num: int = 15):
|
|
"""Display ETF Holdings. [Source: StockAnalysis]"""
|
|
|
|
if imps.DEBUG:
|
|
logger.debug("etfs")
|
|
|
|
holdings = stockanalysis_model.get_etf_holdings(etf.upper())
|
|
holdings = holdings.iloc[:num]
|
|
|
|
if holdings.empty:
|
|
raise Exception("No company holdings found!\n")
|
|
|
|
title = f"ETF: {etf.upper()} Holdings"
|
|
|
|
dindex = len(holdings.index)
|
|
if dindex > 15:
|
|
embeds: list = []
|
|
# Output
|
|
i, i2, end = 0, 0, 15
|
|
df_pg, embeds_img, images_list = [], [], []
|
|
while i < dindex:
|
|
df_pg = holdings.iloc[i:end]
|
|
df_pg.append(df_pg)
|
|
fig = imps.plot_df(
|
|
df_pg,
|
|
fig_size=(500, (40 + (40 * dindex))),
|
|
col_width=[1, 2, 2],
|
|
tbl_header=imps.PLT_TBL_HEADER,
|
|
tbl_cells=imps.PLT_TBL_CELLS,
|
|
font=imps.PLT_TBL_FONT,
|
|
row_fill_color=imps.PLT_TBL_ROW_COLORS,
|
|
paper_bgcolor="rgba(0, 0, 0, 0)",
|
|
)
|
|
fig.update_traces(cells=(dict(align=["center", "right"])))
|
|
imagefile = "etf-holdings.png"
|
|
imagefile = imps.save_image(imagefile, fig)
|
|
|
|
if imps.IMAGES_URL or not imps.IMG_HOST_ACTIVE:
|
|
image_link = imps.multi_image(imagefile)
|
|
images_list.append(imagefile)
|
|
else:
|
|
image_link = imps.multi_image(imagefile)
|
|
|
|
embeds_img.append(
|
|
f"{image_link}",
|
|
)
|
|
embeds.append(
|
|
disnake.Embed(
|
|
title=title,
|
|
colour=imps.COLOR,
|
|
),
|
|
)
|
|
i2 += 1
|
|
i += 15
|
|
end += 15
|
|
|
|
# Author/Footer
|
|
for i in range(0, i2):
|
|
embeds[i].set_author(
|
|
name=imps.AUTHOR_NAME,
|
|
url=imps.AUTHOR_URL,
|
|
icon_url=imps.AUTHOR_ICON_URL,
|
|
)
|
|
embeds[i].set_footer(
|
|
text=imps.AUTHOR_NAME,
|
|
icon_url=imps.AUTHOR_ICON_URL,
|
|
)
|
|
|
|
i = 0
|
|
for i in range(0, i2):
|
|
embeds[i].set_image(url=embeds_img[i])
|
|
|
|
i += 1
|
|
embeds[0].set_footer(text=f"Page 1 of {len(embeds)}")
|
|
choices = [
|
|
disnake.SelectOption(label="Home", value="0", emoji="🟢"),
|
|
]
|
|
|
|
output = {
|
|
"view": imps.Menu,
|
|
"title": title,
|
|
"embed": embeds,
|
|
"choices": choices,
|
|
"embeds_img": embeds_img,
|
|
"images_list": images_list,
|
|
}
|
|
else:
|
|
fig = imps.plot_df(
|
|
holdings,
|
|
fig_size=(500, (40 + (40 * dindex))),
|
|
col_width=[1, 2, 2],
|
|
tbl_header=imps.PLT_TBL_HEADER,
|
|
tbl_cells=imps.PLT_TBL_CELLS,
|
|
font=imps.PLT_TBL_FONT,
|
|
row_fill_color=imps.PLT_TBL_ROW_COLORS,
|
|
paper_bgcolor="rgba(0, 0, 0, 0)",
|
|
)
|
|
fig.update_traces(cells=(dict(align=["center", "right"])))
|
|
imagefile = imps.save_image("etf-holdings.png", fig)
|
|
|
|
output = {
|
|
"title": title,
|
|
"imagefile": imagefile,
|
|
}
|
|
|
|
return output
|