Fix stocks/ta/fib datetime plotting (#4875)

This commit is contained in:
clairericks
2023-04-24 13:44:36 +01:00
committed by GitHub
parent d5917587bc
commit fc66bd7ff8
2 changed files with 4 additions and 4 deletions

View File

@@ -51,12 +51,12 @@ def calculate_fib_levels(
return pd.DataFrame(), pd.Timestamp(), pd.Timestamp(), 0, 0, ""
if start_date and end_date:
if start_date not in data.index:
date0 = data.index[data.index.get_loc(start_date, method="nearest")]
date0 = data.index[data.index.get_indexer([end_date], method="nearest")[0]]
console.print(f"Start date not in data. Using nearest: {date0}")
else:
date0 = start_date
if end_date not in data.index:
date1 = data.index[data.index.get_loc(end_date, method="nearest")]
date1 = data.index[data.index.get_indexer([end_date], method="nearest")[0]]
console.print(f"End date not in data. Using nearest: {date1}")
else:
date1 = end_date

View File

@@ -146,8 +146,8 @@ class Custom(PltTA):
"<b>0.65</b>",
"<b>1</b>",
]
min_date = min_date.to_pydatetime()
max_date = max_date.to_pydatetime()
min_date = pd.to_datetime(min_date).to_pydatetime()
max_date = pd.to_datetime(max_date).to_pydatetime()
self.df_fib = df_fib
fig.add_scatter(