home *** CD-ROM | disk | FTP | other *** search
- import string
- import win32com.axscript.axscript
- import winerror
- from win32com.axscript import axscript
- from win32com.server import exception, util
- import pythoncom
-
- class AXEngine:
- def __init__(self, site, engine):
- self.eScript = self.eParse = self.eSafety = None
- if type(engine) == type(''):
- engine = pythoncom.CoCreateInstance(engine,
- None,
- pythoncom.CLSCTX_SERVER,
- pythoncom.IID_IUnknown)
-
- self.eScript = engine.QueryInterface(axscript.IID_IActiveScript)
- self.eParse = engine.QueryInterface(axscript.IID_IActiveScriptParse)
- self.eSafety = engine.QueryInterface(axscript.IID_IObjectSafety)
-
- self.eScript.SetScriptSite(site)
- self.eParse.InitNew()
- def __del__(self):
- self.Close()
- def GetScriptDispatch(self, name = None):
- return self.eScript.GetScriptDispatch(name)
- def AddNamedItem(self, item, flags):
- return self.eScript.AddNamedItem(item, flags)
- # Some helpers.
- def AddCode(self, code, flags=0):
- self.eParse.ParseScriptText(code, None, None, None, 0, 0, flags)
- def EvalCode(self, code):
- return self.eParse.ParseScriptText(code, None, None, None, 0, 0, axscript.SCRIPTTEXT_ISEXPRESSION)
- def Start(self):
- # Should maybe check state?
- # Do I need to transition through?
- self.eScript.SetScriptState(axscript.SCRIPTSTATE_STARTED)
- # self.eScript.SetScriptState(axscript.SCRIPTSTATE_CONNECTED)
-
- def Close(self):
- if self.eScript:
- self.eScript.Close()
- self.eScript = self.eParse = self.eSafety = None
- def SetScriptState(self, state):
- self.eScript.SetScriptState(state)
-
- IActiveScriptSite_methods = [
- 'GetLCID',
- 'GetItemInfo',
- 'GetDocVersionString',
- 'OnScriptTerminate',
- 'OnStateChange',
- 'OnScriptError',
- 'OnEnterScript',
- 'OnLeaveScript',
- ]
-
- class AXSite:
- """An Active Scripting site. A Site can have exactly one engine.
- """
- _public_methods_ = IActiveScriptSite_methods
- _com_interfaces_ = [ axscript.IID_IActiveScriptSite ]
-
- def __init__(self, objModel={}, engine = None, lcid=0):
- self.lcid = lcid
-
- self.objModel = { }
- for name, object in objModel.items():
- # Gregs code did string.lower this - I think that is callers job if he wants!
- self.objModel[name] = object
-
- self.engine = None
- if engine:
- self._AddEngine(engine)
-
- def AddEngine(self, engine):
- """Adds a new engine to the site.
- engine can be a string, or a fully wrapped engine object.
- """
- if type(engine)==type(''):
- newEngine = AXEngine(util.wrap(self), engine)
- else:
- newEngine = engine
- self.engine = newEngine
- flags = axscript.SCRIPTITEM_ISVISIBLE | axscript.SCRIPTITEM_NOCODE | axscript.SCRIPTITEM_GLOBALMEMBERS | axscript.SCRIPTITEM_ISPERSISTENT
- for name in self.objModel.keys():
- newEngine.AddNamedItem(name, flags)
- newEngine.SetScriptState(axscript.SCRIPTSTATE_INITIALIZED)
- return newEngine
-
- # B/W compat
- _AddEngine = AddEngine
-
- def _Close(self):
- self.engine.Close()
- self.objModel = {}
-
- def GetLCID(self):
- return self.lcid
-
- def GetItemInfo(self, name, returnMask):
- # name = string.lower(name)
- if not self.objModel.has_key(name):
- raise exception.Exception(scode=winerror.TYPE_E_ELEMENTNOTFOUND, desc='item not found')
-
- ### for now, we don't have any type information
-
- if returnMask & axscript.SCRIPTINFO_IUNKNOWN:
- return (self.objModel[name], None)
-
- return (None, None)
-
- def GetDocVersionString(self):
- return 'Python AXHost version 1.0'
-
- def OnScriptTerminate(self, result, excepInfo):
- pass
-
- def OnStateChange(self, state):
- pass
-
- def OnScriptError(self, errorInterface):
- return winerror.S_FALSE
-
- def OnEnterScript(self):
- pass
-
- def OnLeaveScript(self):
- pass
-