Return to iperf3_to_gnuplot.py CVS log

Up to [ELWIX - Embedded LightWeight unIX -] / embedaddon / iperf / contrib

Annotation of embedaddon/iperf/contrib/iperf3_to_gnuplot.py, revision 1.1

1.1     ! misho       1: #!/usr/bin/env python
        !             2: 
        !             3: """
        !             4: Extract iperf data from json blob and format for gnuplot.
        !             5: """
        !             6: 
        !             7: import json
        !             8: import os
        !             9: import sys
        !            10: 
        !            11: import os.path
        !            12: from optparse import OptionParser
        !            13: 
        !            14: import pprint
        !            15: # for debugging, so output to stderr to keep verbose
        !            16: # output out of any redirected stdout.
        !            17: pp = pprint.PrettyPrinter(indent=4, stream=sys.stderr)
        !            18: 
        !            19: def generate_output(iperf, options):
        !            20:     for i in iperf.get('intervals'):
        !            21:         for ii in i.get('streams'):
        !            22:             if options.verbose: pp.pprint(ii)
        !            23:             row = '{0} {1} {2} {3} {4}\n'.format(
        !            24:                         round(float(ii.get('start')), 4),
        !            25:                         ii.get('bytes'),
        !            26:                         # to Gbits/sec
        !            27:                         round(float(ii.get('bits_per_second')) / (1000*1000*1000), 3),
        !            28:                         ii.get('retransmits'),
        !            29:                         round(float(ii.get('snd_cwnd')) / (1000*1000), 2)
        !            30:                     )
        !            31:             yield row
        !            32: 
        !            33: def main():
        !            34:     usage = '%prog [ -f FILE | -o OUT | -v ]'
        !            35:     parser = OptionParser(usage=usage)
        !            36:     parser.add_option('-f', '--file', metavar='FILE',
        !            37:         type='string', dest='filename', 
        !            38:         help='Input filename.')
        !            39:     parser.add_option('-o', '--output', metavar='OUT',
        !            40:             type='string', dest='output', 
        !            41:             help='Optional file to append output to.')
        !            42:     parser.add_option('-v', '--verbose',
        !            43:         dest='verbose', action='store_true', default=False,
        !            44:         help='Verbose debug output to stderr.')
        !            45:     options, args = parser.parse_args()
        !            46:     
        !            47:     if not options.filename:
        !            48:         parser.error('Filename is required.')
        !            49:     
        !            50:     file_path = os.path.normpath(options.filename)
        !            51:     
        !            52:     if not os.path.exists(file_path):
        !            53:         parser.error('{f} does not exist'.format(f=file_path))
        !            54: 
        !            55:     with open(file_path,'r') as fh:
        !            56:         data = fh.read()
        !            57:     
        !            58:     try:
        !            59:         iperf = json.loads(data)
        !            60:     except Exception as e:
        !            61:         parser.error('Could not parse JSON from file (ex): {0}'.format(str(e)))
        !            62: 
        !            63:     if options.output:
        !            64:         absp = os.path.abspath(options.output)
        !            65:         d,f = os.path.split(absp)
        !            66:         if not os.path.exists(d):
        !            67:             parser.error('Output file directory path {0} does not exist'.format(d))
        !            68:         fh = open(absp, 'a')
        !            69:     else:
        !            70:         fh = sys.stdout
        !            71: 
        !            72:     for i in generate_output(iperf, options):
        !            73:         fh.write(i)
        !            74: 
        !            75: 
        !            76: if __name__ == '__main__':
        !            77:     main()