openseries 1.5.4__py3-none-any.whl → 1.5.6__py3-none-any.whl

openseries/_common_model.py

@@ -38,6 +38,7 @@ from openseries.types import (
     CountriesType,
     DaysInYearType,
     LiteralBarPlotMode,
+    LiteralJsonOutput,
     LiteralLinePlotMode,
     LiteralNanMethod,
     LiteralPlotlyJSlib,
@@ -651,16 +652,18 @@ class _CommonModel(BaseModel):
 
     def to_json(
         self: Self,
+        what_output: LiteralJsonOutput,
         filename: str,
         directory: Optional[DirectoryPath] = None,
     ) -> list[dict[str, Union[str, bool, ValueType, list[str], list[float]]]]:
         """
         Dump timeseries data into a json file.
 
-        The label and tsdf parameters are deleted before the json file is saved
-
         Parameters
         ----------
+        what_output: LiteralJsonOutput
+            Choice on whether the raw values or the tsdf Dataframe values are
+            returned as json and exported as json file.
         filename: str
             Filename including filetype
         directory: DirectoryPath, optional
@@ -669,7 +672,7 @@ class _CommonModel(BaseModel):
         Returns
         -------
         list[Dict[str, Union[str, bool, ValueType, list[str], list[float]]]]
-            A list of dictionaries with the raw original data of the series
+            A list of dictionaries with the data of the series
 
         """
         if directory:
@@ -683,21 +686,31 @@ class _CommonModel(BaseModel):
         data = dict(self.__dict__)
         output = []
         if "label" in data:
+            if what_output == "tsdf":
+                values = self.tsdf.iloc[:, 0].tolist()
+            else:
+                values = list(cast(list[float], data.get("values")))
             for item in cleaner_list:
                 data.pop(item)
+            valuetype = cast(ValueType, data.get("valuetype")).value
+            data.update({"valuetype": valuetype})
+            data.update({"values": values})
             output.append(dict(data))
         else:
             for serie in cast(list[Any], data.get("constituents")):
+                if what_output == "tsdf":
+                    values = serie.tsdf.iloc[:, 0].tolist()
+                else:
+                    values = list(serie.values)
                 itemdata = dict(serie.__dict__)
                 for item in cleaner_list:
                     itemdata.pop(item)
+                valuetype = cast(ValueType, itemdata["valuetype"]).value
+                itemdata.update({"valuetype": valuetype})
+                itemdata.update({"values": values})
                 output.append(dict(itemdata))
 
-        with Path.open(
-            dirpath.joinpath(filename),
-            "w",
-            encoding="utf-8",
-        ) as jsonfile:
+        with dirpath.joinpath(filename).open(mode="w", encoding="utf-8") as jsonfile:
             dump(output, jsonfile, indent=2, sort_keys=False)
 
         return output

openseries/datefixer.py

@@ -62,7 +62,7 @@ def holiday_calendar(
     endyear += 1
     if startyear == endyear:
         endyear += 1
-    years = list(range(startyear, endyear))
+    years = list(range(int(startyear), int(endyear)))
 
     if isinstance(countries, str) and countries in list_supported_countries():
         staging = country_holidays(country=countries, years=years)
@@ -412,9 +412,11 @@ def do_resample_to_business_period_ends(
     data.index = Index(d.date() for d in DatetimeIndex(data.index))
 
     if newhead.index[0] not in data.index:
+        # noinspection PyUnreachableCode
         data = concat([data, newhead])
 
     if newtail.index[0] not in data.index:
+        # noinspection PyUnreachableCode
         data = concat([data, newtail])
 
     data = data.sort_index()

openseries/types.py

@@ -113,6 +113,7 @@ PlotlyLayoutType = dict[
 
 CaptorLogoType = dict[str, Union[str, float]]
 
+LiteralJsonOutput = Literal["values", "tsdf"]
 LiteralTrunc = Literal["before", "after", "both"]
 LiteralLinePlotMode = Literal[
     "lines",

{openseries-1.5.4.dist-info → openseries-1.5.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: openseries
-Version: 1.5.4
+Version: 1.5.6
 Summary: Tools for analyzing financial timeseries.
 Home-page: https://github.com/CaptorAB/openseries
 License: BSD-3-Clause
@@ -20,7 +20,7 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Topic :: Office/Business :: Financial :: Investment
-Requires-Dist: holidays (>=0.30,<1.0)
+Requires-Dist: holidays (>=0.30,<0.50)
 Requires-Dist: numpy (>=1.23.2,<=2.0.0)
 Requires-Dist: openpyxl (>=3.1.2,<4.0.0)
 Requires-Dist: pandas (>=2.1.2,<3.0.0)
@@ -302,7 +302,7 @@ make lint
 | `value_nan_handle`                 | `OpenTimeSeries`, `OpenFrame` | Fills `Nan` in a value series with the preceding non-Nan value.                                                                                          |
 | `return_nan_handle`                | `OpenTimeSeries`, `OpenFrame` | Replaces `Nan` in a return series with a 0.0 `float`.                                                                                                    |
 | `to_cumret`                        | `OpenTimeSeries`, `OpenFrame` | Converts a return series into a value series and/or resets a value series to be rebased from 1.0.                                                        |
-| `to_json`                          | `OpenTimeSeries`, `OpenFrame` | Method to export the raw unaltered data to a json file.                                                                                                  |
+| `to_json`                          | `OpenTimeSeries`, `OpenFrame` | Method to export object data to a json file.                                                                                                             |
 | `to_xlsx`                          | `OpenTimeSeries`, `OpenFrame` | Method to save the data in the .tsdf DataFrame to an Excel file.                                                                                         |
 | `value_to_ret`                     | `OpenTimeSeries`, `OpenFrame` | Converts a value series into a percentage return series.                                                                                                 |
 | `value_to_diff`                    | `OpenTimeSeries`, `OpenFrame` | Converts a value series into a series of differences.                                                                                                    |

{openseries-1.5.4.dist-info → openseries-1.5.6.dist-info}/RECORD

@@ -1,15 +1,15 @@
 openseries/__init__.py,sha256=W429Ojwa-wPgHV5PDAOQOBOAzPOR4wrVHRwdZQjRKcQ,41
-openseries/_common_model.py,sha256=ETr8vueNHrMC5dZGHP2V3DowIg-jzgmhdOwCI_4258M,74705
+openseries/_common_model.py,sha256=gHJfvxsOjrjGnuoTm62LRYho_VeEhRKmbY9xM6tiRUk,75497
 openseries/_risk.py,sha256=4ckiA-0-uuoUOsuc_uElUA_2rLS_U3xJyiva2BX4W1s,3300
-openseries/datefixer.py,sha256=ZOSPp4kLkMEsZv50GQaSo2vAEDVaXEr9iX3wTO7ZdB4,12378
+openseries/datefixer.py,sha256=DtluyFPfrnXTqRnSfYyNfAE4KzAOiFwgDW8bJM3NjBA,12470
 openseries/frame.py,sha256=SH1sPUFoJS71o2uVJBLoDl3rSC8AP7wQZw3baMD9T4U,74019
 openseries/load_plotly.py,sha256=L4A3Fa5Jk47FY7pcg0NPZ0gm0JD_uCXjVlGC7FWVSJg,1856
 openseries/plotly_captor_logo.json,sha256=F5nhMzEyxKywtjvQqMTKgKRCJQYMDIiBgDSxdte8Clo,178
 openseries/plotly_layouts.json,sha256=ahx8-dL4_RPzvHtBOX0SiL0AH7xQJzNRSDhGrSmU-Og,1429
 openseries/series.py,sha256=pIpiD3v8z7SyIppBatbsPuqOWCf3qI_us2145JnPUkQ,28313
 openseries/simulation.py,sha256=VYxc-e5VSyC55DdfACpQen-necYbhso-6RMyOhYX-5k,13905
-openseries/types.py,sha256=fSSg9hzQd9cTY8Q3MVdCOtoSu4zFxX4GSk6778LdCNo,7643
-openseries-1.5.4.dist-info/LICENSE.md,sha256=cPUabMxJ6-ziqzqS6aLGkR-ilIOKe_s3Qtyp0ioTmo0,1521
-openseries-1.5.4.dist-info/METADATA,sha256=JhTQclCLTBKzyUE3uD_P-Oa6Ayfu2pAgljK-yODn_YU,44222
-openseries-1.5.4.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-openseries-1.5.4.dist-info/RECORD,,
+openseries/types.py,sha256=yJmGZcBFwvMQQEaRVb0vhVMP463xu7MSydjWr12X38M,7689
+openseries-1.5.6.dist-info/LICENSE.md,sha256=cPUabMxJ6-ziqzqS6aLGkR-ilIOKe_s3Qtyp0ioTmo0,1521
+openseries-1.5.6.dist-info/METADATA,sha256=aE0BWMcUlTQgJxWNK52MCYkO9dFFccqzn4BpR6_R9bA,44223
+openseries-1.5.6.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+openseries-1.5.6.dist-info/RECORD,,