FIx for GetAllVintageSeriesResult.to_pd_data_frame (#67)

mb-jp · web-flow · commit 9c2ae576ffef · 2024-08-16T10:45:33.000+02:00
diff --git a/macrobond_data_api/common/types/get_all_vintage_series_result.py b/macrobond_data_api/common/types/get_all_vintage_series_result.py
@@ -1,4 +1,5 @@
 from dataclasses import dataclass
+from itertools import chain
 
 from typing import TYPE_CHECKING, Any, Dict, Sequence, overload, List
 
@@ -36,9 +37,12 @@ def to_pd_data_frame(self) -> "DataFrame":
         """
         import pandas  # pylint: disable=import-outside-toplevel
 
+        dates = list(set(chain.from_iterable((x.dates for x in self.series))))
+        dates.sort()
+
         df = pandas.DataFrame(
             {
-                **{"date": self.series[len(self.series) - 1].dates},
+                **{"date": dates},
             }
         )
 

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,5 @@`
`1`	`1`	`from dataclasses import dataclass`
	`2`	`+from itertools import chain`
`2`	`3`
`3`	`4`	`from typing import TYPE_CHECKING, Any, Dict, Sequence, overload, List`
`4`	`5`
`@@ -36,9 +37,12 @@ def to_pd_data_frame(self) -> "DataFrame":`
`36`	`37`	`"""`
`37`	`38`	`import pandas # pylint: disable=import-outside-toplevel`
`38`	`39`
	`40`	`+ dates = list(set(chain.from_iterable((x.dates for x in self.series))))`
	`41`	`+ dates.sort()`
	`42`	`+`
`39`	`43`	`df = pandas.DataFrame(`
`40`	`44`	`{`
`41`		`- **{"date": self.series[len(self.series) - 1].dates},`
	`45`	`+ **{"date": dates},`
`42`	`46`	`}`
`43`	`47`	`)`
`44`	`48`