Add CSV Bulk Index Command

Default.sublime-commands

@@ -113,4 +113,7 @@
     {"command": "dumpdata",                "caption": "Elasticsearch: Dump Data"},
     {"command": "loaddata",                "caption": "Elasticsearch: Load Data"},
     {"command": "copy_index",              "caption": "Elasticsearch: Copy Data from ..."},
+
+    {"command": "csv_convert_bulk_format",  "caption": "Elasticsearch: CSV Convert to Bulk Index format"},
+    {"command": "csv_bulk_index",           "caption": "Elasticsearch: CSV Bulk Index"},
 ]

README.md

@@ -244,6 +244,8 @@ open the Command Palette (``Shift + Command + P``) and enter ``Elasticsearch ...
 * Elasticsearch: Dump Data
 * Elasticsearch: Load Data
 * Elasticsearch: Copy Data from ...
+* Elasticsearch: CSV Convert to Bulk Index format
+* Elasticsearch: CSV Bulk Index
 
 ### Command for Sublime User Settings
 

convert.py

@@ -0,0 +1,40 @@
+import sublime
+import sublime_plugin
+import csv
+import io
+import json
+
+
+class CsvConvertBulkFormatCommand(sublime_plugin.TextCommand):
+
+    def new_file(self, title=''):
+        view = sublime.active_window().new_file()
+        view.set_name('**Bulk {} Data**'.format(title))
+        view.set_scratch(True)
+        view.set_syntax_file('Packages/JavaScript/JSON.tmLanguage')
+        return view
+
+    def run(self, edit, action='index'):
+
+        text = self.view.substr(sublime.Region(0, self.view.size()))
+
+        output_view = self.new_file(action)
+
+        for doc in csv.DictReader(io.StringIO(text.strip())):
+            bulk_action = {action: {}}
+
+            if 'id' in doc:
+                bulk_action[action]['_id'] = doc['id']
+
+            data = []
+            data.append(json.dumps(bulk_action))
+
+            if action in ('index', 'create'):
+                data.append(json.dumps(doc, ensure_ascii=False))
+            elif action == 'update':
+                data.append(json.dumps({'doc': doc}, ensure_ascii=False))
+
+            output_text = '\n'.join(data) + '\n'
+
+            output_view.insert(
+                edit, output_view.size(), output_text)

helpers.py

@@ -134,3 +134,10 @@ def on_done(self, index):
 
     def run(self):
         self.show_select_inputfiles(self.on_done)
+
+
+class CsvBulkIndexCommand(HelperBaseCommand):
+
+    def run(self):
+        self.window.run_command('csv_convert_bulk_format')
+        self.window.run_command('bulk')

indices.py

@@ -207,7 +207,8 @@ class PutIndexSettingsCommand(IndicesClientCommand):
     result_window_title = "Put Index Settings"
 
     def run(self):
-        self.request_indices_api('put_settings', body=self.selection())
+        self.request_indices_api(
+            'put_settings', index=self.index, body=self.selection())
 
 
 class PutIndexWarmerCommand(IndicesClientCommand):