runmonitor.py

   1 #!/usr/bin/python
   2 # -*- coding: utf-8 -*-
   3
   4 from __future__ import absolute_import
   5
   6 import os
   7 import sys
   8 from optparse import OptionParser
   9
  10 sys.path.insert(0, os.path.dirname(__file__) or '.')
  11
  12 if __name__ == '__main__':
  13     from bwmon import monitor
  14
  15     parser = OptionParser()
  16     parser.add_option('--include', dest='include_filter', type='string', action='append', help='include only processes that match the given regex')
  17     parser.add_option('--exclude', dest='exclude_filter', type='string', action='append', help='exclude processes that match the given regex')
  18     parser.add_option('--bandwidth', dest='bandwidth', action='store_true', default=False, help='print bandwidth instead of traffic')
  19
  20     (options, args) = parser.parse_args()
  21
  22     m = monitor.Monitor()
  23     m.set_filter(options.include_filter, options.exclude_filter)
  24     if options.bandwidth:
  25         m.loop(monitor.BANDWIDTH)
  26     else:
  27         m.loop(monitor.TRAFFIC)
  28