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