From ca59f137b691e4b1427bdddda384e5b98f8044da Mon Sep 17 00:00:00 2001
From: Honza Kral <honza.kral@gmail.com>
Date: Mon, 22 Jul 2013 00:31:13 +0200
Subject: [PATCH] extract _bulk_body

---
 elasticsearch/client/__init__.py | 31 +++++++++++++------------------
 1 file changed, 13 insertions(+), 18 deletions(-)

diff --git a/elasticsearch/client/__init__.py b/elasticsearch/client/__init__.py
index a8d5e2d0..2c9843af 100644
--- a/elasticsearch/client/__init__.py
+++ b/elasticsearch/client/__init__.py
@@ -56,6 +56,17 @@ class Elasticsearch(object):
         self.indices = IndicesClient(self)
         self.cluster = ClusterClient(self)
 
+    def _bulk_body(self, body):
+        # if not passed in a string, serialize items and join by newline
+        if not isinstance(body, (type(''), type(u''))):
+            body = '\n'.join(map(self.transport.serializer.dumps, body))
+
+        # bulk body must end with a newline
+        if not body.endswith('\n'):
+            body += '\n'
+
+        return body
+
     @query_params()
     def ping(self, params=None):
         """ Returns True if the cluster is up, False otherwise. """
@@ -361,15 +372,7 @@ class Elasticsearch(object):
         :arg refresh: Refresh the index after performing the operation
         :arg replication: Explicitely set the replication type, default u'sync'
         """
-        # if not passed in a string, serialize items and join by newline
-        if not isinstance(body, (type(''), type(u''))):
-            body = '\n'.join(map(self.transport.serializer.dumps, body))
-
-        # bulk body must end with a newline
-        if not body.endswith('\n'):
-            body += '\n'
-
-        status, data = self.transport.perform_request('POST', _make_path(index, doc_type, '_bulk'), params=params, body=body)
+        status, data = self.transport.perform_request('POST', _make_path(index, doc_type, '_bulk'), params=params, body=self._bulk_body(body))
         return data
 
     @query_params('search_type')
@@ -383,14 +386,6 @@ class Elasticsearch(object):
         :arg doc_type: A comma-separated list of document types to use as default
         :arg search_type: Search operation type
         """
-        # if not passed in a string, serialize items and join by newline
-        if not isinstance(body, (type(''), type(u''))):
-            body = '\n'.join(map(self.transport.serializer.dumps, body))
-
-        # bulk body must end with a newline
-        if not body.endswith('\n'):
-            body += '\n'
-
-        status, data = self.transport.perform_request('GET', _make_path(index, doc_type, '_msearch'), params=params, body=body)
+        status, data = self.transport.perform_request('GET', _make_path(index, doc_type, '_msearch'), params=params, body=self._bulk_body(body))
         return data