411 lines
14 KiB
Python
411 lines
14 KiB
Python
import datetime
|
|
from collections import UserDict, UserList
|
|
from dataclasses import dataclass
|
|
from numbers import Number
|
|
from typing import Iterable, List, Literal, Mapping, Sequence, Tuple, Union
|
|
|
|
|
|
@dataclass
|
|
class FincalOptions:
|
|
date_format: str = "%Y-%m-%d"
|
|
closest: str = "before" # after
|
|
|
|
|
|
@dataclass(frozen=True)
|
|
class Frequency:
|
|
name: str
|
|
freq_type: str
|
|
value: int
|
|
days: int
|
|
symbol: str
|
|
|
|
|
|
class AllFrequencies:
|
|
D = Frequency("daily", "days", 1, 1, "D")
|
|
W = Frequency("weekly", "days", 7, 7, "W")
|
|
M = Frequency("monthly", "months", 1, 30, "M")
|
|
Q = Frequency("quarterly", "months", 3, 91, "Q")
|
|
H = Frequency("half-yearly", "months", 6, 182, "H")
|
|
Y = Frequency("annual", "years", 1, 365, "Y")
|
|
|
|
|
|
def _preprocess_timeseries(
|
|
data: Union[
|
|
Sequence[Iterable[Union[str, datetime.datetime, float]]],
|
|
Sequence[Mapping[str, Union[float, datetime.datetime]]],
|
|
Sequence[Mapping[Union[str, datetime.datetime], float]],
|
|
Mapping[Union[str, datetime.datetime], float],
|
|
],
|
|
date_format: str,
|
|
) -> List[Tuple[datetime.datetime, float]]:
|
|
"""Converts any type of list to the correct type"""
|
|
|
|
if isinstance(data, Sequence):
|
|
if isinstance(data[0], Mapping):
|
|
if len(data[0].keys()) == 2:
|
|
current_data = [tuple(i.values()) for i in data]
|
|
elif len(data[0].keys()) == 1:
|
|
current_data = [tuple(*i.items()) for i in data]
|
|
else:
|
|
raise TypeError("Could not parse the data")
|
|
current_data = _preprocess_timeseries(current_data, date_format)
|
|
|
|
elif isinstance(data[0], Sequence):
|
|
if isinstance(data[0][0], str):
|
|
current_data = []
|
|
for i in data:
|
|
row = datetime.datetime.strptime(i[0], date_format), i[1]
|
|
current_data.append(row)
|
|
elif isinstance(data[0][0], datetime.datetime):
|
|
current_data = [(i, j) for i, j in data]
|
|
else:
|
|
raise TypeError("Could not parse the data")
|
|
else:
|
|
raise TypeError("Could not parse the data")
|
|
|
|
elif isinstance(data, Mapping):
|
|
current_data = [(k, v) for k, v in data.items()]
|
|
current_data = _preprocess_timeseries(current_data, date_format)
|
|
|
|
else:
|
|
raise TypeError("Could not parse the data")
|
|
current_data.sort()
|
|
return current_data
|
|
|
|
|
|
def _preprocess_match_options(as_on_match: str, prior_match: str, closest: str) -> datetime.timedelta:
|
|
"""Checks the arguments and returns appropriate timedelta objects"""
|
|
|
|
deltas = {"exact": 0, "previous": -1, "next": 1}
|
|
if closest not in deltas.keys():
|
|
raise ValueError(f"Invalid closest argument: {closest}")
|
|
|
|
as_on_match = closest if as_on_match == "closest" else as_on_match
|
|
prior_match = closest if prior_match == "closest" else prior_match
|
|
|
|
if as_on_match in deltas.keys():
|
|
as_on_delta = datetime.timedelta(days=deltas[as_on_match])
|
|
else:
|
|
raise ValueError(f"Invalid as_on_match argument: {as_on_match}")
|
|
|
|
if prior_match in deltas.keys():
|
|
prior_delta = datetime.timedelta(days=deltas[prior_match])
|
|
else:
|
|
raise ValueError(f"Invalid prior_match argument: {prior_match}")
|
|
|
|
return as_on_delta, prior_delta
|
|
|
|
|
|
def _parse_date(date: str, date_format: str = None):
|
|
"""Parses date and handles errors"""
|
|
|
|
if isinstance(date, (datetime.datetime, datetime.date)):
|
|
return datetime.datetime.fromordinal(date.toordinal())
|
|
|
|
if date_format is None:
|
|
date_format = FincalOptions.date_format
|
|
|
|
try:
|
|
date = datetime.datetime.strptime(date, date_format)
|
|
except TypeError:
|
|
raise ValueError("Date does not seem to be valid date-like string")
|
|
except ValueError:
|
|
raise ValueError("Date could not be parsed. Have you set the correct date format in FincalOptions.date_format?")
|
|
return date
|
|
|
|
|
|
class _IndexSlicer:
|
|
"""Class to create a slice using iloc in TimeSeriesCore"""
|
|
|
|
def __init__(self, parent_obj):
|
|
self.parent = parent_obj
|
|
|
|
def __getitem__(self, n):
|
|
if isinstance(n, int):
|
|
keys = [self.parent.dates[n]]
|
|
else:
|
|
keys = self.parent.dates[n]
|
|
item = [(key, self.parent.data[key]) for key in keys]
|
|
if len(item) == 1:
|
|
return item[0]
|
|
|
|
return item
|
|
|
|
|
|
class Series(UserList):
|
|
"""Container for a series of objects, all objects must be of the same type"""
|
|
|
|
def __init__(
|
|
self,
|
|
data,
|
|
data_type: Literal['date', 'number', 'bool'],
|
|
date_format: str = None,
|
|
):
|
|
types_dict = {
|
|
'date': datetime.datetime,
|
|
'datetime': datetime.datetime,
|
|
'datetime.datetime': datetime.datetime,
|
|
'float': float,
|
|
'int': float,
|
|
'number': float,
|
|
'bool': bool
|
|
}
|
|
|
|
if data_type not in types_dict.keys():
|
|
raise ValueError("Unsupported value for data type")
|
|
|
|
if not isinstance(data, Sequence):
|
|
raise TypeError("Series object can only be created using Sequence types")
|
|
|
|
if data_type in ['date', 'datetime', 'datetime.datetime']:
|
|
data = [_parse_date(i, date_format) for i in data]
|
|
else:
|
|
func = types_dict[data_type]
|
|
data = [func(i) for i in data]
|
|
|
|
self.dtype = types_dict[data_type]
|
|
self.data = data
|
|
|
|
def __repr__(self):
|
|
return f"{self.__class__.__name__}({self.data}, data_type='{self.data_type.__name__}')"
|
|
|
|
def __getitem__(self, i):
|
|
if isinstance(i, slice):
|
|
return self.__class__(self.data[i], str(self.dtype.__name__))
|
|
else:
|
|
return self.data[i]
|
|
|
|
def __gt__(self, other):
|
|
if self.dtype == bool:
|
|
raise TypeError("> not supported for boolean series")
|
|
|
|
if isinstance(other, (str, datetime.datetime, datetime.date)):
|
|
other = _parse_date(other)
|
|
|
|
if self.dtype == float and isinstance(other, Number) or isinstance(other, self.dtype):
|
|
gt = Series([i > other for i in self.data], 'bool')
|
|
else:
|
|
raise Exception(f"Cannot compare type {self.dtype.__name__} to {type(other).__name__}")
|
|
|
|
return gt
|
|
|
|
def __ge__(self, other):
|
|
if self.dtype == bool:
|
|
raise TypeError(">= not supported for boolean series")
|
|
|
|
if isinstance(other, (str, datetime.datetime, datetime.date)):
|
|
other = _parse_date(other)
|
|
|
|
if self.dtype == float and isinstance(other, Number) or isinstance(other, self.dtype):
|
|
ge = Series([i >= other for i in self.data], 'bool')
|
|
else:
|
|
raise Exception(f"Cannot compare type {self.dtype.__name__} to {type(other).__name__}")
|
|
|
|
return ge
|
|
|
|
def __lt__(self, other):
|
|
if self.dtype == bool:
|
|
raise TypeError("< not supported for boolean series")
|
|
|
|
if isinstance(other, (str, datetime.datetime, datetime.date)):
|
|
other = _parse_date(other)
|
|
|
|
if self.dtype == float and isinstance(other, Number) or isinstance(other, self.dtype):
|
|
lt = Series([i < other for i in self.data], 'bool')
|
|
else:
|
|
raise Exception(f"Cannot compare type {self.dtype.__name__} to {type(other).__name__}")
|
|
return lt
|
|
|
|
def __le__(self, other):
|
|
if self.dtype == bool:
|
|
raise TypeError("<= not supported for boolean series")
|
|
|
|
if isinstance(other, (str, datetime.datetime, datetime.date)):
|
|
other = _parse_date(other)
|
|
|
|
if self.dtype == float and isinstance(other, Number) or isinstance(other, self.dtype):
|
|
le = Series([i <= other for i in self.data], 'bool')
|
|
else:
|
|
raise Exception(f"Cannot compare type {self.dtype.__name__} to {type(other).__name__}")
|
|
return le
|
|
|
|
def __eq__(self, other):
|
|
if isinstance(other, (str, datetime.datetime, datetime.date)):
|
|
other = _parse_date(other)
|
|
|
|
if self.dtype == float and isinstance(other, Number) or isinstance(other, self.dtype):
|
|
eq = Series([i == other for i in self.data], 'bool')
|
|
else:
|
|
raise Exception(f"Cannot compare type {self.dtype.__name__} to {type(other).__name__}")
|
|
return eq
|
|
|
|
|
|
class TimeSeriesCore(UserDict):
|
|
"""Defines the core building blocks of a TimeSeries object"""
|
|
|
|
def __init__(
|
|
self, data: List[Iterable], frequency: Literal["D", "W", "M", "Q", "H", "Y"], date_format: str = "%Y-%m-%d"
|
|
):
|
|
"""Instantiate a TimeSeries object
|
|
|
|
Parameters
|
|
----------
|
|
data : List[tuple]
|
|
Time Series data in the form of list of tuples.
|
|
The first element of each tuple should be a date and second element should be a value.
|
|
|
|
date_format : str, optional, default "%Y-%m-%d"
|
|
Specify the format of the date
|
|
Required only if the first argument of tuples is a string. Otherwise ignored.
|
|
|
|
frequency : str, optional, default "infer"
|
|
The frequency of the time series. Default is infer.
|
|
The class will try to infer the frequency automatically and adjust to the closest member.
|
|
Note that inferring frequencies can fail if the data is too irregular.
|
|
Valid values are {D, W, M, Q, H, Y}
|
|
"""
|
|
|
|
data = _preprocess_timeseries(data, date_format=date_format)
|
|
|
|
self.data = dict(data)
|
|
if len(self.data) != len(data):
|
|
print("Warning: The input data contains duplicate dates which have been ignored.")
|
|
self.frequency = getattr(AllFrequencies, frequency)
|
|
self.iter_num = -1
|
|
self._dates = None
|
|
self._values = None
|
|
self._start_date = None
|
|
self._end_date = None
|
|
|
|
@property
|
|
def dates(self):
|
|
if self._dates is None or len(self._dates) != len(self.data):
|
|
self._dates = list(self.data.keys())
|
|
|
|
return Series(self._dates, 'date')
|
|
|
|
@property
|
|
def values(self):
|
|
if self._values is None or len(self._values) != len(self.data):
|
|
self._values = list(self.data.values())
|
|
|
|
return Series(self._values, 'number')
|
|
|
|
@property
|
|
def start_date(self):
|
|
return self.dates[0]
|
|
|
|
@property
|
|
def end_date(self):
|
|
return self.dates[-1]
|
|
|
|
def _get_printable_slice(self, n: int):
|
|
"""Returns a slice of the dataframe from beginning and end"""
|
|
|
|
printable = {}
|
|
iter_f = iter(self.data)
|
|
first_n = [next(iter_f) for i in range(n // 2)]
|
|
|
|
iter_b = reversed(self.data)
|
|
last_n = [next(iter_b) for i in range(n // 2)]
|
|
last_n.sort()
|
|
|
|
printable["start"] = [str((i, self.data[i])) for i in first_n]
|
|
printable["end"] = [str((i, self.data[i])) for i in last_n]
|
|
return printable
|
|
|
|
def __repr__(self):
|
|
if len(self.data) > 6:
|
|
printable = self._get_printable_slice(6)
|
|
printable_str = "{}([{}\n\t ...\n\t {}], frequency={})".format(
|
|
self.__class__.__name__,
|
|
",\n\t ".join(printable["start"]),
|
|
",\n\t ".join(printable["end"]),
|
|
repr(self.frequency.symbol),
|
|
)
|
|
else:
|
|
printable_str = "{}([{}], frequency={})".format(
|
|
self.__class__.__name__,
|
|
",\n\t".join([str(i) for i in self.data.items()]),
|
|
repr(self.frequency.symbol),
|
|
)
|
|
return printable_str
|
|
|
|
def __str__(self):
|
|
if len(self.data) > 6:
|
|
printable = self._get_printable_slice(6)
|
|
printable_str = "[{}\n ...\n {}]".format(
|
|
",\n ".join(printable["start"]),
|
|
",\n ".join(printable["end"]),
|
|
)
|
|
else:
|
|
printable_str = "[{}]".format(",\n ".join([str(i) for i in self.data.items()]))
|
|
return printable_str
|
|
|
|
def __getitem__(self, key):
|
|
if isinstance(key, Series):
|
|
if not key.dtype == bool:
|
|
raise ValueError(f"Cannot slice {self.__class__.__name__} using a Series of {key.dtype.__name__}")
|
|
elif len(key) != len(self.dates):
|
|
raise Exception(f"Length of Series: {len(key)} did not match length of object: {len(self.dates)}")
|
|
else:
|
|
dates = self.dates
|
|
dates_to_return = [dates[i] for i, j in enumerate(key) if j]
|
|
data_to_return = [(key, self.data[key]) for key in dates_to_return]
|
|
return self.__class__(data_to_return, frequency=self.frequency.symbol)
|
|
|
|
if isinstance(key, int):
|
|
raise KeyError(f"{key}. For index based slicing, use .iloc[{key}]")
|
|
elif isinstance(key, (datetime.datetime, datetime.date)):
|
|
key = _parse_date(key)
|
|
item = (key, self.data[key])
|
|
elif isinstance(key, str):
|
|
if key == "dates":
|
|
return self.dates
|
|
elif key == "values":
|
|
return self.values
|
|
|
|
dt_key = _parse_date(key)
|
|
item = (dt_key, self.data[dt_key])
|
|
|
|
elif isinstance(key, Sequence):
|
|
keys = [_parse_date(i) for i in key]
|
|
item = [(k, self.data[k]) for k in keys]
|
|
else:
|
|
raise TypeError(f"Invalid type {repr(type(key).__name__)} for slicing.")
|
|
return item
|
|
|
|
def __iter__(self):
|
|
self.n = 0
|
|
return self
|
|
|
|
def __next__(self):
|
|
if self.n >= len(self.dates):
|
|
raise StopIteration
|
|
else:
|
|
key = self.dates[self.n]
|
|
self.n += 1
|
|
return key, self.data[key]
|
|
|
|
def head(self, n: int = 6):
|
|
"""Returns the first n items of the TimeSeries object"""
|
|
|
|
keys = list(self.data.keys())
|
|
keys = keys[:n]
|
|
result = [(key, self.data[key]) for key in keys]
|
|
return result
|
|
|
|
def tail(self, n: int = 6):
|
|
"""Returns the last n items of the TimeSeries object"""
|
|
|
|
keys = list(self.data.keys())
|
|
keys = keys[-n:]
|
|
result = [(key, self.data[key]) for key in keys]
|
|
return result
|
|
|
|
@property
|
|
def iloc(self):
|
|
"""Returns an item or a set of items based on index"""
|
|
|
|
return _IndexSlicer(self)
|