maxenceg2m/iTunes
1
0
Fork 0
Code Issues 2 Pull Requests Projects Releases Wiki Activity

Add ELS option in send_data.py

Browse Source
This commit is contained in:
Maxence G. de Montauzan
2017-04-21 16:09:31 +02:00
parent f4c8efb8d1
commit 8aee169713
1 changed files with 13 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
send_data.py
View File

@@ -7,8 +7,6 @@
import argparse import argparse
import requests import requests
ELASTICSEARCH_URL = 'http://localhost:9200/'
def send_data(file_name, quiet=False): def send_data(file_name, quiet=False):
""" """
Send a file to ELS Send a file to ELS
@@ -44,6 +42,9 @@ def delete_index(index_name, quiet=False):
#### main block #### #### main block ####
ELASTICSEARCH_URL = 'http://localhost:9200/'
INDEX_NAME = "itunessongs"
# Get options # Get options
parser = argparse.ArgumentParser( parser = argparse.ArgumentParser(
description=''' description='''
@@ -68,13 +69,21 @@ parser.add_argument('-A', '--ALL', action='store_true',
help='Send all') help='Send all')
parser.add_argument('-D', '--DELETE', action='store_true', parser.add_argument('-D', '--DELETE', action='store_true',
help='Delete old index') 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', parser.add_argument('-q', '--quiet', action='store_true',
help="Disable main output") help="Disable main output")
INDEX_NAME = "itunessongs"
if __name__ == '__main__': if __name__ == '__main__':
args = parser.parse_args() 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: if args.DELETE:
delete_index(INDEX_NAME, args.quiet) delete_index(INDEX_NAME, args.quiet)