Files
quantconnect--lean/Common/Exceptions/KeyErrorPythonExceptionInterpreter.cs
T
Martin-Molinero b77f0122b2
Regression Tests / build (push) Has been cancelled
Build & Test Lean / build (push) Has been cancelled
Improve python exception parsing (#5831)
* Improve python exception parsing

- Improve python exception parsing adding support for line shift. Adding
  unit tests

* PythonException revert change

* Centralized and normalize algorithm runtime handling

* Adding support for C# line and file exception report
2021-08-06 20:11:54 -03:00

70 lines
2.9 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 System;
using Python.Runtime;
using QuantConnect.Util;
using System.Collections.Generic;
namespace QuantConnect.Exceptions
{
/// <summary>
/// Interprets <see cref="KeyErrorPythonExceptionInterpreter"/> instances
/// </summary>
public class KeyErrorPythonExceptionInterpreter : PythonExceptionInterpreter
{
/// <summary>
/// Determines the order that an instance of this class should be called
/// </summary>
public override int Order => 0;
/// <summary>
/// Determines if this interpreter should be applied to the specified exception.
/// </summary>
/// <param name="exception">The exception to check</param>
/// <returns>True if the exception can be interpreted, false otherwise</returns>
public override bool CanInterpret(Exception exception)
{
return base.CanInterpret(exception) &&
exception.Message.Contains("KeyError");
}
/// <summary>
/// Interprets the specified exception into a new exception
/// </summary>
/// <param name="exception">The exception to be interpreted</param>
/// <param name="innerInterpreter">An interpreter that should be applied to the inner exception.</param>
/// <returns>The interpreted exception</returns>
public override Exception Interpret(Exception exception, IExceptionInterpreter innerInterpreter)
{
var pe = (PythonException)exception;
var key = string.Empty;
if (pe.Message.Contains("["))
{
key = pe.Message.GetStringBetweenChars('[', ']');
}
else if (pe.Message.Contains("\'"))
{
key = pe.Message.GetStringBetweenChars('\'', '\'');
}
var message = $"Trying to retrieve an element from a collection using a key that does not exist in that collection throws a KeyError exception. To prevent the exception, ensure that the {key} key exist in the collection and/or that collection is not empty.";
message += PythonUtil.PythonExceptionStackParser(pe.StackTrace);
return new KeyNotFoundException(message, pe);
}
}
}