pep8 conversion of python algos (#7942)
* pep8 conversion of python algos * adding 10 more pep8 converted algos
This commit is contained in:
Ashutosh
@@ -20,15 +20,15 @@ from math import copysign
|
||||
### <meta name="tag" content="regression test" />
|
||||
class UpdateOrderRegressionAlgorithm(QCAlgorithm):
|
||||
|
||||
def Initialize(self):
|
||||
def initialize(self):
|
||||
'''Initialise the data and resolution required, as well as the cash and start-end dates for your algorithm. All algorithms must initialized.'''
|
||||
|
||||
self.SetStartDate(2013,1,1) #Set Start Date
|
||||
self.SetEndDate(2015,1,1) #Set End Date
|
||||
self.SetCash(100000) #Set Strategy Cash
|
||||
self.set_start_date(2013,1,1) #Set Start Date
|
||||
self.set_end_date(2015,1,1) #Set End Date
|
||||
self.set_cash(100000) #Set Strategy Cash
|
||||
# Find more symbols here: http://quantconnect.com/data
|
||||
|
||||
self.security = self.AddEquity("SPY", Resolution.Daily)
|
||||
self.security = self.add_equity("SPY", Resolution.DAILY)
|
||||
|
||||
self.last_month = -1
|
||||
self.quantity = 100
|
||||
@@ -39,86 +39,86 @@ class UpdateOrderRegressionAlgorithm(QCAlgorithm):
|
||||
self.limit_percentage = 0.025
|
||||
self.limit_percentage_delta = 0.005
|
||||
|
||||
OrderTypeEnum = [OrderType.Market, OrderType.Limit, OrderType.StopMarket, OrderType.StopLimit, OrderType.MarketOnOpen, OrderType.MarketOnClose, OrderType.TrailingStop]
|
||||
self.order_types_queue = CircularQueue[OrderType](OrderTypeEnum)
|
||||
self.order_types_queue.CircleCompleted += self.onCircleCompleted
|
||||
order_type_enum = [OrderType.MARKET, OrderType.LIMIT, OrderType.STOP_MARKET, OrderType.STOP_LIMIT, OrderType.MARKET_ON_OPEN, OrderType.MARKET_ON_CLOSE, OrderType.TRAILING_STOP]
|
||||
self.order_types_queue = CircularQueue[OrderType](order_type_enum)
|
||||
self.order_types_queue.circle_completed += self.on_circle_completed
|
||||
self.tickets = []
|
||||
|
||||
|
||||
def onCircleCompleted(self, sender, event):
|
||||
def on_circle_completed(self, sender, event):
|
||||
'''Flip our signs when we've gone through all the order types'''
|
||||
self.quantity *= -1
|
||||
|
||||
|
||||
def OnData(self, data):
|
||||
def on_data(self, data):
|
||||
'''OnData event is the primary entry point for your algorithm. Each new data point will be pumped in here.'''
|
||||
if not data.ContainsKey("SPY"):
|
||||
if not data.contains_key("SPY"):
|
||||
return
|
||||
|
||||
if self.Time.month != self.last_month:
|
||||
if self.time.month != self.last_month:
|
||||
# we'll submit the next type of order from the queue
|
||||
orderType = self.order_types_queue.Dequeue()
|
||||
order_type = self.order_types_queue.dequeue()
|
||||
#Log("")
|
||||
self.Log("\r\n--------------MONTH: {0}:: {1}\r\n".format(self.Time.strftime("%B"), orderType))
|
||||
self.Log("\r\n--------------MONTH: {0}:: {1}".format(self.time.strftime("%B"), order_type))
|
||||
#Log("")
|
||||
self.last_month = self.Time.month
|
||||
self.Log("ORDER TYPE:: {0}".format(orderType))
|
||||
isLong = self.quantity > 0
|
||||
stopPrice = (1 + self.stop_percentage)*data["SPY"].High if isLong else (1 - self.stop_percentage)*data["SPY"].Low
|
||||
limitPrice = (1 - self.limit_percentage)*stopPrice if isLong else (1 + self.limit_percentage)*stopPrice
|
||||
self.last_month = self.time.month
|
||||
self.log("ORDER TYPE:: {0}".format(order_type))
|
||||
is_long = self.quantity > 0
|
||||
stop_price = (1 + self.stop_percentage)*data["SPY"].high if is_long else (1 - self.stop_percentage)*data["SPY"].low
|
||||
limit_price = (1 - self.limit_percentage)*stop_price if is_long else (1 + self.limit_percentage)*stop_price
|
||||
|
||||
if orderType == OrderType.Limit:
|
||||
limitPrice = (1 + self.limit_percentage)*data["SPY"].High if not isLong else (1 - self.limit_percentage)*data["SPY"].Low
|
||||
if order_type == OrderType.LIMIT:
|
||||
limit_price = (1 + self.limit_percentage)*data["SPY"].high if not is_long else (1 - self.limit_percentage)*data["SPY"].low
|
||||
|
||||
request = SubmitOrderRequest(orderType, self.security.Symbol.SecurityType, "SPY", self.quantity, stopPrice, limitPrice, 0, 0.01, True,
|
||||
self.UtcTime, str(orderType))
|
||||
ticket = self.Transactions.AddOrder(request)
|
||||
request = SubmitOrderRequest(order_type, self.security.symbol.security_type, "SPY", self.quantity, stop_price, limit_price, 0, 0.01, True,
|
||||
self.utc_time, str(order_type))
|
||||
ticket = self.transactions.add_order(request)
|
||||
self.tickets.append(ticket)
|
||||
|
||||
elif len(self.tickets) > 0:
|
||||
ticket = self.tickets[-1]
|
||||
|
||||
if self.Time.day > 8 and self.Time.day < 14:
|
||||
if len(ticket.UpdateRequests) == 0 and ticket.Status is not OrderStatus.Filled:
|
||||
self.Log("TICKET:: {0}".format(ticket))
|
||||
updateOrderFields = UpdateOrderFields()
|
||||
updateOrderFields.Quantity = ticket.Quantity + copysign(self.delta_quantity, self.quantity)
|
||||
updateOrderFields.Tag = "Change quantity: {0}".format(self.Time.day)
|
||||
ticket.Update(updateOrderFields)
|
||||
if self.time.day > 8 and self.time.day < 14:
|
||||
if len(ticket.update_requests) == 0 and ticket.status is not OrderStatus.FILLED:
|
||||
self.log("TICKET:: {0}".format(ticket))
|
||||
update_order_fields = UpdateOrderFields()
|
||||
update_order_fields.quantity = ticket.quantity + copysign(self.delta_quantity, self.quantity)
|
||||
update_order_fields.tag = "Change quantity: {0}".format(self.time.day)
|
||||
ticket.update(update_order_fields)
|
||||
|
||||
elif self.Time.day > 13 and self.Time.day < 20:
|
||||
if len(ticket.UpdateRequests) == 1 and ticket.Status is not OrderStatus.Filled:
|
||||
self.Log("TICKET:: {0}".format(ticket))
|
||||
updateOrderFields = UpdateOrderFields()
|
||||
updateOrderFields.LimitPrice = self.security.Price*(1 - copysign(self.limit_percentage_delta, ticket.Quantity))
|
||||
updateOrderFields.StopPrice = self.security.Price*(1 + copysign(self.stop_percentage_delta, ticket.Quantity)) if ticket.OrderType != OrderType.TrailingStop else None
|
||||
updateOrderFields.Tag = "Change prices: {0}".format(self.Time.day)
|
||||
ticket.Update(updateOrderFields)
|
||||
elif self.time.day > 13 and self.time.day < 20:
|
||||
if len(ticket.update_requests) == 1 and ticket.status is not OrderStatus.FILLED:
|
||||
self.log("TICKET:: {0}".format(ticket))
|
||||
update_order_fields = UpdateOrderFields()
|
||||
update_order_fields.limit_price = self.security.price*(1 - copysign(self.limit_percentage_delta, ticket.quantity))
|
||||
update_order_fields.stop_price = self.security.price*(1 + copysign(self.stop_percentage_delta, ticket.quantity)) if ticket.order_type != OrderType.TRAILING_STOP else None
|
||||
update_order_fields.tag = "Change prices: {0}".format(self.time.day)
|
||||
ticket.update(update_order_fields)
|
||||
else:
|
||||
if len(ticket.UpdateRequests) == 2 and ticket.Status is not OrderStatus.Filled:
|
||||
self.Log("TICKET:: {0}".format(ticket))
|
||||
ticket.Cancel("{0} and is still open!".format(self.Time.day))
|
||||
self.Log("CANCELLED:: {0}".format(ticket.CancelRequest))
|
||||
if len(ticket.update_requests) == 2 and ticket.status is not OrderStatus.FILLED:
|
||||
self.log("TICKET:: {0}".format(ticket))
|
||||
ticket.cancel("{0} and is still open!".format(self.time.day))
|
||||
self.log("CANCELLED:: {0}".format(ticket.cancel_request))
|
||||
|
||||
|
||||
def OnOrderEvent(self, orderEvent):
|
||||
order = self.Transactions.GetOrderById(orderEvent.OrderId)
|
||||
ticket = self.Transactions.GetOrderTicket(orderEvent.OrderId)
|
||||
def on_order_event(self, orderEvent):
|
||||
order = self.transactions.get_order_by_id(orderEvent.order_id)
|
||||
ticket = self.transactions.get_order_ticket(orderEvent.order_id)
|
||||
|
||||
#order cancelations update CanceledTime
|
||||
if order.Status == OrderStatus.Canceled and order.CanceledTime != orderEvent.UtcTime:
|
||||
if order.status == OrderStatus.CANCELED and order.canceled_time != orderEvent.utc_time:
|
||||
raise ValueError("Expected canceled order CanceledTime to equal canceled order event time.")
|
||||
|
||||
#fills update LastFillTime
|
||||
if (order.Status == OrderStatus.Filled or order.Status == OrderStatus.PartiallyFilled) and order.LastFillTime != orderEvent.UtcTime:
|
||||
if (order.status == OrderStatus.FILLED or order.status == OrderStatus.PARTIALLY_FILLED) and order.last_fill_time != orderEvent.utc_time:
|
||||
raise ValueError("Expected filled order LastFillTime to equal fill order event time.")
|
||||
|
||||
# check the ticket to see if the update was successfully processed
|
||||
if len([ur for ur in ticket.UpdateRequests if ur.Response is not None and ur.Response.IsSuccess]) > 0 and order.CreatedTime != self.UtcTime and order.LastUpdateTime is None:
|
||||
if len([ur for ur in ticket.update_requests if ur.response is not None and ur.response.is_success]) > 0 and order.created_time != self.utc_time and order.last_update_time is None:
|
||||
raise ValueError("Expected updated order LastUpdateTime to equal submitted update order event time")
|
||||
|
||||
if orderEvent.Status == OrderStatus.Filled:
|
||||
self.Log("FILLED:: {0} FILL PRICE:: {1}".format(self.Transactions.GetOrderById(orderEvent.OrderId), orderEvent.FillPrice))
|
||||
if orderEvent.status == OrderStatus.FILLED:
|
||||
self.log("FILLED:: {0} FILL PRICE:: {1}".format(self.transactions.get_order_by_id(orderEvent.order_id), orderEvent.fill_price))
|
||||
else:
|
||||
self.Log(orderEvent.ToString())
|
||||
self.Log("TICKET:: {0}".format(ticket))
|
||||
self.log(orderEvent.to_string())
|
||||
self.log("TICKET:: {0}".format(ticket))
|
||||
|
||||
Reference in New Issue
Block a user