add optional host and path param on load.py (elastic#601)

If you want/need to load data into a different es cluster just specify either the * -H/--host option * -p/--path option ex: python load.py --host http://123.123.123.123:9200 -p /home/code/elasticsearch

Author: fxdgear

example/load.py

@@ -6,6 +6,7 @@
 from datetime import datetime
 import logging
 import sys
+import argparse
 
 import git
 
@@ -174,11 +175,25 @@ def load_repo(client, path=None, index='git'):
  tracer.setLevel(logging.INFO)
  tracer.addHandler(logging.FileHandler('/tmp/es_trace.log'))
 
+ parser = argparse.ArgumentParser()
+ parser.add_argument(
+ "-H", "--host",
+ action="store",
+ default="localhost:9200",
+ help="The elasticsearch host you wish to connect too. (Default: localhost:9200)")
+ parser.add_argument(
+ "-p", "--path",
+ action="store",
+ default=None,
+ help="Path to git repo. Commits used as data to load into Elasticsearch. (Default: None")
+
+ args = parser.parse_args()
+
  # instantiate es client, connects to localhost:9200 by default
- es = Elasticsearch()
+ es = Elasticsearch(args.host)
 
  # we load the repo and all commits
- load_repo(es, path=sys.argv[1] if len(sys.argv) == 2 else None)
+ load_repo(es, path=args.path)
 
  # run the bulk operations
  success, _ = bulk(es, REPO_ACTIONS, index='git', raise_on_error=True)