runpipe.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 bwmon.pipe import Pipe, PipeMonitor
   9
  10 sys.path.insert(0, os.path.dirname(__file__) or '.')
  11
  12 if __name__ == '__main__':
  13     if len(sys.argv) > 2:
  14         port = newport = int(sys.argv[1])
  15         newhost = sys.argv[2]
  16         if len(sys.argv) == 4:
  17             newport = int(sys.argv[3])
  18         pipe = Pipe(port, newhost, newport)
  19         monitor = PipeMonitor(pipe)
  20         pipe.start()
  21         try:
  22             monitor.run()
  23         except KeyboardInterrupt:
  24             pipe.close()
  25             print 'Waiting for threads to finish...'
  26
  27     else:
  28         import sys
  29         print >> sys.stderr, '''
  30 Usage:
  31     runpipe.py <port> <remotehost>
  32     runpipe.py <localport> <remotehost> <remoteport>
  33
  34 '''
  35