mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-10-29 05:21:48 +00:00
02901f1473
* first round of removing spaces * remove more print statemnts * remove more spaces * bullbear color * spacc color * improve stalker readability * delete more spaces * reddit colors * rever hood * flake8 * forgot this one * fix econometrics tests * fix econometrics tests * fix more tests Co-authored-by: James Maslek <jmaslek11@gmail.com>
115 lines
3.1 KiB
Python
115 lines
3.1 KiB
Python
"""Yahoo Finance view"""
|
|
__docformat__ = "numpy"
|
|
|
|
from typing import Optional, List
|
|
import logging
|
|
import os
|
|
|
|
import pandas as pd
|
|
from matplotlib import pyplot as plt
|
|
|
|
from openbb_terminal.config_terminal import theme
|
|
from openbb_terminal.config_plot import PLOT_DPI
|
|
from openbb_terminal.decorators import log_start_end
|
|
from openbb_terminal.etf import yfinance_model
|
|
from openbb_terminal.helper_funcs import (
|
|
export_data,
|
|
plot_autoscale,
|
|
print_rich_table,
|
|
is_valid_axes_count,
|
|
)
|
|
from openbb_terminal.rich_config import console
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
@log_start_end(log=logger)
|
|
def display_etf_weightings(
|
|
name: str,
|
|
raw: bool = False,
|
|
min_pct_to_display: float = 5,
|
|
export: str = "",
|
|
external_axes: Optional[List[plt.Axes]] = None,
|
|
):
|
|
"""Display sector weightings allocation of ETF. [Source: Yahoo Finance]
|
|
|
|
Parameters
|
|
----------
|
|
name: str
|
|
ETF name
|
|
raw: bool
|
|
Display sector weighting allocation
|
|
min_pct_to_display: float
|
|
Minimum percentage to display sector
|
|
export: str
|
|
Type of format to export data
|
|
external_axes : Optional[List[plt.Axes]], optional
|
|
External axes (1 axis is expected in the list), by default None
|
|
"""
|
|
sectors = yfinance_model.get_etf_sector_weightings(name)
|
|
if not sectors:
|
|
console.print("No data was found for that ETF\n")
|
|
return
|
|
|
|
holdings = pd.DataFrame(sectors, index=[0]).T
|
|
|
|
title = f"Sector holdings of {name}"
|
|
|
|
if raw:
|
|
console.print(f"\n{title}")
|
|
holdings.columns = ["% of holdings in the sector"]
|
|
print_rich_table(
|
|
holdings,
|
|
headers=list(holdings.columns),
|
|
show_index=True,
|
|
title="Sector Weightings Allocation",
|
|
)
|
|
|
|
else:
|
|
main_holdings = holdings[holdings.values > min_pct_to_display].to_dict()[
|
|
holdings.columns[0]
|
|
]
|
|
if len(main_holdings) < len(holdings):
|
|
main_holdings["Others"] = 100 - sum(main_holdings.values())
|
|
|
|
legend, values = zip(*main_holdings.items())
|
|
leg = [f"{le}\n{round(v,2)}%" for le, v in zip(legend, values)]
|
|
|
|
if external_axes is None:
|
|
_, ax = plt.subplots(figsize=plot_autoscale(), dpi=PLOT_DPI)
|
|
elif is_valid_axes_count(external_axes, 1):
|
|
(ax,) = external_axes
|
|
else:
|
|
return
|
|
|
|
ax.pie(
|
|
values,
|
|
labels=leg,
|
|
wedgeprops=theme.pie_wedgeprops,
|
|
colors=theme.get_colors(),
|
|
startangle=theme.pie_startangle,
|
|
)
|
|
ax.set_title(title)
|
|
theme.style_primary_axis(ax)
|
|
if external_axes is None:
|
|
theme.visualize_output()
|
|
|
|
export_data(export, os.path.dirname(os.path.abspath(__file__)), "weights", holdings)
|
|
|
|
|
|
@log_start_end(log=logger)
|
|
def display_etf_description(name: str):
|
|
"""Display ETF description summary. [Source: Yahoo Finance]
|
|
|
|
Parameters
|
|
----------
|
|
name: str
|
|
ETF name
|
|
"""
|
|
description = yfinance_model.get_etf_summary_description(name)
|
|
if not description:
|
|
console.print("No data was found for that ETF\n")
|
|
return
|
|
|
|
console.print(description, "\n")
|