]> SALOME platform Git repositories - modules/kernel.git/blob - bin/runConsole.py
Salome HOME
5b1e504197104110553d6f4f3a284bb67049b44c
[modules/kernel.git] / bin / runConsole.py
1 #  -*- coding: iso-8859-1 -*-
2 # Copyright (C) 2007-2017  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, or (at your option) any later version.
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 from optparse import OptionParser
25 import os
26 import sys
27 import user
28 import pickle
29
30 # Use to display newlines (\n) in epilog
31 class MyParser(OptionParser):
32   def format_epilog(self, formatter):
33     return self.epilog
34 #
35
36 def __parse_args(args):
37   if args is None:
38     args = []
39
40   usage = "Usage: salome connect [-p port] [ -c command | script | - ]"
41   epilog  = """
42 Connects a Python console to a local SALOME instance.\n
43 If port is given, try to connect to corresponding instance.
44 If port is not given, or does not correspond to a running instance,
45 ask user to select a port from list of available SALOME instances.\n
46
47 The -c option can be used to specify the command to execute in the interpreter.
48 A script can also be used.
49 For example:
50        salome connect -p 2810 -c 'print "Hello"'
51        salome connect -p 2810 hello.py
52 """
53   parser = MyParser(usage=usage, epilog=epilog)
54   parser.add_option("-p", metavar="<port>", default=0,
55                     action="store", type="string", dest="port",
56                     help="The port to connect to."
57                     )
58   parser.add_option('-c', dest="command", default=None,
59                     help="The command to execute in the interpreter."
60                     )
61   try:
62     (options, args) = parser.parse_args(args)
63   except Exception, e:
64     print e
65     return
66
67   return options, args
68 #
69
70 def __show_running_instances(list_of_instances):
71   print '-'*10
72   print "Running instances:"
73   for i in range(len(list_of_instances)):
74     host, port, _ = list_of_instances[i]
75     print "   [%d] %s:%s"%(i+1, host, port)
76   print '-'*10
77 #
78
79 def __choose_in(choices):
80   __show_running_instances(choices)
81   rep = raw_input("Please enter the number of instance to use (0 to cancel): ")
82   if rep == '0':
83     return None, None, None
84   elif rep in [str(i) for i in range(1, len(choices)+1)]:
85     return choices[int(rep)-1]
86   else:
87     print "*** Invalid number! ***"
88     return __choose_in(choices)
89 #
90
91 def __get_running_session(requested_port=None, lastInstanceByDefault=False):
92   import glob
93   import salome_utils
94   from ORBConfigFile import readORBConfigFile
95
96   omniorbUserPath = os.getenv("OMNIORB_USER_PATH")
97   files = glob.glob(os.path.join(omniorbUserPath,".omniORB_"+salome_utils.getUserName()+"_*[!last].cfg"))
98   available_connexions = []
99   for filename in files:
100     host, port = readORBConfigFile(filename)
101     available_connexions.append((host, port, filename))
102
103   host, port, filename = None, None, None
104   if requested_port:
105     print "Search for running instance on port %s..."%requested_port
106     found = [(h,p,f) for h,p,f in available_connexions if int(p) == int(requested_port)]
107     if not found:
108       print "   ...no running instance found"
109     elif len(found) == 1:
110       host, port, filename = found[0]
111       print "   ...found unique instance: %s:%s"%(host,port)
112     else:
113       print "   ...multiple instances found ; please choose one in the following:"
114       host, port, filename = __choose_in(found)
115   else: # no requested port
116     if not available_connexions:
117       print "No running instance found"
118     elif len(available_connexions) == 1:
119       host, port, filename = available_connexions[0]
120       print "Found unique instance: %s:%s"%(host,port)
121     else:
122       print "Multiple instances found ; please choose one in the following:"
123       host, port, filename = __choose_in(available_connexions)
124       pass
125
126   if port:
127     print "Selected instance: %s:%s"%(host, port)
128   else:
129     print "Cancel."
130
131   return host, port, filename
132 #
133
134 import CORBA
135 import CosNaming
136 import orbmodule
137
138 class client(orbmodule.client):
139   def initNS(self,args):
140     # Obtain a reference to the root naming context
141     obj = self.orb.resolve_initial_references("NameService")
142     try:
143       self.rootContext = obj._narrow(CosNaming.NamingContext)
144       return
145     except (CORBA.TRANSIENT,CORBA.OBJECT_NOT_EXIST,CORBA.COMM_FAILURE):
146       print "It's not a valid naming service"
147       self.rootContext = None
148       sys.stdout.flush()
149       raise
150 #
151
152 def start_client():
153   try:
154     clt = client()
155   except Exception:
156     import traceback
157     traceback.print_exc()
158     sys.exit(1)
159   #
160
161   session_server = clt.Resolve('/Kernel/Session')
162   if session_server:
163     session = clt.waitNS("/Kernel/Session")
164   catalog = clt.waitNS("/Kernel/ModulCatalog")
165   study = clt.waitNS("/Study")
166
167   import salome
168   salome.salome_init()
169   from salome import lcc
170   print "--> now connected to SALOME"
171 #
172
173 def _prompt(environment=None, commands=None, message="Connecting to SALOME"):
174   if environment is None:
175     environment = globals().copy()
176     environment.update(locals())
177   if commands is None:
178     commands = []
179
180   import code
181   import rlcompleter
182   import readline
183   readline.set_completer(rlcompleter.Completer(environment).complete)
184   readline.parse_and_bind("tab: complete")
185   # calling this with globals ensures we can see the environment
186   print message
187   shell = code.InteractiveConsole(environment)
188   for cmd in commands:
189     print "Execute command:", cmd
190     shell.push(cmd)
191     pass
192   shell.interact()
193 #
194
195 def connect(args=None, env=None):
196   if env is not None:
197     os.environ = env
198   options, args = __parse_args(args)
199   host, port, filename = __get_running_session(options.port)
200   if not port:
201     return 0
202
203   cmd = [
204     "os.environ['OMNIORB_CONFIG'] = '%s'"%filename,
205     "start_client()"
206     ]
207   if options.command:
208     cmd.append(options.command)
209   if args: # unprocessed: may be scripts
210     for arg in args:
211       cmd.append("execfile('%s')"%os.path.abspath(os.path.expanduser(arg)))
212
213   if port:
214     import subprocess
215     absoluteAppliPath = os.getenv('ABSOLUTE_APPLI_PATH','')
216     env_copy = os.environ.copy()
217     proc = subprocess.Popen(['python', os.path.join(absoluteAppliPath,"bin","salome","runConsole.py"), pickle.dumps(cmd)], shell=False, close_fds=True, env=env_copy)
218     return proc.communicate()
219 #
220
221 if __name__ == "__main__":
222   if len(sys.argv) == 2:
223     cmd = pickle.loads(sys.argv[1])
224     sys.argv = []
225     _prompt(commands=cmd)
226   else:
227     print "runConsole.py: incorrect usage!"
228 #