src/NamingService/SALOME_NamingServicePy.py

   1 #! /usr/bin/env python
   2 #  -*- coding: iso-8859-1 -*-
   3 #  Copyright (C) 2007-2008  CEA/DEN, EDF R&D, OPEN CASCADE
   4 #
   5 #  Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
   6 #  CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
   7 #
   8 #  This library is free software; you can redistribute it and/or
   9 #  modify it under the terms of the GNU Lesser General Public
  10 #  License as published by the Free Software Foundation; either
  11 #  version 2.1 of the License.
  12 #
  13 #  This library is distributed in the hope that it will be useful,
  14 #  but WITHOUT ANY WARRANTY; without even the implied warranty of
  15 #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  16 #  Lesser General Public License for more details.
  17 #
  18 #  You should have received a copy of the GNU Lesser General Public
  19 #  License along with this library; if not, write to the Free Software
  20 #  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  21 #
  22 #  See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  23 #
  24 #  SALOME NamingService : wrapping NamingService services
  25 #  File   : SALOME_NamingServicePy.py
  26 #  Author : Estelle Deville, CEA
  27 #  Module : SALOME
  28 #  $Header$
  29 #
  30 ## @package SALOME_NamingServicePy
  31 # \brief Module to manage SALOME naming service from python
  32 #
  33 #
  34
  35 import sys
  36 import time
  37 from omniORB import CORBA
  38 import CosNaming
  39 import string
  40 from string import *
  41
  42 from SALOME_utilities import *
  43 #=============================================================================
  44
  45 class SALOME_NamingServicePy_i(object):
  46     """
  47       A class to manage SALOME naming service from python code
  48     """
  49     _orb = None
  50     _root_context=None
  51     _current_context=None
  52     _obj=None
  53
  54     #-------------------------------------------------------------------------
  55
  56     def __init__(self, orb=None):
  57         """
  58         Standard Constructor, with ORB reference.
  59
  60         Initializes the naming service root context
  61         """
  62         #MESSAGE ( "SALOME_NamingServicePy_i::__init__" )
  63         if orb is None:
  64           orb=CORBA.ORB_init([''], CORBA.ORB_ID)
  65         self._orb = orb
  66         # initialize root context and current context
  67         ok = 0
  68         steps = 240
  69         while steps > 0 and ok == 0:
  70           try:
  71             obj =self._orb.resolve_initial_references("NameService")
  72             self._root_context =obj._narrow(CosNaming.NamingContext)
  73             self._current_context = self._root_context
  74
  75
  76             if self._root_context is None :
  77               #MESSAGE ( "Name Service Reference is invalid" )
  78               #sys.exit(1)
  79               MESSAGE(" Name service not found")
  80             else:
  81               ok = 1
  82           except (CORBA.TRANSIENT,CORBA.OBJECT_NOT_EXIST,CORBA.COMM_FAILURE):
  83             MESSAGE(" Name service not found")
  84           time.sleep(0.25)
  85           steps = steps - 1
  86         if steps == 0 and self._root_context is None:
  87           MESSAGE ( "Name Service Reference is invalid" )
  88           sys.exit(1)
  89
  90     #-------------------------------------------------------------------------
  91
  92     def Register(self,ObjRef, Path):
  93         """ ns.Register(object,pathname )
  94
  95         register a CORBA object under a pathname
  96         """
  97
  98         MESSAGE ( "SALOME_NamingServicePy_i::Register" )
  99         _not_exist = 0
 100         path_list = list(Path)
 101         if path_list[0]=='/':
 102             self._current_context = self._root_context
 103             #delete first '/' before split
 104             Path=Path[1:]
 105
 106         result_resolve_path = string.split(Path,'/')
 107         if len(result_resolve_path)>1:
 108             # A directory is treated (not only an object name)
 109             # We had to test if the directory where ObjRef should be recorded
 110             # is already done
 111             # If not, the new context has to be created
 112             _context_name = []
 113             for i in range(len(result_resolve_path)-1):
 114                 _context_name.append(CosNaming.NameComponent(result_resolve_path[i],"dir"))
 115
 116             try:
 117                 obj = self._current_context.resolve(_context_name)
 118                 self._current_context = obj._narrow(CosNaming.NamingContext)
 119             except CosNaming.NamingContext.NotFound, ex:
 120                 _not_exist = 1
 121             except CosNaming.NamingContext.InvalidName, ex:
 122                 MESSAGE ( "Register : CosNaming.NamingContext.InvalidName" )
 123             except CosNaming.NamingContext.CannotProceed, ex:
 124                 MESSAGE ( "Register : CosNaming.NamingContext.CannotProceed" )
 125             except (CORBA.TRANSIENT,CORBA.OBJECT_NOT_EXIST,CORBA.COMM_FAILURE):
 126                 MESSAGE ( "Register : CORBA.TRANSIENT,CORBA.OBJECT_NOT_EXIST,CORBA.COMM_FAILURE" )
 127
 128             if _not_exist:
 129                 # at least one context of the complete path is not created, we had
 130                 # to create it or them
 131                 _context_name = []
 132                 for i in range(len(result_resolve_path)-1):
 133                     _context_name = [CosNaming.NameComponent(result_resolve_path[i],"dir")]
 134
 135                     try:
 136                         obj = self._current_context.resolve(_context_name)
 137                         self._current_context = obj._narrow(CosNaming.NamingContext)
 138                     except CosNaming.NamingContext.NotFound, ex:
 139                         #This context is not created. It will be done
 140                         self._current_context = self._current_context.bind_new_context(_context_name)
 141
 142         #The current directory is now the directory where the object should
 143         #be recorded
 144
 145         _context_name = [CosNaming.NameComponent(result_resolve_path[len(result_resolve_path)-1],"object")]
 146         try:
 147             self._current_context.bind(_context_name,ObjRef)
 148         except CosNaming.NamingContext.NotFound, ex:
 149             MESSAGE ( "Register : CosNaming.NamingContext.NotFound" )
 150         except CosNaming.NamingContext.InvalidName, ex:
 151             MESSAGE ( "Register : CosNaming.NamingContext.InvalidName" )
 152         except CosNaming.NamingContext.CannotProceed, ex:
 153             MESSAGE ( "Register : CosNaming.NamingContext.CannotProceed" )
 154         except CosNaming.NamingContext.AlreadyBound, ex:
 155             MESSAGE ( "Register : CosNaming.NamingContext.AlreadyBound, object will be rebind" )
 156             self._current_context.rebind(_context_name,ObjRef)
 157         except (CORBA.TRANSIENT,CORBA.OBJECT_NOT_EXIST,CORBA.COMM_FAILURE):
 158             MESSAGE ( "Register : CORBA.TRANSIENT,CORBA.OBJECT_NOT_EXIST,CORBA.COMM_FAILURE" )
 159
 160     #-------------------------------------------------------------------------
 161
 162     def Resolve(self, Path):
 163         """ ns.Resolve(pathname) -> object
 164
 165         find a CORBA object (ior) by its pathname
 166         """
 167         #MESSAGE ( "SALOME_NamingServicePy_i::Resolve" )
 168         path_list = list(Path)
 169         if path_list[0]=='/':
 170             self._current_context = self._root_context
 171             #delete first '/' before split
 172             Path=Path[1:]
 173
 174         result_resolve_path = string.split(Path,'/')
 175         _context_name=[]
 176         for i in range(len(result_resolve_path)-1):
 177             _context_name.append(CosNaming.NameComponent(result_resolve_path[i],"dir"))
 178         _context_name.append(CosNaming.NameComponent(result_resolve_path[len(result_resolve_path)-1],"object"))
 179         try:
 180             self._obj = self._current_context.resolve(_context_name)
 181         except CosNaming.NamingContext.NotFound, ex:
 182             MESSAGE ( "Resolve : CosNaming.NamingContext.NotFound" )
 183             self._obj = None
 184         except CosNaming.NamingContext.InvalidName, ex:
 185             MESSAGE ( "Resolve : CosNaming.NamingContext.InvalidName" )
 186             self._obj = None
 187         except CosNaming.NamingContext.CannotProceed, ex:
 188             MESSAGE ( "Resolve : CosNaming.NamingContext.CannotProceed" )
 189             self._obj = None
 190         except (CORBA.TRANSIENT,CORBA.OBJECT_NOT_EXIST,CORBA.COMM_FAILURE):
 191             MESSAGE ( "Resolve : CORBA.TRANSIENT,CORBA.OBJECT_NOT_EXIST,CORBA.COMM_FAILURE" )
 192             self._obj = None
 193         return self._obj
 194
 195
 196     #-------------------------------------------------------------------------
 197
 198     def Create_Directory(self,ObjRef, Path):
 199         """ ns.Create_Directory(ObjRef, Path)
 200
 201         create a sub directory
 202         """
 203         MESSAGE ( "SALOME_NamingServicePy_i::Create_Directory" )
 204         _not_exist = 0
 205         path_list = list(Path)
 206         if path_list[0]=='/':
 207             self._current_context = self._root_context
 208             #delete first '/' before split
 209             Path=Path[1:]
 210
 211         result_resolve_path = string.split(Path,'/')
 212         _context_name = []
 213         for i in range(len(result_resolve_path)):
 214             _context_name[CosNaming.NameComponent(result_resolve_path[i],"dir")]
 215             try:
 216                 obj = self._current_context.resolve(_context_name)
 217                 self._current_context = obj._narrow(CosNaming.NamingContext)
 218             except CosNaming.NamingContext.NotFound, ex:
 219                 self._current_context = self._current_context.bind_new_context(_context_name)
 220             except CosNaming.NamingContext.InvalidName, ex:
 221                 MESSAGE ( "Create_Directory : CosNaming.NamingContext.InvalidName" )
 222             except CosNaming.NamingContext.CannotProceed, ex:
 223                 MESSAGE ( "Create_Directory : CosNaming.NamingContext.CannotProceed" )
 224             except (CORBA.TRANSIENT,CORBA.OBJECT_NOT_EXIST,CORBA.COMM_FAILURE):
 225                 MESSAGE ( "Create_Directory : CORBA.TRANSIENT,CORBA.OBJECT_NOT_EXIST,CORBA.COMM_FAILURE" )
 226
 227     def Destroy_Name(self,Path):
 228       """ ns.Destroy_Name(Path)
 229
 230         remove a name in naming service
 231       """
 232       resolve_path=string.split(Path,'/')
 233       if resolve_path[0] == '': del resolve_path[0]
 234       dir_path=resolve_path[:-1]
 235       context_name=[]
 236       for e in dir_path:
 237          context_name.append(CosNaming.NameComponent(e,"dir"))
 238       context_name.append(CosNaming.NameComponent(resolve_path[-1],"object"))
 239
 240       try:
 241         self._root_context.unbind(context_name)
 242       except CosNaming.NamingContext.NotFound, ex:
 243         return
 244       except CORBA.Exception,ex:
 245         return
 246
 247     def Destroy_FullDirectory(self,Path):
 248       """ ns.Destroy_FullDirectory(Path)
 249
 250         remove recursively a directory
 251       """
 252       context_name=[]
 253       for e in string.split(Path,'/'):
 254         if e == '':continue
 255         context_name.append(CosNaming.NameComponent(e,"dir"))
 256
 257       try:
 258         context=self._root_context.resolve(context_name)
 259       except CosNaming.NamingContext.NotFound, ex:
 260         return
 261       except CORBA.Exception,ex:
 262         return
 263
 264       bl,bi=context.list(0)
 265       if bi is not None:
 266          ok,b=bi.next_one()
 267          while(ok):
 268             for s in b.binding_name :
 269                if s.kind == "object":
 270                   context.unbind([s])
 271                elif s.kind == "dir":
 272                   context.unbind([s])
 273             ok,b=bi.next_one()
 274
 275       context.destroy()
 276       self._root_context.unbind(context_name)