Add ability to print out the CSV headers

This commit is contained in:
Matt Martz 2016-08-29 09:42:14 -05:00
parent 884c7fed87
commit 01abb3ae71
1 changed files with 13 additions and 0 deletions

View File

@ -1100,6 +1100,14 @@ def version():
sys.exit(0)
def csv_header():
"""Print the CSV Headers"""
print_('Server ID,Sponsor,Server Name,Timestamp,Distance,Ping,Download,'
'Upload')
sys.exit(0)
def parse_args():
"""Function to handle building and parsing of command line arguments"""
description = (
@ -1132,6 +1140,8 @@ def parse_args():
parser.add_argument('--csv-delimiter', default=',', type=PARSER_TYPE_STR,
help='Single character delimiter to use in CSV '
'output. Default ","')
parser.add_argument('--csv-header', action='store_true', default=False,
help='Print CSV headers')
parser.add_argument('--json', action='store_true', default=False,
help='Suppress verbose output, only show basic '
'information in JSON format')
@ -1207,6 +1217,9 @@ def shell():
if args.version:
version()
if args.csv_header:
csv_header()
if len(args.csv_delimiter) != 1:
raise SystemExit('--csv-delimiter must be a single character')