jerry/OpenBB
1
0
Fork 0
mirror of https://github.com/OpenBB-finance/OpenBB.git synced 2026-10-27 12:40:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

[BugFix] FRED Release Table: Fix Date Parsing For Format Variation (#6608)

Browse Source 
* fix date parsing to handle additional format types

* use date()

* black
This commit is contained in:
Danglewood
2024-08-03 17:15:22 -07:00
committed by GitHub
parent 804a67f24f
commit d5dfe8430d
1 changed files with 22 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
openbb_platform/providers/fred/openbb_fred/models/release_table.py
View File

@@ -189,15 +189,29 @@ class FredReleaseTableFetcher(
if "line" in df.columns:
df["line"] = df.line.astype(int)
# Some dates are in the format 'Jan 2021' and others are '2021-01-01'.
try:
df["observation_date"] = to_datetime(
df["observation_date"], format="%b %Y"
).dt.date
except ValueError:
df["observation_date"] = to_datetime(
df["observation_date"], format="%Y-%m-%d"
).dt.date
def apply_date_format(x):
"""Apply the date format."""
x = x.replace(" ", "-")
if x.startswith("Q"):
new_x = x.split("-")[-1]
q_dict = {
"Q1": "-03-31",
"Q2": "-06-30",
"Q3": "-09-30",
"Q4": "-12-31",
}
return new_x + q_dict[x.split("-")[0]]
try:
return to_datetime(x).date()
except ValueError:
try:
return to_datetime(x, format="%b-%Y").date()
except ValueError:
return x
df["observation_date"] = df.observation_date.apply(apply_date_format)
children = (
df.groupby("parent_id")["element_id"]
.apply(lambda x: x.sort_values().unique().tolist())