bin/addToKillList.py

   1 #! /usr/bin/env python
   2 #  -*- coding: iso-8859-1 -*-
   3 # Copyright (C) 2007-2016  CEA/DEN, EDF R&D, OPEN CASCADE
   4 #
   5 # Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
   6 # CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
   7 #
   8 # This library is free software; you can redistribute it and/or
   9 # modify it under the terms of the GNU Lesser General Public
  10 # License as published by the Free Software Foundation; either
  11 # version 2.1 of the License, or (at your option) any later version.
  12 #
  13 # This library is distributed in the hope that it will be useful,
  14 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  15 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  16 # Lesser General Public License for more details.
  17 #
  18 # You should have received a copy of the GNU Lesser General Public
  19 # License along with this library; if not, write to the Free Software
  20 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  21 #
  22 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  23 #
  24
  25 import os, sys, pickle, signal
  26 from launchConfigureParser import verbose
  27
  28 ########## adds to the kill list of SALOME one more process ##########
  29
  30 def findFileDict():
  31     """
  32     Detect current SALOME session's port number.
  33     Returns port number.
  34     """
  35     from salome_utils import getPortNumber
  36     port = getPortNumber()
  37     if verbose(): print "myport = ", port
  38     return port
  39
  40 def addToKillList(command_pid, command, port=None):
  41     """
  42     Add the process to the SALOME processes dictionary file.
  43     Parameters:
  44     - command_pid : command PID
  45     - command     : command (string or list of strings)
  46     - [port]      : SALOME port number; if this parameter is None (default),
  47     it is detected automatically
  48     """
  49     # retrieve current processes dictionary
  50     from killSalomeWithPort import getPiDict
  51     if port is None: port=findFileDict()
  52     filedict = getPiDict(port)
  53
  54     try:
  55         with open(filedict, 'r') as fpid:
  56             process_ids=pickle.load(fpid)
  57     except:
  58         process_ids=[]
  59         pass
  60     # check if PID is already in dictionary
  61     already_in=False
  62     for process_id in process_ids:
  63         for pid in process_id.keys():
  64             if int(pid) == int(command_pid):
  65                 already_in=True
  66                 break
  67             pass
  68         if already_in: break
  69         pass
  70
  71     # add process to the dictionary
  72     if not already_in:
  73         import types
  74         if type(command) == types.ListType: command=" ".join([str(c) for c in command])
  75         command=command.split()[0]
  76         try:
  77             if verbose(): print "addToKillList: %s : %s" % ( str(command_pid), command )
  78             process_ids.append({int(command_pid): [command]})
  79             dir = os.path.dirname(filedict)
  80             if not os.path.exists(dir): os.makedirs(dir, 0777)
  81             with open(filedict,'w') as fpid:
  82                 pickle.dump(process_ids, fpid)
  83         except:
  84             if verbose(): print "addToKillList: can not add command %s : %s to the kill list" % ( str(command_pid), command )
  85             pass
  86         pass
  87     pass
  88
  89 def killList(port=None):
  90     """
  91     Kill all the processes listed in the SALOME processes dictionary file.
  92     - [port]      : SALOME port number; if this parameter is None (default),
  93     it is detected automatically
  94     """
  95     # retrieve processes dictionary
  96     from killSalomeWithPort import getPiDict
  97     if port is None: port=findFileDict()
  98
  99     # new-style dot-prefixed pidict file
 100     filedict=getPiDict(port)
 101     # provide compatibility with old-style pidict file (not dot-prefixed)
 102     if not os.path.exists(filedict): filedict = getPiDict(port, hidden=False)
 103
 104     try:
 105         with open(filedict, 'r') as fpid:
 106             process_ids=pickle.load(fpid)
 107     except:
 108         process_ids=[]
 109         pass
 110     # kill processes
 111     for process_id in process_ids:
 112         #print process_id
 113         for pid, cmd in process_id.items():
 114             try:
 115                 os.kill(int(pid),signal.SIGKILL)
 116             except:
 117                 print "  ------------------ process %s : %s inexistant"% (pid, cmd[0])
 118                 pass
 119             pass
 120         pass
 121     # remove processes dictionary file
 122     os.remove(filedict)
 123     pass
 124
 125 if __name__ == "__main__":
 126     if verbose(): print sys.argv
 127     addToKillList(sys.argv[1], sys.argv[2])
 128     pass