analyse script for "nethogs -t > filename"

list_nethogs.py

#! /usr/bin/env python

# analyse script for "nethogs -t > filename"
import sys


def process_line(line, sent, received):
    """
    @params: line:     string: line to process
    @params: sent:     int:    threshold for sent
    @params: received: int:    threshold for received
    @returns: matched line or None
    """
    parts = line.split("\t")
    try:
        if len(parts) < 3:
            return None
        s = float(parts[-2])
        r = float(parts[-1].strip("\n"))
        if s > sent or r > received:
            return line
        return None
    except Exception as e:
        print("unrecognized line: {}, error: {}".format(line, e))
        return None


if __name__ == "__main__":
    if len(sys.argv) < 2:
        exit()

    sent, received = 500, 500
    if len(sys.argv) > 2:
        sent = float(sys.argv[2])
    if len(sys.argv) > 3:
        received = float(sys.argv[3])

    path = sys.argv[1]
    with open(path, "r") as f:
        for line in f:
            if process_line(line, sent, received):
                print(line)