Add ELS option in send_data.py

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)