bin/addToKillList.py

   1 #!/usr/bin/env python
   2
   3 import os, sys, pickle, string, signal
   4
   5 ########## adds to the kill list of SALOME one more process ##########
   6
   7 def findFileDict():
   8     if os.environ.has_key('NSPORT'):
   9         my_port = os.environ['NSPORT']
  10         pass
  11     else:
  12         my_port = 2809
  13         try:
  14             file = open(os.environ["OMNIORB_CONFIG"], "r")
  15             s = file.read()
  16             while len(s):
  17                 l = string.split(s, ":")
  18                 if string.split(l[0], " ")[0] == "ORBInitRef" or string.split(l[0], " ")[0] == "InitRef" :
  19                     my_port = int(l[len(l)-1])
  20                     pass
  21                 s = file.read()
  22                 pass
  23             pass
  24         except:
  25             pass
  26         pass
  27     print "myport = ", my_port
  28     return my_port
  29
  30 def addToKillList(command_pid, command):
  31     my_port = findFileDict()
  32     filedict=os.getenv("HOME")+'/'+os.getenv('USER')+"_"+str(my_port)+'_SALOME_pidict'
  33     try:
  34         fpid=open(filedict, 'r')
  35         process_ids=pickle.load(fpid)
  36         fpid.close()
  37     except:
  38         process_ids=[{}]
  39         pass
  40
  41     already_in=0
  42     for process_id in process_ids:
  43         print process_id
  44         for pid, cmd in process_id.items():
  45             #print "see process %s : %s"% (pid, cmd[0])
  46             if pid == command_pid:
  47                 already_in=1
  48                 pass
  49             pass
  50         pass
  51
  52     command=(command.split(" "))[0]
  53     if already_in == 0:
  54         try:
  55             process_ids.append({command_pid: [command]})
  56             fpid=open(filedict,'w')
  57             pickle.dump(process_ids, fpid)
  58             fpid.close()
  59         except:
  60             print "addToKillList: can not add command %s to the kill list"% filedict
  61             pass
  62         pass
  63     pass
  64
  65 def killList():
  66     my_port = findFileDict()
  67     filedict=os.getenv("HOME")+'/'+os.getenv('USER')+"_"+str(my_port)+'_SALOME_pidict'
  68     try:
  69         fpid=open(filedict, 'r')
  70         process_ids=pickle.load(fpid)
  71         fpid.close()
  72     except:
  73         process_ids=[{}]
  74         pass
  75
  76     for process_id in process_ids:
  77         print process_id
  78         for pid, cmd in process_id.items():
  79             print "stop process %s : %s"% (pid, cmd[0])
  80             try:
  81                 os.kill(int(pid),signal.SIGKILL)
  82             except:
  83                 print "  ------------------ process %s : %s inexistant"% (pid, cmd[0])
  84                 pass
  85             pass
  86         pass
  87     os.remove(filedict)
  88     pass
  89
  90
  91
  92 if __name__ == "__main__":
  93     print sys.argv
  94     addToKillList(sys.argv[1], sys.argv[2])