Files
quantconnect--lean/Common/Python/BrokerageMessageHandlerPythonWrapper.cs
T
Jhonathan Abreu 3dab2e42eb Handle/filter new brokerage-side orders (#7706)
* Handle/filter brokerage side orders

* Add unit tests

* Minor fix

* Add unrequested security for new brokerage-side orders

* Add unit test for algorithms overriding the brokerage message handler

* Add unit test for python algorithm overriding the brokerage message handler

* Minor change

* Address peer review
2024-01-23 12:01:03 -04:00

63 lines
2.3 KiB
C#

/*
* 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.
*/
using Python.Runtime;
using QuantConnect.Brokerages;
namespace QuantConnect.Python
{
/// <summary>
/// Provides a wrapper for <see cref="IBrokerageMessageHandler"/> implementations written in python
/// </summary>
public class BrokerageMessageHandlerPythonWrapper : IBrokerageMessageHandler
{
private readonly dynamic _model;
/// <summary>
/// Initializes a new instance of the <see cref="BrokerageMessageHandlerPythonWrapper"/> class
/// </summary>
/// <param name="model">The python implementation of <see cref="IBrokerageMessageHandler"/></param>
public BrokerageMessageHandlerPythonWrapper(PyObject model)
{
_model = model.ValidateImplementationOf<IBrokerageMessageHandler>();
}
/// <summary>
/// Handles the message
/// </summary>
/// <param name="message">The message to be handled</param>
public void HandleMessage(BrokerageMessageEvent message)
{
using (Py.GIL())
{
_model.HandleMessage(message);
}
}
/// <summary>
/// Handles a new order placed manually in the brokerage side
/// </summary>
/// <param name="eventArgs">The new order event</param>
/// <returns>Whether the order should be added to the transaction handler</returns>
public bool HandleOrder(NewBrokerageOrderNotificationEventArgs eventArgs)
{
using (Py.GIL())
{
return _model.HandleOrder(eventArgs);
}
}
}
}