From ff865cb2b9461856ccbda5b4c409abd2a54fc93a Mon Sep 17 00:00:00 2001
From: gouravkr <gourav2711@gmail.com>
Date: Mon, 21 Mar 2022 20:47:55 +0530
Subject: [PATCH] Added expand function, not fully working yet

---
 fincal/fincal.py | 26 +++++++++++++++++++++++++-
 1 file changed, 25 insertions(+), 1 deletion(-)

diff --git a/fincal/fincal.py b/fincal/fincal.py
index da830e6..9117ec8 100644
--- a/fincal/fincal.py
+++ b/fincal/fincal.py
@@ -7,7 +7,7 @@ from typing import Iterable, List, Literal, Mapping, Union
 
 from dateutil.relativedelta import relativedelta
 
-from .core import AllFrequencies, Series, TimeSeriesCore, date_parser
+from .core import AllFrequencies, Frequency, Series, TimeSeriesCore, date_parser
 from .utils import (
     FincalOptions,
     _find_closest_date,
@@ -535,6 +535,30 @@ class TimeSeries(TimeSeriesCore):
 
         return max_drawdown
 
+    def expand(
+        self, to_frequency: Literal["D", "W", "M", "Q", "H"], method: Literal["ffill", "bfill", "interpolate"]
+    ) -> TimeSeries:
+        try:
+            to_frequency: Frequency = getattr(AllFrequencies, to_frequency)
+        except AttributeError:
+            raise ValueError(f"Invalid argument for to_frequency {to_frequency}")
+
+        if to_frequency.days >= self.frequency.days:
+            raise ValueError("TimeSeries can be only expanded to a higher frequency")
+
+        new_dates = create_date_series(self.start_date, self.end_date, frequency=to_frequency.symbol)
+        new_ts: dict = {dt: self.data.get(dt, None) for dt in new_dates}
+        output_ts: TimeSeries = TimeSeries(new_ts, frequency=to_frequency.symbol)
+
+        if method == "ffill":
+            output_ts.ffill(inplace=True)
+        elif method == "bfill":
+            output_ts.bfill(inplace=True)
+        else:
+            raise NotImplementedError(f"Method {method} not implemented")
+
+        return output_ts
+
 
 if __name__ == "__main__":
     date_series = [