1 # -*- coding: iso-8859-1 -*-
2 # Copyright (C) 2007-2016 CEA/DEN, EDF R&D, OPEN CASCADE
4 # Copyright (C) 2003-2007 OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
5 # CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
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.
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.
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
21 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
24 from optparse import OptionParser
30 # Use to display newlines (\n) in epilog
31 class MyParser(OptionParser):
32 def format_epilog(self, formatter):
36 def __parse_args(args):
40 usage = "Usage: salome connect [-p port] [ -c command | script | - ]"
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
47 The -c option can be used to specify the command to execute in the interpreter.
48 A script can also be used.
50 salome connect -p 2810 -c 'print "Hello"'
51 salome connect -p 2810 hello.py
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."
58 parser.add_option('-c', dest="command", default=None,
59 help="The command to execute in the interpreter."
62 (options, args) = parser.parse_args(args)
70 def __show_running_instances(list_of_instances):
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)
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): ")
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]
87 print "*** Invalid number! ***"
88 return __choose_in(choices)
91 def __get_running_session(requested_port=None, lastInstanceByDefault=False):
94 from ORBConfigFile import readORBConfigFile
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))
103 host, port, filename = None, None, None
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)]
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)
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)
122 print "Multiple instances found ; please choose one in the following:"
123 host, port, filename = __choose_in(available_connexions)
127 print "Selected instance: %s:%s"%(host, port)
131 return host, port, filename
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")
143 self.rootContext = obj._narrow(CosNaming.NamingContext)
145 except (CORBA.TRANSIENT,CORBA.OBJECT_NOT_EXIST,CORBA.COMM_FAILURE):
146 print "It's not a valid naming service"
147 self.rootContext = None
157 traceback.print_exc()
161 session_server = clt.Resolve('/Kernel/Session')
163 session = clt.waitNS("/Kernel/Session")
164 catalog = clt.waitNS("/Kernel/ModulCatalog")
165 studyMgr = clt.waitNS("/myStudyManager")
169 from salome import lcc
170 print "--> now connected to SALOME"
173 def _prompt(environment=None, commands=None, message="Connecting to SALOME"):
174 if environment is None:
175 environment = globals().copy()
176 environment.update(locals())
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
187 shell = code.InteractiveConsole(environment)
189 print "Execute command:", cmd
195 def connect(args=None, env=None):
198 options, args = __parse_args(args)
199 host, port, filename = __get_running_session(options.port)
204 "os.environ['OMNIORB_CONFIG'] = '%s'"%filename,
208 cmd.append(options.command)
209 if args: # unprocessed: may be scripts
211 cmd.append("execfile('%s')"%os.path.abspath(os.path.expanduser(arg)))
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()
221 if __name__ == "__main__":
222 if len(sys.argv) == 2:
223 cmd = pickle.loads(sys.argv[1])
225 _prompt(commands=cmd)
227 print "runConsole.py: incorrect usage!"