Skip to content

Commit

Permalink
Add ohlcv property setters
Browse files Browse the repository at this point in the history
  • Loading branch information
MDUYN committed Dec 24, 2023
1 parent 7270115 commit 43b31d1
Show file tree
Hide file tree
Showing 2 changed files with 26 additions and 2 deletions.
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
from abc import abstractmethod, ABC
from typing import Callable
from datetime import datetime


Expand Down Expand Up @@ -30,7 +31,14 @@ def prepare_data(
pass

@abstractmethod
def get_data(self, market_credential_service, backtest_index_date, **kwargs):
def get_data(
self,
market_credential_service,
backtest_index_date,
from_time_stamp=None,
to_time_stamp=None,
**kwargs
):
pass

@property
Expand Down Expand Up @@ -162,6 +170,22 @@ def get_end_date(self):
def end_date_func(self):
return self._end_date_func

@end_date.setter
def end_date(self, value):
self._end_date = value

@start_date.setter
def start_date(self, value):
self._start_date = value

@end_date_func.setter
def end_date_func(self, func: Callable):
self._end_date_func = func

@start_date_func.setter
def start_date_func(self, func: Callable):
self._start_date_func = func


class TickerMarketDataSource(MarketDataSource, ABC):

Expand Down
2 changes: 1 addition & 1 deletion version.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
VERSION = (2, 0, 2, 'alpha', 0)
VERSION = (2, 0, 3, 'alpha', 0)


def get_version(version=None):
Expand Down

0 comments on commit 43b31d1

Please sign in to comment.