mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-10-28 13:01:38 +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>
91 lines
2.5 KiB
Python
91 lines
2.5 KiB
Python
"""Custom Indicator Models"""
|
|
__docformat__ = "numpy"
|
|
|
|
import logging
|
|
from typing import Any, Tuple
|
|
|
|
import pandas as pd
|
|
|
|
from openbb_terminal.decorators import log_start_end
|
|
from openbb_terminal.rich_config import console
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
@log_start_end(log=logger)
|
|
def calculate_fib_levels(
|
|
df_stock: pd.DataFrame,
|
|
period: int = 120,
|
|
open_date: Any = None,
|
|
close_date: Any = None,
|
|
) -> Tuple[pd.DataFrame, pd.Timestamp, pd.Timestamp, float, float]:
|
|
"""Calculate Fibonacci levels
|
|
|
|
Parameters
|
|
----------
|
|
df_stock : pd.DataFrame
|
|
Dataframe of prices
|
|
period : int
|
|
Days to look back for retracement
|
|
open_date : Any
|
|
Custom start date for retracement
|
|
close_date : Any
|
|
Custom end date for retracement
|
|
|
|
Returns
|
|
-------
|
|
df : pd.DataFrame
|
|
Dataframe of fib levels
|
|
min_date: pd.Timestamp
|
|
Date of min point
|
|
max_date: pd.Timestamp:
|
|
Date of max point
|
|
min_pr: float
|
|
Price at min point
|
|
max_pr: float
|
|
Price at max point
|
|
"""
|
|
if open_date and close_date:
|
|
if open_date not in df_stock.index:
|
|
date0 = df_stock.index[df_stock.index.get_loc(open_date, method="nearest")]
|
|
console.print(f"Start date not in df_stock. Using nearest: {date0}")
|
|
else:
|
|
date0 = open_date
|
|
if close_date not in df_stock.index:
|
|
date1 = df_stock.index[df_stock.index.get_loc(close_date, method="nearest")]
|
|
console.print(f"End date not in df_stock. Using nearest: {date1}")
|
|
else:
|
|
date1 = close_date
|
|
|
|
df_stock0 = df_stock.loc[date0, "Adj Close"]
|
|
df_stock1 = df_stock.loc[date1, "Adj Close"]
|
|
|
|
min_pr = min(df_stock0, df_stock1)
|
|
max_pr = max(df_stock0, df_stock1)
|
|
|
|
if min_pr == df_stock0:
|
|
min_date = date0
|
|
max_date = date1
|
|
else:
|
|
min_date = date1
|
|
max_date = date0
|
|
else:
|
|
data_to_use = df_stock.iloc[period:]["Adj Close"]
|
|
|
|
min_pr = data_to_use.min()
|
|
min_date = data_to_use.idxmin()
|
|
max_pr = data_to_use.max()
|
|
max_date = data_to_use.idxmax()
|
|
|
|
fib_levels = [0, 0.235, 0.382, 0.5, 0.618, 0.65, 1]
|
|
price_dif = max_pr - min_pr
|
|
|
|
levels = [round(max_pr - price_dif * f_lev, 2) for f_lev in fib_levels]
|
|
|
|
df = pd.DataFrame()
|
|
df["Level"] = fib_levels
|
|
df["Level"] = df["Level"].apply(lambda x: str(x * 100) + "%")
|
|
df["Price"] = levels
|
|
|
|
return df, min_date, max_date, min_pr, max_pr
|