iperf-3.3.tar.gz

#!/usr/bin/env python

“““
Extract iperf data from json blob and format for gnuplot.
“““

import json
import os
import sys

from optparse import OptionParser

import pprint
# for debugging so output to stderr to keep verbose
# output out of any redirected stdout.
pp = pprint.PrettyPrinter（indent=4 stream=sys.stderr）


def generate_output（iperf options）:
    “““Do the actual formatting.“““
    for i in iperf.get（‘intervals‘）:
        for ii in i.get（‘streams‘）:
            if options.verbose:
                pp.pprint（ii）
            row = ‘{0} {1} {2} {3} {4}\n‘.format（
                round（float（ii.get（‘start‘）） 4）
                ii.get（‘bytes‘）
                # to Gbits/sec
                round（float（ii.get（‘bits_per_second‘）） / （1000*1000*1000） 3）
                ii.get（‘retransmits‘）
                round（float（ii.get（‘snd_cwnd‘）） / （1000*1000） 2）
            ）
            yield row


def summed_output（iperf options）:
    “““Format summed output.“““

    for i in iperf.get（‘intervals‘）:

        row_header = None

        byte = list（）
        bits_per_second = list（）
        retransmits = list（）
        snd_cwnd = list（）

        for ii in i.get（‘streams‘）:
            if options.verbose:
                pp.pprint（i）
            # grab the first start value
            if row_header is None:
                row_header = round（float（ii.get（‘start‘）） 2）
            # aggregate the rest of the values
            byte.append（ii.get（‘bytes‘））
            bits_per_second.append（float（ii.get（‘bits_per_second‘）） / （1000*1000*1000））
            retransmits.append（ii.get（‘retransmits‘））
            snd_cwnd.append（float（ii.get（‘snd_cwnd‘）） / （1000*1000））

        row = ‘{h} {b} {bps} {r} {s}\n‘.format（
            h=row_header
            b=sum（byte）
            bps=round（sum（bits_per_second） 3）
            r=sum（retransmits）
            s=round（sum（snd_cwnd） / len（snd_cwnd） 2）
        ）

        yield row


def main（）:
    “““Execute the read and formatting.“““
    usage = ‘%prog [ -f FILE | -o OUT | -v ]‘
    parser = OptionParser（usage=usage）
    parser.add_option（‘-f‘ ‘--file‘ metavar=‘FILE‘
                      type=‘string‘ dest=‘filename‘
                      help=‘Input filename.‘）
    parser.add_option（‘-o‘ ‘--output‘ metavar=‘OUT‘
                      type=‘string‘ dest=‘output‘
                      help=‘Optional file to append output to.‘）
    parser.add_option（‘-s‘ ‘--sum‘
                      dest=‘summed‘ action=‘store_true‘ default=False
                      help=‘Summed version of the output.‘）
    parser.add_option（‘-v‘ ‘--verbose‘
                      dest=‘verbose‘ action=‘store_true‘ default=False
                      help=‘Verbose debug output to stderr.‘）
    options _ = parser.parse_args（）

    if not options.filename:
        parser.error（‘Filename is required.‘）

    file_path = os.path.normpath（options.filename）

    if not os.path.exists（file_path）:
        parser.error（‘{f