From 8aee16971313730d77fc9c2fb70eb7f4be11ee3c Mon Sep 17 00:00:00 2001
From: "Maxence G. de Montauzan" <maxence@gdemontauzan.fr>
Date: Fri, 21 Apr 2017 16:09:31 +0200
Subject: [PATCH] Add ELS option in send_data.py

---
 send_data.py | 17 +++++++++++++----
 1 file changed, 13 insertions(+), 4 deletions(-)

diff --git a/send_data.py b/send_data.py
index 31ef0c0..7ba8b6e 100644
--- a/send_data.py
+++ b/send_data.py
@@ -7,8 +7,6 @@
 import argparse
 import requests
 
-ELASTICSEARCH_URL = 'http://localhost:9200/'
-
 def send_data(file_name, quiet=False):
     """
         Send a file to ELS
@@ -44,6 +42,9 @@ def delete_index(index_name, quiet=False):
 
 #### main block ####
 
+ELASTICSEARCH_URL = 'http://localhost:9200/'
+INDEX_NAME = "itunessongs"
+
 # Get options
 parser = argparse.ArgumentParser(
     description='''
@@ -68,13 +69,21 @@ parser.add_argument('-A', '--ALL', action='store_true',
                     help='Send all')
 parser.add_argument('-D', '--DELETE', action='store_true',
                     help='Delete old index')
+parser.add_argument('-els', '--elasticsearch_url', default=ELASTICSEARCH_URL,
+                    help="Elasticsearch URL to send data (default: {})".format(ELASTICSEARCH_URL))
+parser.add_argument('-i', '--index_name', default=INDEX_NAME,
+                    help="Index name in Elasticsearch for data (default: {})".format(INDEX_NAME))
 parser.add_argument('-q', '--quiet', action='store_true',
                     help="Disable main output")
 
-INDEX_NAME = "itunessongs"
-
 if __name__ == '__main__':
     args = parser.parse_args()
+    INDEX_NAME = args.index_name
+    ELASTICSEARCH_URL = args.elasticsearch_url
+
+    if not args.quiet:
+        print("Elasticsearch URL:\t" + ELASTICSEARCH_URL)
+        print("Index name:\t\t" + INDEX_NAME)
 
     if args.DELETE:
         delete_index(INDEX_NAME, args.quiet)