Add "-p" option to print pretty json.

This commit is contained in:
Yaser Amiri 2018-04-19 21:02:07 +04:30
parent 5bbf5e7172
commit f6482274ee

View File

@ -1,7 +1,7 @@
#!/usr/bin/env python #!/usr/bin/env python
import socket import socket
import sys import sys
from pprint import pprint
from argparse import ArgumentParser, SUPPRESS from argparse import ArgumentParser, SUPPRESS
from datetime import datetime from datetime import datetime
from ssl import PROTOCOL_TLSv1 from ssl import PROTOCOL_TLSv1
@ -98,8 +98,10 @@ def show_result(user_args):
# Enable JSON output if -j argument specified # Enable JSON output if -j argument specified
if user_args.json_true: if user_args.json_true:
print(context) if user_args.pretty_output:
pprint(context)
else:
print(context)
def filter_hostname(host): def filter_hostname(host):
"""Remove unused characters and split by address and port.""" """Remove unused characters and split by address and port."""
@ -119,6 +121,9 @@ def get_args():
parser.add_argument("-j", "--json", dest="json_true", parser.add_argument("-j", "--json", dest="json_true",
action="store_true", default=False, action="store_true", default=False,
help="Enable JSON in the output") help="Enable JSON in the output")
parser.add_argument("-p", "--pretty", dest="pretty_output",
action="store_true",default=False,
help="Print pretty and more human readable Json")
parser.add_argument("-h", "--help", default=SUPPRESS, parser.add_argument("-h", "--help", default=SUPPRESS,
action='help', action='help',
help='Show this help message and exit') help='Show this help message and exit')