Changeset 240 for tbroadcast/v2/scripts/tbroadcast

Timestamp:

Aug 28, 2006, 1:20:02 PM (18 years ago)

Author:

garonne

Message:

improve tbroadcast script

File:

• tbroadcast/v2/scripts/tbroadcast (modified) (2 diffs)

tbroadcast/v2/scripts/tbroadcast

@@ -1 @@
-#!/usr/bin/env python2
+#!/usr/bin/env python
 #----------------------------------#
 # -- Author: V.Garonne
@@ -13 +13 @@
 from tbroadcast import Scheduler
 
+def usage():
+    print 'Usage : > tbroadcast [global options] [<command>]'
+    print '# command :'
+    print '#   <command>: command to execute'
+    print '# global options :'
+    print '#   -nb=<num_worker>]                  : Total number of threads'
+    print '#   -help]                             : Print help'
+
 if __name__ == '__main__':
-
-    if len(sys.argv) ==1 :
-        print 'Usage : > tbroadcast [-nb=<num_worker>] <command>'
-        sys.exit(-1)    
-    num_worker = 20    
-    for arg in sys.argv:            
+    # Default options
+    num_worker = 20
+    command    = ''    
+    for arg in sys.argv[1:len(sys.argv)]:            
          if arg[0]=='-':
-             if string.split(arg,'=')[0] == '-nb':
+             option = string.split(arg,'=')[0]
+             if option == '-nb':
                 num_worker = int (string.split(arg,'=')[1])         
+             else:
+                usage()
+                sys.exit(-1)   
+                
          else:
               command = arg
-                      
-    # we create a pool of num_worker worker threads
-    #print 'num_worker', num_worker , command
+
     master = Scheduler (num_workers=num_worker)        
     master.execute_all (command)