@Override
  protected @Nullable PyExpression getCondition(PyStatement statement) {
    if (statement instanceof PyIfStatement ifStatement) {
      PyExpression condition = ifStatement.getIfPart().getCondition();
      if (condition == null) {
        return null;
      }
      return ((PyBinaryExpression)condition).getLeftExpression();
    }
    return null;
  }
@Override
  protected @Nullable PyExpression getCondition(PyStatement statement)