From ebf7a36ba1938e2002c1582cf87d4f6d3309b1a1 Mon Sep 17 00:00:00 2001 From: Denis Tereshkin Date: Mon, 24 Dec 2018 21:19:45 +0700 Subject: [PATCH] Son of a git --- src/naiback/strategy/strategy.py | 16 ++++++++++++---- 1 file changed, 12 insertions(+), 4 deletions(-) diff --git a/src/naiback/strategy/strategy.py b/src/naiback/strategy/strategy.py index 2812d77..5a47523 100644 --- a/src/naiback/strategy/strategy.py +++ b/src/naiback/strategy/strategy.py @@ -108,14 +108,18 @@ class Strategy: return dates - def buy_at_open(self, bar, ticker): + def buy_at_open(self, bar, ticker=None): + if ticker is None: + ticker = 0 if isinstance(ticker, int): ticker = self.all_bars[ticker].ticker bars = self._get_bars(ticker) self.broker.set_timestamp(bars.timestamp[bar]) return self.broker.add_position(ticker, bars.open[bar], self.trade_size, bar) - def buy_at_limit(self, bar, price, ticker): + def buy_at_limit(self, bar, price, ticker=None): + if ticker is None: + ticker = 0 if isinstance(ticker, int): ticker = self.all_bars[ticker].ticker bars = self._get_bars(ticker) @@ -128,7 +132,9 @@ class Strategy: else: return None - def buy_at_stop(self, bar, price, ticker): + def buy_at_stop(self, bar, price, ticker=None): + if ticker is None: + ticker = 0 if isinstance(ticker, int): ticker = self.all_bars[ticker].ticker bars = self._get_bars(ticker) @@ -141,7 +147,9 @@ class Strategy: else: return None - def buy_at_close(self, bar, ticker): + def buy_at_close(self, bar, ticker=None): + if ticker is None: + ticker = 0 if isinstance(ticker, int): ticker = self.all_bars[ticker].ticker bars = self._get_bars(ticker)