Salome HOME
Fix bug to ensure compatibility with Python < 2.6
[modules/kernel.git] / bin / orbmodule.py
1 #  -*- coding: iso-8859-1 -*-
2 # Copyright (C) 2007-2013  CEA/DEN, EDF R&D, OPEN CASCADE
3 #
4 # Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
5 # CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
6 #
7 # This library is free software; you can redistribute it and/or
8 # modify it under the terms of the GNU Lesser General Public
9 # License as published by the Free Software Foundation; either
10 # version 2.1 of the License.
11 #
12 # This library is distributed in the hope that it will be useful,
13 # but WITHOUT ANY WARRANTY; without even the implied warranty of
14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15 # Lesser General Public License for more details.
16 #
17 # You should have received a copy of the GNU Lesser General Public
18 # License along with this library; if not, write to the Free Software
19 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
20 #
21 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
22 #
23
24 ## @package orbmodule
25 # \brief Module that provides a client for %SALOME
26 #
27
28 import sys,os,time
29 import string
30 from nameserver import *
31 from omniORB import CORBA
32 from launchConfigureParser import verbose
33
34 # Import the stubs for the Naming service
35 import CosNaming
36
37 # -----------------------------------------------------------------------------
38
39 class client:
40    """Client for SALOME"""
41
42    def __init__(self,args=None):
43       # Initialise the ORB
44       self.orb=CORBA.ORB_init(sys.argv, CORBA.ORB_ID)
45
46       # Initialise the Naming Service
47       self.initNS(args or {})
48
49    # --------------------------------------------------------------------------
50
51    def initNS(self,args):
52       # Obtain a reference to the root naming context
53       obj         = self.orb.resolve_initial_references("NameService")
54       try:
55           self.rootContext = obj._narrow(CosNaming.NamingContext)
56           return
57       except (CORBA.TRANSIENT,CORBA.OBJECT_NOT_EXIST,CORBA.COMM_FAILURE):
58           self.rootContext = None
59           if verbose(): print "Launch Naming Service++",
60
61       # On lance le Naming Server (doit etre dans le PATH)
62       test = True
63       if args['wake_up_session']:
64          test = False
65          pass
66       if test:
67          NamingServer(args).run()
68          pass
69       print "Searching Naming Service ",
70       ncount=0
71       delta=0.1
72       while(ncount < 100):
73           ncount += 1
74           try:
75               obj = self.orb.resolve_initial_references("NameService")
76               self.rootContext = obj._narrow(CosNaming.NamingContext)
77               break
78           except (CORBA.TRANSIENT,CORBA.OBJECT_NOT_EXIST,CORBA.COMM_FAILURE):
79               self.rootContext = None
80               sys.stdout.write('+')
81               sys.stdout.flush()
82               time.sleep(delta)
83
84       if self.rootContext is None:
85           print "Failed to narrow the root naming context"
86           sys.exit(1)
87       print " found in %s seconds " % ((ncount-1)*delta)
88
89    # --------------------------------------------------------------------------
90
91    def showNScontext(self,context,dec=''):
92       bl,bi=context.list(0)
93       if bi is not None:
94          ok,b=bi.next_one()
95          while(ok):
96             for s in b.binding_name :
97                print "%s%s.%s" %(dec,s.id,s.kind)
98                if s.kind == "dir":
99                   obj=context.resolve([s])
100                   scontext = obj._narrow(CosNaming.NamingContext)
101                   self.showNScontext(scontext,dec=dec+'  ')
102             ok,b=bi.next_one()
103
104    # --------------------------------------------------------------------------
105
106    def showNS(self):
107       """ Show the content of SALOME naming service """
108       self.showNScontext(self.rootContext)
109
110    # --------------------------------------------------------------------------
111
112    def Resolve(self, Path):
113       resolve_path=string.split(Path,'/')
114       if resolve_path[0] == '': del resolve_path[0]
115       dir_path=resolve_path[:-1]
116       context_name=[]
117       for e in dir_path:
118          context_name.append(CosNaming.NameComponent(e,"dir"))
119       context_name.append(CosNaming.NameComponent(resolve_path[-1],"object"))
120
121       try:
122           obj = self.rootContext.resolve(context_name)
123       except CosNaming.NamingContext.NotFound, ex:
124           obj = None
125       except CosNaming.NamingContext.InvalidName, ex:
126           obj = None
127       except CosNaming.NamingContext.CannotProceed, ex:
128           obj = None
129       except (CORBA.TRANSIENT,CORBA.OBJECT_NOT_EXIST,CORBA.COMM_FAILURE):
130           obj = None
131       return obj
132
133    # --------------------------------------------------------------------------
134
135    def waitNS(self,name,typobj=None,maxcount=240):
136       count=0
137       delta=0.5
138       print "Searching %s in Naming Service " % name,
139       while(1):
140           count += 1
141           if count > maxcount : raise RuntimeError, "Impossible de trouver %s" % name
142           obj=self.Resolve(name)
143           if obj :
144               print " found in %s seconds " % ((count-1)*delta)
145               break
146           else:
147               sys.stdout.write('+')
148               sys.stdout.flush()
149               time.sleep(delta)
150
151       if typobj is None:return obj
152
153       nobj = obj._narrow(typobj)
154       if nobj is None:
155             print "%s exists but is not a %s" % (name,typobj)
156       return nobj
157
158    if sys.platform != "win32":
159     def waitNSPID(self, theName, thePID, theTypObj = None):
160       aCount = 0
161       aDelta = 0.5
162       anObj = None
163       print "Searching %s in Naming Service " % theName,
164       while(1):
165          try:
166            os.kill(thePID,0)
167          except:
168            raise RuntimeError, "Process %d for %s not found" % (thePID,theName)
169          aCount += 1
170          anObj = self.Resolve(theName)
171          if anObj:
172             print " found in %s seconds " % ((aCount-1)*aDelta)
173             break
174          else:
175             sys.stdout.write('+')
176             sys.stdout.flush()
177             time.sleep(aDelta)
178             pass
179          pass
180
181       if theTypObj is None:
182          return anObj
183
184       anObject = anObj._narrow(theTypObj)
185       if anObject is None:
186          print "%s exists but is not a %s" % (theName,theTypObj)
187       return anObject
188
189
190    # --------------------------------------------------------------------------
191
192    def ResolveLogger(self, name):
193       context_name=[]
194       context_name.append(CosNaming.NameComponent(name,""))
195
196       try:
197           obj = self.rootContext.resolve(context_name)
198       except CosNaming.NamingContext.NotFound, ex:
199           obj = None
200       except CosNaming.NamingContext.InvalidName, ex:
201           obj = None
202       except CosNaming.NamingContext.CannotProceed, ex:
203           obj = None
204       except (CORBA.TRANSIENT,CORBA.OBJECT_NOT_EXIST,CORBA.COMM_FAILURE):
205           obj = None
206       return obj
207
208    # --------------------------------------------------------------------------
209
210    def waitLogger(self,name,typobj=None,maxcount=40):
211       count=0
212       delta=0.5
213       print "Searching %s in Naming Service " % name,
214       while(1):
215           count += 1
216           if count > maxcount : raise RuntimeError, "Impossible de trouver %s" % name
217           obj=self.ResolveLogger(name)
218           if obj :
219               print " found in %s seconds " % ((count-1)*delta)
220               break
221           else:
222               sys.stdout.write('+')
223               sys.stdout.flush()
224               time.sleep(delta)
225
226       if typobj is None:return obj
227
228       nobj = obj._narrow(typobj)
229       if nobj is None:
230             print "%s exists but is not a %s" % (name,typobj)
231       return nobj