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     from killSalomeWithPort import getPiDict
  33     filedict=getPiDict(my_port)
  34     try:
  35         fpid=open(filedict, 'r')
  36         process_ids=pickle.load(fpid)
  37         fpid.close()
  38     except:
  39         process_ids=[{}]
  40         pass
  41
  42     already_in=0
  43     for process_id in process_ids:
  44         print process_id
  45         for pid, cmd in process_id.items():
  46             #print "see process %s : %s"% (pid, cmd[0])
  47             if pid == command_pid:
  48                 already_in=1
  49                 pass
  50             pass
  51         pass
  52
  53     command=(command.split(" "))[0]
  54     if already_in == 0:
  55         try:
  56             process_ids.append({command_pid: [command]})
  57             fpid=open(filedict,'w')
  58             pickle.dump(process_ids, fpid)
  59             fpid.close()
  60         except:
  61             print "addToKillList: can not add command %s to the kill list"% filedict
  62             pass
  63         pass
  64     pass
  65
  66 def killList():
  67     my_port = findFileDict()
  68     from killSalomeWithPort import getPiDict
  69     filedict=getPiDict(my_port)
  70     try:
  71         fpid=open(filedict, 'r')
  72         process_ids=pickle.load(fpid)
  73         fpid.close()
  74     except:
  75         process_ids=[{}]
  76         pass
  77
  78     for process_id in process_ids:
  79         print process_id
  80         for pid, cmd in process_id.items():
  81             print "stop process %s : %s"% (pid, cmd[0])
  82             try:
  83                 os.kill(int(pid),signal.SIGKILL)
  84             except:
  85                 print "  ------------------ process %s : %s inexistant"% (pid, cmd[0])
  86                 pass
  87             pass
  88         pass
  89     os.remove(filedict)
  90     pass
  91
  92
  93
  94 if __name__ == "__main__":
  95     print sys.argv
  96     addToKillList(sys.argv[1], sys.argv[2])