src/Container/SALOME_PyNode.py

   1 #! /usr/bin/env python
   2 #  -*- coding: iso-8859-1 -*-
   3 # Copyright (C) 2007-2015  CEA/DEN, EDF R&D, OPEN CASCADE
   4 #
   5 # This library is free software; you can redistribute it and/or
   6 # modify it under the terms of the GNU Lesser General Public
   7 # License as published by the Free Software Foundation; either
   8 # version 2.1 of the License, or (at your option) any later version.
   9 #
  10 # This library is distributed in the hope that it will be useful,
  11 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13 # Lesser General Public License for more details.
  14 #
  15 # You should have received a copy of the GNU Lesser General Public
  16 # License along with this library; if not, write to the Free Software
  17 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  18 #
  19 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  20 #
  21
  22 #  File   : SALOME_PyNode.py
  23 #  Author : Christian CAREMOLI, EDF
  24 #  Module : SALOME
  25 #  $Header$
  26 #
  27 import sys,traceback,string
  28 import linecache
  29 import cPickle
  30 import Engines__POA
  31 import SALOME__POA
  32 import SALOME
  33
  34 class Generic(SALOME__POA.GenericObj):
  35   """A Python implementation of the GenericObj CORBA IDL"""
  36   def __init__(self,poa):
  37     self.poa=poa
  38     self.cnt=1
  39
  40   def Register(self):
  41     self.cnt+=1
  42
  43   def UnRegister(self):
  44     self.cnt-=1
  45     if self.cnt <= 0:
  46       oid=self.poa.servant_to_id(self)
  47       self.poa.deactivate_object(oid)
  48
  49   def Destroy(self):
  50     print "WARNING SALOME::GenericObj::Destroy() function is obsolete! Use UnRegister() instead."
  51     self.UnRegister()
  52
  53 class PyNode_i (Engines__POA.PyNode,Generic):
  54   """The implementation of the PyNode CORBA IDL"""
  55   def __init__(self, nodeName,code,poa,my_container):
  56     """Initialize the node : compilation in the local context"""
  57     Generic.__init__(self,poa)
  58     self.nodeName=nodeName
  59     self.code=code
  60     self.my_container=my_container._container
  61     linecache.cache[nodeName]=0,None,string.split(code,'\n'),nodeName
  62     ccode=compile(code,nodeName,'exec')
  63     self.context={}
  64     self.context["my_container"] = self.my_container
  65     exec ccode in self.context
  66
  67   def executeAnotherPieceOfCode(self,code):
  68     """Called for initialization of container lodging self."""
  69     try:
  70       ccode=compile(code,self.nodeName,'exec')
  71       exec ccode in self.context
  72     except:
  73       raise SALOME.SALOME_Exception(SALOME.ExceptionStruct(SALOME.BAD_PARAM,"","PyScriptNode (%s) : code to be executed \"%s\"" %(self.nodeName,code),0))
  74
  75   def execute(self,funcName,argsin):
  76     """Execute the function funcName found in local context with pickled args (argsin)"""
  77     try:
  78       argsin,kws=cPickle.loads(argsin)
  79       func=self.context[funcName]
  80       argsout=func(*argsin,**kws)
  81       argsout=cPickle.dumps(argsout,-1)
  82       return argsout
  83     except:
  84       exc_typ,exc_val,exc_fr=sys.exc_info()
  85       l=traceback.format_exception(exc_typ,exc_val,exc_fr)
  86       raise SALOME.SALOME_Exception(SALOME.ExceptionStruct(SALOME.BAD_PARAM,"".join(l),"PyNode: %s, function: %s" % (self.nodeName,funcName),0))
  87
  88 class PyScriptNode_i (Engines__POA.PyScriptNode,Generic):
  89   """The implementation of the PyScriptNode CORBA IDL that executes a script"""
  90   def __init__(self, nodeName,code,poa,my_container):
  91     """Initialize the node : compilation in the local context"""
  92     Generic.__init__(self,poa)
  93     self.nodeName=nodeName
  94     self.code=code
  95     self.my_container=my_container._container
  96     linecache.cache[nodeName]=0,None,string.split(code,'\n'),nodeName
  97     self.ccode=compile(code,nodeName,'exec')
  98     self.context={}
  99     self.context["my_container"] = self.my_container
 100
 101   def executeAnotherPieceOfCode(self,code):
 102     """Called for initialization of container lodging self."""
 103     try:
 104       ccode=compile(code,self.nodeName,'exec')
 105       exec ccode in self.context
 106     except:
 107       raise SALOME.SALOME_Exception(SALOME.ExceptionStruct(SALOME.BAD_PARAM,"","PyScriptNode (%s) : code to be executed \"%s\"" %(self.nodeName,code),0))
 108
 109   def assignNewCompiledCode(self,codeStr):
 110     try:
 111       self.code=codeStr
 112       self.ccode=compile(codeStr,self.nodeName,'exec')
 113     except:
 114       raise SALOME.SALOME_Exception(SALOME.ExceptionStruct(SALOME.BAD_PARAM,"","PyScriptNode.assignNewCompiledCode (%s) : code to be executed \"%s\"" %(self.nodeName,codeStr),0))
 115
 116   def execute(self,outargsname,argsin):
 117     """Execute the script stored in attribute ccode with pickled args (argsin)"""
 118     try:
 119       argsname,kws=cPickle.loads(argsin)
 120       self.context.update(kws)
 121       exec self.ccode in self.context
 122       argsout=[]
 123       for arg in outargsname:
 124         if not self.context.has_key(arg):
 125           raise KeyError("There is no variable %s in context" % arg)
 126         argsout.append(self.context[arg])
 127       argsout=cPickle.dumps(tuple(argsout),-1)
 128       return argsout
 129     except:
 130       exc_typ,exc_val,exc_fr=sys.exc_info()
 131       l=traceback.format_exception(exc_typ,exc_val,exc_fr)
 132       raise SALOME.SALOME_Exception(SALOME.ExceptionStruct(SALOME.BAD_PARAM,"".join(l),"PyScriptNode: %s, outargsname: %s" % (self.nodeName,outargsname),0))