changed pos = [] to *pos in date_parser decorator
This commit is contained in:
parent
347634f536
commit
60f25de710
@ -17,14 +17,14 @@ class Frequency:
|
||||
symbol: str
|
||||
|
||||
|
||||
def date_parser(pos):
|
||||
def date_parser(*pos):
|
||||
def parse_dates(func):
|
||||
def wrapper_func(*args, **kwargs):
|
||||
date_format = kwargs.get("date_format", None)
|
||||
args = list(args)
|
||||
sig = inspect.signature(func)
|
||||
params = [i[0] for i in sig.parameters.items()]
|
||||
# print(params)
|
||||
|
||||
for j in pos:
|
||||
kwarg = params[j]
|
||||
date = kwargs.get(kwarg, None)
|
||||
|
@ -184,7 +184,7 @@ class TimeSeries(TimeSeriesCore):
|
||||
|
||||
return self.__class__(new_ts, frequency=self.frequency.symbol)
|
||||
|
||||
@date_parser(pos=[1])
|
||||
@date_parser(1)
|
||||
def calculate_returns(
|
||||
self,
|
||||
as_on: Union[str, datetime.datetime],
|
||||
@ -277,7 +277,7 @@ class TimeSeries(TimeSeriesCore):
|
||||
returns = returns ** (1 / years)
|
||||
return (current[0] if return_actual_date else as_on), returns - 1
|
||||
|
||||
@date_parser(pos=[1, 2])
|
||||
@date_parser(1, 2)
|
||||
def calculate_rolling_returns(
|
||||
self,
|
||||
from_date: Union[datetime.date, str],
|
||||
|
Loading…
Reference in New Issue
Block a user