Files
quantconnect--lean/Algorithm.Python/StopLimitOrderRegressionAlgorithm.py
T

78 lines
3.6 KiB
Python

# QUANTCONNECT.COM - Democratizing Finance, Empowering Individuals.
# Lean Algorithmic Trading Engine v2.0. Copyright 2014 QuantConnect Corporation.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
from AlgorithmImports import *
### <summary>
### Basic algorithm demonstrating how to place stop limit orders.
### </summary>
### <meta name="tag" content="trading and orders" />
### <meta name="tag" content="placing orders" />
### <meta name="tag" content="stop limit order"/>
class StopLimitOrderRegressionAlgorithm(QCAlgorithm):
'''Basic algorithm demonstrating how to place stop limit orders.'''
Tolerance = 0.001
FastPeriod = 30
SlowPeriod = 60
def Initialize(self):
self.SetStartDate(2013, 1, 1)
self.SetEndDate(2017, 1, 1)
self.SetCash(100000)
self._symbol = self.AddEquity("SPY", Resolution.Daily).Symbol
self._fast = self.EMA(self._symbol, self.FastPeriod, Resolution.Daily)
self._slow = self.EMA(self._symbol, self.SlowPeriod, Resolution.Daily)
self._buyOrderTicket: OrderTicket = None
self._sellOrderTicket: OrderTicket = None
self._previousSlice: Slice = None
def OnData(self, slice: Slice):
if not self.IsReady():
return
security = self.Securities[self._symbol]
if self._buyOrderTicket is None and self.TrendIsUp():
self._buyOrderTicket = self.StopLimitOrder(self._symbol, 100, stopPrice=security.High * 1.10, limitPrice=security.High * 1.11)
elif self._buyOrderTicket.Status == OrderStatus.Filled and self._sellOrderTicket is None and self.TrendIsDown():
self._sellOrderTicket = self.StopLimitOrder(self._symbol, -100, stopPrice=security.Low * 0.99, limitPrice=security.Low * 0.98)
def OnOrderEvent(self, orderEvent: OrderEvent):
if orderEvent.Status == OrderStatus.Filled:
order: StopLimitOrder = self.Transactions.GetOrderById(orderEvent.OrderId)
if not order.StopTriggered:
raise Exception("StopLimitOrder StopTriggered should haven been set if the order filled.")
if orderEvent.Direction == OrderDirection.Buy:
limitPrice = self._buyOrderTicket.Get(OrderField.LimitPrice)
if orderEvent.FillPrice > limitPrice:
raise Exception(f"Buy stop limit order should have filled with price less than or equal to the limit price {limitPrice}. "
f"Fill price: {orderEvent.FillPrice}")
else:
limitPrice = self._sellOrderTicket.Get(OrderField.LimitPrice)
if orderEvent.FillPrice < limitPrice:
raise Exception(f"Sell stop limit order should have filled with price greater than or equal to the limit price {limitPrice}. "
f"Fill price: {orderEvent.FillPrice}")
def IsReady(self):
return self._fast.IsReady and self._slow.IsReady
def TrendIsUp(self):
return self.IsReady() and self._fast.Current.Value > self._slow.Current.Value * (1 + self.Tolerance)
def TrendIsDown(self):
return self.IsReady() and self._fast.Current.Value < self._slow.Current.Value * (1 + self.Tolerance)