-#!/usr/bin/env python
+#! /usr/bin/env python
+# -*- coding: iso-8859-1 -*-
+# Copyright (C) 2007-2016 CEA/DEN, EDF R&D, OPEN CASCADE
+#
+# Copyright (C) 2003-2007 OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
+# CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
+#
+# This library is free software; you can redistribute it and/or
+# modify it under the terms of the GNU Lesser General Public
+# License as published by the Free Software Foundation; either
+# version 2.1 of the License, or (at your option) any later version.
+#
+# This library is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+# Lesser General Public License for more details.
+#
+# You should have received a copy of the GNU Lesser General Public
+# License along with this library; if not, write to the Free Software
+# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
+#
+# See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
+#
-import os, sys, pickle, string
+import os, sys, pickle, signal
+from launchConfigureParser import verbose
########## adds to the kill list of SALOME one more process ##########
-def addToKillList(command_pid, command):
- command=(command.split(" "))[0]
- my_port = 2809
- try:
- file = open(os.environ["OMNIORB_CONFIG"], "r")
- s = file.read()
- while len(s):
- l = string.split(s, ":")
- if string.split(l[0], " ")[0] == "ORBInitRef" or string.split(l[0], " ")[0] == "InitRef" :
- my_port = int(l[len(l)-1])
- pass
- s = file.read()
- pass
- except:
- pass
- #print "myport = ", my_port
+def findFileDict():
+ """
+ Detect current SALOME session's port number.
+ Returns port number.
+ """
+ from salome_utils import getPortNumber
+ port = getPortNumber()
+ if verbose(): print "myport = ", port
+ return port
+
+def addToKillList(command_pid, command, port=None):
+ """
+ Add the process to the SALOME processes dictionary file.
+ Parameters:
+ - command_pid : command PID
+ - command : command (string or list of strings)
+ - [port] : SALOME port number; if this parameter is None (default),
+ it is detected automatically
+ """
+ # retrieve current processes dictionary
+ from killSalomeWithPort import getPiDict
+ if port is None: port=findFileDict()
+ filedict = getPiDict(port)
- filedict=os.getenv("HOME")+'/'+os.getenv('USER')+"_"+str(my_port)+'_SALOME_pidict'
try:
- fpid=open(filedict, 'r')
- process_ids=pickle.load(fpid)
- fpid.close()
+ with open(filedict, 'r') as fpid:
+ process_ids=pickle.load(fpid)
except:
- process_ids=[{}]
+ process_ids=[]
pass
-
- already_in=0
+ # check if PID is already in dictionary
+ already_in=False
for process_id in process_ids:
- print process_id
- for pid, cmd in process_id.items():
- #print "see process %s : %s"% (pid, cmd[0])
- if pid == command_pid:
- already_in=1
+ for pid in process_id.keys():
+ if int(pid) == int(command_pid):
+ already_in=True
+ break
+ pass
+ if already_in: break
pass
- if already_in == 0:
+ # add process to the dictionary
+ if not already_in:
+ import types
+ if type(command) == types.ListType: command=" ".join([str(c) for c in command])
+ command=command.split()[0]
try:
- process_ids.append({command_pid: [command]})
- fpid=open(filedict,'w')
- pickle.dump(process_ids, fpid)
- fpid.close()
+ if verbose(): print "addToKillList: %s : %s" % ( str(command_pid), command )
+ process_ids.append({int(command_pid): [command]})
+ dir = os.path.dirname(filedict)
+ if not os.path.exists(dir): os.makedirs(dir, 0777)
+ with open(filedict,'w') as fpid:
+ pickle.dump(process_ids, fpid)
except:
- print "addToKillList: can not add command %s to the kill list"% filedict
- pass
- pass
+ if verbose(): print "addToKillList: can not add command %s : %s to the kill list" % ( str(command_pid), command )
+ pass
+ pass
+ pass
+
+def killList(port=None):
+ """
+ Kill all the processes listed in the SALOME processes dictionary file.
+ - [port] : SALOME port number; if this parameter is None (default),
+ it is detected automatically
+ """
+ # retrieve processes dictionary
+ from killSalomeWithPort import getPiDict
+ if port is None: port=findFileDict()
+
+ # new-style dot-prefixed pidict file
+ filedict=getPiDict(port)
+ # provide compatibility with old-style pidict file (not dot-prefixed)
+ if not os.path.exists(filedict): filedict = getPiDict(port, hidden=False)
+
+ try:
+ with open(filedict, 'r') as fpid:
+ process_ids=pickle.load(fpid)
+ except:
+ process_ids=[]
+ pass
+ # kill processes
+ for process_id in process_ids:
+ #print process_id
+ for pid, cmd in process_id.items():
+ try:
+ os.kill(int(pid),signal.SIGKILL)
+ except:
+ print " ------------------ process %s : %s inexistant"% (pid, cmd[0])
+ pass
+ pass
+ pass
+ # remove processes dictionary file
+ os.remove(filedict)
+ pass
if __name__ == "__main__":
- print sys.argv
+ if verbose(): print sys.argv
addToKillList(sys.argv[1], sys.argv[2])
+ pass
