Salome HOME
Integration of [CEA 13233] support new xdr.h location
[modules/kernel.git] / bin / killSalome.py
index f03beaa04d28ebd179296ea6bf7b4d00fb38aef2..cafe632f7664b8e982f0e3ef61ce08f7ef7b5ef9 100755 (executable)
@@ -1,6 +1,6 @@
-#! /usr/bin/env python
+#! /usr/bin/env python3
 #  -*- coding: iso-8859-1 -*-
-# Copyright (C) 2007-2013  CEA/DEN, EDF R&D, OPEN CASCADE
+# Copyright (C) 2007-2019  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
@@ -8,7 +8,7 @@
 # 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.
+# 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
@@ -30,12 +30,13 @@ import os, sys, re, signal
 
 from killSalomeWithPort import killMyPort, getPiDict
 #from salome_utils import getHostName, getShortHostName
+from salome_utils import getUserName
 
 def killAllPorts():
     """
     Kill all SALOME sessions belonging to the user.
     """
-    user = os.getenv('USER')
+    user = getUserName()
     #hostname  = getHostName()
     #shostname = getShortHostName()
     # new-style dot-prefixed pidict file
@@ -47,7 +48,7 @@ def killAllPorts():
     #if hostname in fpidict:
     #    fpidict = fpidict.replace(hostname, shostname+".*")
     fpidict   = fpidict.replace('#####', '(\d*)')
-    fnamere   = re.compile("^%s$" % fpidict)
+    fnamere   = re.compile("^%s" % fpidict)
     try:
         for f in os.listdir(dirpidict):
             mo = fnamere.match(f)
@@ -81,9 +82,9 @@ def killAllPorts():
         pass
     # kill other processes
     if sys.platform != 'win32':
-        import commands
+        import subprocess
         cmd = "ps -fea | grep '%s' | grep 'ghs3d' | grep 'f /tmp/GHS3D_' | grep -v 'grep' | awk '{print $2}'" % user
-        prc = commands.getoutput(cmd)
+        prc = subprocess.getoutput(cmd)
         for field in prc.split():
             try:
                 os.kill(int(field), signal.SIGKILL)
@@ -91,8 +92,27 @@ def killAllPorts():
                 pass
             pass
         pass
+        # kill ompi-server needed for MPI containers coupling
+        cmd = "ps -fea | grep '%s' | grep 'ompi-server' | grep -v 'grep' | awk '{print $2}'" % user
+        prc = subprocess.getoutput(cmd)
+        for field in prc.split():
+            try:
+                os.kill(int(field), signal.SIGKILL)
+            except:
+                pass
+            pass
+        pass
+        # delete uri files needed by ompi-server
+        cmd = "rm -f " + os.environ['HOME'] + "/.urifile_*"
+        os.system(cmd)
     pass
 
 if __name__ == "__main__":
+    try:
+        from salomeContextUtils import setOmniOrbUserPath
+        setOmniOrbUserPath()
+    except Exception as e:
+        print(e)
+        sys.exit(1)
     killAllPorts()
     pass