"""AXScript Client Framework
This module provides a core framework for an ActiveX Scripting client.
Derived classes actually implement the AX Client itself, including the
scoping rules, etc.
There are classes defined for the engine itself, and for ScriptItems
"""
import sys
from win32com.axscript import axscript
import win32com.server.util
import win32com.client.connect # Need simple connection point support
import win32api, winerror
import pythoncom
import types
import re
def RemoveCR(text):
# No longer just "RemoveCR" - should be renamed to
# FixNewlines, or something. Idea is to fix arbitary newlines into
# something Python can compile...
return re.sub('(\r\n)|\r|(\n\r)','\n',text)
SCRIPTTEXT_FORCEEXECUTION = -2147483648 # 0x80000000
SCRIPTTEXT_ISEXPRESSION = 0x00000020
SCRIPTTEXT_ISPERSISTENT = 0x00000040
from win32com.server.exception import Exception,IsCOMServerException
import error # ax.client.error
state_map = {
axscript.SCRIPTSTATE_UNINITIALIZED: "SCRIPTSTATE_UNINITIALIZED",
axscript.SCRIPTSTATE_INITIALIZED: "SCRIPTSTATE_INITIALIZED",
axscript.SCRIPTSTATE_STARTED: "SCRIPTSTATE_STARTED",
axscript.SCRIPTSTATE_CONNECTED: "SCRIPTSTATE_CONNECTED",
axscript.SCRIPTSTATE_DISCONNECTED: "SCRIPTSTATE_DISCONNECTED",
axscript.SCRIPTSTATE_CLOSED: "SCRIPTSTATE_CLOSED",
}
def profile(fn, *args):
import profile
prof = profile.Profile()
try:
# roll on 1.6 :-)
# return prof.runcall(fn, *args)
return prof.runcall(*(fn,) + args)
finally:
import pstats
# Damn - really want to send this to Excel!
# width, list = pstats.Stats(prof).strip_dirs().get_print_list([])
pstats.Stats(prof).strip_dirs().sort_stats("time").print_stats()
class SafeOutput:
softspace=1
def __init__(self, redir=None):
if redir is None: redir = sys.stdout
self.redir=redir
def write(self,message):
try:
self.redir.write(message)
except:
win32api.OutputDebugString(message.encode('mbcs'))
def flush(self):
pass
def close(self):
pass
# Make sure we have a valid sys.stdout/stderr, otherwise out
# print and trace statements may raise an exception
def MakeValidSysOuts():
if not isinstance(sys.stdout, SafeOutput):
sys.stdout = sys.stderr = SafeOutput()
# and for the sake of working around something I can't understand...
# prevent keyboard interrupts from killing IIS
import signal
def noOp(a,b):
# it would be nice to get to the bottom of this, so a warning to
# the debug console can't hurt.
print "WARNING: Ignoring keyboard interrupt from ActiveScripting engine"
# If someone else has already redirected, then assume they know what they are doing!
if signal.getsignal(signal.SIGINT) == signal.default_int_handler:
try:
signal.signal(signal.SIGINT, noOp)
except ValueError:
# Not the main thread - can't do much.
pass
def trace(*args):
"""A function used instead of "print" for debugging output.
"""
for arg in args:
print arg,
print
def RaiseAssert(scode, desc):
"""A debugging function that raises an exception considered an "Assertion".
"""
print "**************** ASSERTION FAILED *******************"
print desc
raise Exception(scode, desc)
class AXScriptCodeBlock:
"""An object which represents a chunk of code in an AX Script
"""
def __init__(self, name, codeText, sourceContextCookie, startLineNumber, flags):
self.name = name
self.codeText = codeText
self.codeObject = None
self.sourceContextCookie = sourceContextCookie
self.startLineNumber = startLineNumber
self.flags = flags
self.beenExecuted = 0
def GetFileName(self):
# Gets the "file name" for Python - uses <...> so Python doesnt think
# it is a real file.
return "<%s>" % self.name
def GetDisplayName(self):
return self.name
def GetLineNo(self, no):
pos = -1
for i in range(no-1):
pos = self.codeText.find('\n', pos+1)
if pos==-1: pos=len(self.codeText)
epos = self.codeText.find('\n', pos+1)
if epos==-1:
epos=len(self.codeText)
return self.codeText[pos+1:epos].strip()
class Event:
"""A single event for a ActiveX named object.
"""
def __init__(self):
self.name = "<None>"
def __repr__(self):
return "<%s at %d: %s>" % (self.__class__.__name__, id(self), self.name)
def Reset(self):
pass
def Close(self):
pass
def Build(self, typeinfo, funcdesc):
self.dispid = funcdesc[0]
self.name = typeinfo.GetNames(self.dispid)[0]
# print "Event.Build() - Event Name is ", self.name
class EventSink:
"""A set of events against an item. Note this is a COM client for connection points.
"""
_public_methods_ = []
def __init__(self, myItem, coDispatch):
self.events = {}
self.connection = None
self.coDispatch = coDispatch
self.myScriptItem = myItem
self.myInvokeMethod = myItem.GetEngine().ProcessScriptItemEvent
self.iid = None
def Reset(self):
self.Disconnect()
def Close(self):
self.iid = None
self.myScriptItem = None
self.myInvokeMethod = None
self.coDispatch = None
for event in self.events.itervalues():
event.Reset()
self.events = {}
self.Disconnect()
# COM Connection point methods.
def _query_interface_(self, iid):
if iid==self.iid:
return win32com.server.util.wrap(self)
def _invoke_(self, dispid, lcid, wFlags, args):
try:
event = self.events[dispid]
except:
raise Exception(scode=winerror.DISP_E_MEMBERNOTFOUND)
#print "Invoke for ", event, "on", self.myScriptItem, " - calling", self.myInvokeMethod
return self.myInvokeMethod(self.myScriptItem, event, lcid, wFlags, args)
def GetSourceTypeInfo(self, typeinfo):
"""Gets the typeinfo for the Source Events for the passed typeinfo"""
attr = typeinfo.GetTypeAttr()
cFuncs = attr[6]
typeKind = attr[5]
if typeKind not in [pythoncom.TKIND_COCLASS, pythoncom.TKIND_INTERFACE]:
RaiseAssert(winerror.E_UNEXPECTED, "The typeKind of the object is unexpected")
cImplType = attr[8]
for i in xrange(cImplType):
# Look for the [source, default] interface on the coclass
# that isn't marked as restricted.
flags = typeinfo.GetImplTypeFlags(i)
flagsNeeded = pythoncom.IMPLTYPEFLAG_FDEFAULT | pythoncom.IMPLTYPEFLAG_FSOURCE
if (flags & ( flagsNeeded | pythoncom.IMPLTYPEFLAG_FRESTRICTED))==(flagsNeeded):
# Get the handle to the implemented interface.
href = typeinfo.GetRefTypeOfImplType(i)
return typeinfo.GetRefTypeInfo(href)
def BuildEvents(self):
# See if it is an extender object.
try:
mainTypeInfo = self.coDispatch.QueryInterface(axscript.IID_IProvideMultipleClassInfo)
isMulti = 1
numTypeInfos = mainTypeInfo.GetMultiTypeInfoCount()
except pythoncom.com_error:
isMulti = 0
numTypeInfos = 1
try:
mainTypeInfo = self.coDispatch.QueryInterface(pythoncom.IID_IProvideClassInfo)
except pythoncom.com_error:
numTypeInfos = 0
# Create an event handler for the item.
for item in xrange(numTypeInfos):
if isMulti:
typeinfo, flags = mainTypeInfo.GetInfoOfIndex(item, axscript.MULTICLASSINFO_GETTYPEINFO)
else:
typeinfo = mainTypeInfo.GetClassInfo()
sourceType = self.GetSourceTypeInfo(typeinfo)
cFuncs = 0
if sourceType:
attr = sourceType.GetTypeAttr()
self.iid = attr[0]
cFuncs = attr[6]
for i in xrange(cFuncs):
funcdesc = sourceType.GetFuncDesc(i)
event = Event()
event.Build(sourceType, funcdesc)
self.events[event.dispid] = event
def Connect(self):
if self.connection is not None or self.iid is None: return
# trace("Connect for sink item", self.myScriptItem.name, "with IID",str(self.iid))
self.connection = win32com.client.connect.SimpleConnection(self.coDispatch, self, self.iid)
def Disconnect(self):
if self.connection:
try:
self.connection.Disconnect()
except pythoncom.com_error:
pass # Ignore disconnection errors.
self.connection = None
class ScriptItem:
"""An item (or subitem) that is exposed to the ActiveX script
"""
def __init__(self, parentItem, name, dispatch, flags):
self.parentItem = parentItem
self.dispatch = dispatch
self.name = name
self.flags = flags
self.eventSink = None
self.subItems = {}
self.createdConnections = 0
self.isRegistered = 0
# trace("Creating ScriptItem", name, "of parent", parentItem,"with dispatch", dispatch)
def __repr__(self):
flagsDesc=""
if self.flags is not None and self.flags & axscript.SCRIPTITEM_GLOBALMEMBERS:
flagsDesc = "/Global"
return "<%s at %d: %s%s>" % (self.__class__.__name__, id(self), self.name,flagsDesc)
def _dump_(self, level):
flagDescs = []
if self.flags is not None and self.flags & axscript.SCRIPTITEM_GLOBALMEMBERS:
flagDescs.append("GLOBAL!")
if self.flags is None or self.flags & axscript.SCRIPTITEM_ISVISIBLE == 0:
flagDescs.append("NOT VISIBLE")
if self.flags is not None and self.flags & axscript.SCRIPTITEM_ISSOURCE:
flagDescs.append("EVENT SINK")
if self.flags is not None and self.flags & axscript.SCRIPTITEM_CODEONLY:
flagDescs.append("CODE ONLY")
print " " * level, "Name=", self.name, ", flags=", "/".join(flagDescs), self
for subItem in self.subItems.itervalues():
subItem._dump_(level+1)
def Reset(self):
self.Disconnect()
if self.eventSink:
self.eventSink.Reset()
self.isRegistered = 0
for subItem in self.subItems.itervalues():
subItem.Reset()
def Close(self):
self.Reset()
self.dispatch = None
self.parentItem = None
if self.eventSink:
self.eventSink.Close()
self.eventSink = None
for subItem in self.subItems.itervalues():
subItem.Close()
self.subItems = []
self.createdConnections = 0
def Register(self):
if self.isRegistered: return
# Get the type info to use to build this item.
# if not self.dispatch:
# id = self.parentItem.dispatch.GetIDsOfNames(self.name)
# print "DispID of me is", id
# result = self.parentItem.dispatch.Invoke(id, 0, pythoncom.DISPATCH_PROPERTYGET,1)
# if type(result)==pythoncom.TypeIIDs[pythoncom.IID_IDispatch]:
# self.dispatch = result
# else:
# print "*** No dispatch"
# return
# print "**** Made dispatch"
self.isRegistered = 1
# Register the sub-items.
for item in self.subItems.itervalues():
if not item.isRegistered:
item.Register()
def IsGlobal(self):
return self.flags & axscript.SCRIPTITEM_GLOBALMEMBERS
def IsVisible(self):
return (self.flags & (axscript.SCRIPTITEM_ISVISIBLE | axscript.SCRIPTITEM_ISSOURCE)) != 0
def GetEngine(self):
item = self
while item.parentItem.__class__==self.__class__:
item = item.parentItem
return item.parentItem
def _GetFullItemName(self):
ret = self.name
if self.parentItem:
try:
ret = self.parentItem._GetFullItemName() + "." + ret
except AttributeError:
pass
return ret
def GetSubItemClass(self):
return self.__class__
def GetSubItem(self, name):
return self.subItems[name.lower()]
def GetCreateSubItem(self, parentItem, name, dispatch, flags):
keyName = name.lower()
try:
rc = self.subItems[keyName]
# No changes allowed to existing flags.
if not rc.flags is None and not flags is None and rc.flags != flags:
raise Exception(scode=winerror.E_INVALIDARG)
# Existing item must not have a dispatch.
if not rc.dispatch is None and not dispatch is None:
raise Exception(scode=winerror.E_INVALIDARG)
rc.flags = flags # Setup the real flags.
rc.dispatch = dispatch
except KeyError:
rc = self.subItems[keyName] = self.GetSubItemClass()(parentItem, name, dispatch, flags)
return rc
# if self.dispatch is None:
# RaiseAssert(winerror.E_UNEXPECTED, "??")
def CreateConnections(self):
# Create (but do not connect to) the connection points.
if self.createdConnections: return
self.createdConnections = 1
# Nothing to do unless this is an event source
# This flags means self, _and_ children, are connectable.
if self.flags & axscript.SCRIPTITEM_ISSOURCE:
self.BuildEvents()
self.FindBuildSubItemEvents()
def Connect(self):
# Connect to the already created connection points.
if self.eventSink:
self.eventSink.Connect()
for subItem in self.subItems.itervalues():
subItem.Connect()
def Disconnect(self):
# Disconnect from the connection points.
if self.eventSink:
self.eventSink.Disconnect()
for subItem in self.subItems.itervalues():
subItem.Disconnect()
def BuildEvents(self):
if self.eventSink is not None or self.dispatch is None:
RaiseAssert(winerror.E_UNEXPECTED, "Item already has built events, or no dispatch available?")
# trace("BuildEvents for named item", self._GetFullItemName())
self.eventSink = EventSink(self, self.dispatch)
self.eventSink.BuildEvents()
def FindBuildSubItemEvents(self):
# Called during connection to event source. Seeks out and connects to
# all children. As per the AX spec, this is not recursive
# (ie, children sub-items are not seeked)
try:
multiTypeInfo = self.dispatch.QueryInterface(axscript.IID_IProvideMultipleClassInfo)
numTypeInfos = multiTypeInfo.GetMultiTypeInfoCount()
except pythoncom.com_error:
return
for item in xrange(numTypeInfos):
typeinfo, flags = multiTypeInfo.GetInfoOfIndex(item, axscript.MULTICLASSINFO_GETTYPEINFO)
defaultType = self.GetDefaultSourceTypeInfo(typeinfo)
index = 0
while 1:
try:
fdesc = defaultType.GetFuncDesc(index)
except pythoncom.com_error:
break # No more funcs
index = index + 1
dispid = fdesc[0]
funckind = fdesc[3]
invkind = fdesc[4]
elemdesc = fdesc[8]
funcflags = fdesc[9]
try:
isSubObject = not (funcflags & pythoncom.FUNCFLAG_FRESTRICTED) and \
funckind == pythoncom.FUNC_DISPATCH and \
invkind == pythoncom.INVOKE_PROPERTYGET and \
elemdesc[0][0] == pythoncom.VT_PTR and \
elemdesc[0][1][0] == pythoncom.VT_USERDEFINED
except:
isSubObject = 0
if isSubObject:
try:
# We found a sub-object.
names = typeinfo.GetNames(dispid);
result = self.dispatch.Invoke(dispid, 0x0, pythoncom.DISPATCH_PROPERTYGET, 1)
# IE has an interesting problem - there are lots of synonyms for the same object. Eg
# in a simple form, "window.top", "window.window", "window.parent", "window.self"
# all refer to the same object. Our event implementation code does not differentiate
# eg, "window_onload" will fire for *all* objects named "window". Thus,
# "window" and "window.window" will fire the same event handler :(
# One option would be to check if the sub-object is indeed the
# parent object - however, this would stop "top_onload" from firing,
# as no event handler for "top" would work.
# I think we simply need to connect to a *single* event handler.
# As use in IE is deprecated, I am not solving this now.
if type(result)==pythoncom.TypeIIDs[pythoncom.IID_IDispatch]:
name = names[0]
subObj = self.GetCreateSubItem(self, name, result, axscript.SCRIPTITEM_ISVISIBLE)
#print "subobj", name, "flags are", subObj.flags, "mydisp=", self.dispatch, "result disp=", result, "compare=", self.dispatch==result
subObj.BuildEvents()
subObj.Register()
except pythoncom.com_error:
pass
def GetDefaultSourceTypeInfo(self, typeinfo):
"""Gets the typeinfo for the Default Dispatch for the passed typeinfo"""
attr = typeinfo.GetTypeAttr()
cFuncs = attr[6]
typeKind = attr[5]
if typeKind not in [pythoncom.TKIND_COCLASS, pythoncom.TKIND_INTERFACE]:
RaiseAssert(winerror.E_UNEXPECTED, "The typeKind of the object is unexpected")
cImplType = attr[8]
for i in xrange(cImplType):
# Look for the [source, default] interface on the coclass
# that isn't marked as restricted.
flags = typeinfo.GetImplTypeFlags(i)
if (flags & ( pythoncom.IMPLTYPEFLAG_FDEFAULT | pythoncom.IMPLTYPEFLAG_FSOURCE | pythoncom.IMPLTYPEFLAG_FRESTRICTED))==pythoncom.IMPLTYPEFLAG_FDEFAULT:
# Get the handle to the implemented interface.
href = typeinfo.GetRefTypeOfImplType(i)
defTypeInfo = typeinfo.GetRefTypeInfo(href)
attr = defTypeInfo.GetTypeAttr()
typeKind = attr[5]
typeFlags = attr[11]
if typeKind == pythoncom.TKIND_INTERFACE and typeFlags & pythoncom.TYPEFLAG_FDUAL:
# Get corresponding Disp interface
# -1 is a special value which does this for us.
href = typeinfo.GetRefTypeOfImplType(-1)
return defTypeInfo.GetRefTypeInfo(href)
else:
return defTypeInfo
IActiveScriptMethods = [
"SetScriptSite", "GetScriptSite", "SetScriptState", "GetScriptState",
"Close", "AddNamedItem", "AddTypeLib", "GetScriptDispatch",
"GetCurrentScriptThreadID", "GetScriptThreadID", "GetScriptThreadState",
"InterruptScriptThread", "Clone" ]
IActiveScriptParseMethods = [
"InitNew", "AddScriptlet", "ParseScriptText" ]
IObjectSafetyMethods = [
"GetInterfaceSafetyOptions", "SetInterfaceSafetyOptions"]
# ActiveScriptParseProcedure is a new interface with IIS4/IE4.
IActiveScriptParseProcedureMethods = ['ParseProcedureText']
class COMScript:
"""An ActiveX Scripting engine base class.
This class implements the required COM interfaces for ActiveX scripting.
"""
_public_methods_ = IActiveScriptMethods + IActiveScriptParseMethods + IObjectSafetyMethods + IActiveScriptParseProcedureMethods
_com_interfaces_ = [axscript.IID_IActiveScript, axscript.IID_IActiveScriptParse, axscript.IID_IObjectSafety] #, axscript.IID_IActiveScriptParseProcedure]
def __init__(self):
# Make sure we can print/trace wihout an exception!
MakeValidSysOuts()
# trace("AXScriptEngine object created", self)
self.baseThreadId = -1
self.debugManager = None
self.threadState = axscript.SCRIPTTHREADSTATE_NOTINSCRIPT
self.scriptState = axscript.SCRIPTSTATE_UNINITIALIZED
self.scriptSite = None
self.safetyOptions = 0
self.lcid = 0
self.subItems = {}
self.scriptCodeBlocks = {}
def _query_interface_(self, iid):
if self.debugManager:
return self.debugManager._query_interface_for_debugger_(iid)
# trace("ScriptEngine QI - unknown IID", iid)
return 0
# IActiveScriptParse
def InitNew(self):
if self.scriptSite is not None:
self.SetScriptState(axscript.SCRIPTSTATE_INITIALIZED)
def AddScriptlet(self, defaultName, code, itemName, subItemName, eventName, delimiter, sourceContextCookie, startLineNumber):
# trace ("AddScriptlet", defaultName, code, itemName, subItemName, eventName, delimiter, sourceContextCookie, startLineNumber)
self.DoAddScriptlet(defaultName, code, itemName, subItemName, eventName, delimiter,sourceContextCookie, startLineNumber)
def ParseScriptText(self, code, itemName, context, delimiter, sourceContextCookie, startLineNumber, flags, bWantResult):
# trace ("ParseScriptText", code[:20],"...", itemName, context, delimiter, sourceContextCookie, startLineNumber, flags, bWantResult)
if bWantResult or self.scriptState == axscript.SCRIPTSTATE_STARTED \
or self.scriptState == axscript.SCRIPTSTATE_CONNECTED \
or self.scriptState == axscript.SCRIPTSTATE_DISCONNECTED :
flags = flags | SCRIPTTEXT_FORCEEXECUTION
else:
flags = flags & (~SCRIPTTEXT_FORCEEXECUTION)
if flags & SCRIPTTEXT_FORCEEXECUTION:
# About to execute the code.
self.RegisterNewNamedItems()
return self.DoParseScriptText(code, sourceContextCookie, startLineNumber, bWantResult, flags)
#
# IActiveScriptParseProcedure
def ParseProcedureText( self, code, formalParams, procName, itemName, unkContext, delimiter, contextCookie, startingLineNumber, flags):
trace("ParseProcedureText", code, formalParams, procName, itemName, unkContext, delimiter, contextCookie, startingLineNumber, flags)
# NOTE - this is never called, as we have disabled this interface.
# Problem is, once enabled all even code comes via here, rather than AddScriptlet.
# However, the "procName" is always an empty string - ie, itemName is the object whose event we are handling,
# but no idea what the specific event is!?
# Problem is disabling this block is that AddScriptlet is _not_ passed
# <SCRIPT for="whatever" event="onClick" language="Python">
# (but even for those blocks, the "onClick" information is still missing!?!?!?)
# self.DoAddScriptlet(None, code, itemName, subItemName, eventName, delimiter,sourceContextCookie, startLineNumber)
return None
#
# IActiveScript
def SetScriptSite(self, site):
# We should still work with an existing site (or so MSXML believes :)
self.scriptSite = site
if self.debugManager is not None:
self.debugManager.Close()
import traceback
try:
import win32com.axdebug.axdebug # see if the core exists.
import debug
self.debugManager = debug.DebugManager(self)
except pythoncom.com_error:
# COM errors will occur if the debugger interface has never been
# seen on the target system
trace("Debugging interfaces not available - debugging is disabled..")
self.debugManager = None
except ImportError:
trace("Debugging extensions (axdebug) module does not exist - debugging is disabled..")
self.debugManager = None
except:
traceback.print_exc()
trace("*** Debugger Manager could not initialize - %s: %s" % (sys.exc_info()[0],sys.exc_info()[1]))
self.debugManager = None
try:
self.lcid = site.GetLCID()
except pythoncom.com_error:
self.lcid = win32api.GetUserDefaultLCID()
self.Reset()
def GetScriptSite(self, iid):
if self.scriptSite is None: raise Exception(scode=winerror.S_FALSE)
return self.scriptSite.QueryInterface(iid)
def SetScriptState(self, state):
#print "SetScriptState with %s - currentstate = %s" % (state_map.get(state),state_map.get(self.scriptState))
if state == self.scriptState: return
# If closed, allow no other state transitions
if self.scriptState==axscript.SCRIPTSTATE_CLOSED:
raise Exception(scode=winerror.E_INVALIDARG)
if state==axscript.SCRIPTSTATE_INITIALIZED:
# Re-initialize - shutdown then reset.
if self.scriptState in [axscript.SCRIPTSTATE_CONNECTED, axscript.SCRIPTSTATE_STARTED]:
self.Stop()
elif state==axscript.SCRIPTSTATE_STARTED:
if self.scriptState == axscript.SCRIPTSTATE_CONNECTED:
self.Disconnect()
if self.scriptState == axscript.SCRIPTSTATE_DISCONNECTED:
self.Reset()
self.Run()
self.ChangeScriptState(axscript.SCRIPTSTATE_STARTED)
elif state==axscript.SCRIPTSTATE_CONNECTED:
if self.scriptState in [axscript.SCRIPTSTATE_UNINITIALIZED,axscript.SCRIPTSTATE_INITIALIZED]:
self.ChangeScriptState(axscript.SCRIPTSTATE_STARTED) # report transition through started
self.Run()
if self.scriptState == axscript.SCRIPTSTATE_STARTED:
self.Connect()
self.ChangeScriptState(state)
elif state==axscript.SCRIPTSTATE_DISCONNECTED:
if self.scriptState == axscript.SCRIPTSTATE_CONNECTED:
self.Disconnect()
elif state==axscript.SCRIPTSTATE_CLOSED:
self.Close()
elif state==axscript.SCRIPTSTATE_UNINITIALIZED:
if self.scriptState == axscript.SCRIPTSTATE_STARTED:
self.Stop()
if self.scriptState == axscript.SCRIPTSTATE_CONNECTED:
self.Disconnect()
if self.scriptState == axscript.SCRIPTSTATE_DISCONNECTED:
self.Reset()
self.ChangeScriptState(state)
else:
raise Exception(scode=winerror.E_INVALIDARG)
def GetScriptState(self):
return self.scriptState
def Close(self):
# trace("Close")
if self.scriptState in [axscript.SCRIPTSTATE_CONNECTED, axscript.SCRIPTSTATE_DISCONNECTED]:
self.Stop()
if self.scriptState in [axscript.SCRIPTSTATE_CONNECTED, axscript.SCRIPTSTATE_DISCONNECTED, axscript.SCRIPTSTATE_INITIALIZED, axscript.SCRIPTSTATE_STARTED]:
pass # engine.close??
if self.scriptState in [axscript.SCRIPTSTATE_UNINITIALIZED, axscript.SCRIPTSTATE_CONNECTED, axscript.SCRIPTSTATE_DISCONNECTED, axscript.SCRIPTSTATE_INITIALIZED, axscript.SCRIPTSTATE_STARTED]:
self.ChangeScriptState(axscript.SCRIPTSTATE_CLOSED)
# Completely reset all named items (including persistent)
for item in self.subItems.itervalues():
item.Close()
self.subItems = {}
self.baseThreadId = -1
if self.debugManager:
self.debugManager.Close()
self.debugManager = None
self.scriptSite = None
self.scriptCodeBlocks = {}
self.persistLoaded = 0
def AddNamedItem(self, name, flags):
if self.scriptSite is None: raise Exception(scode=winerror.E_INVALIDARG)
try:
unknown = self.scriptSite.GetItemInfo(name, axscript.SCRIPTINFO_IUNKNOWN)[0]
dispatch = unknown.QueryInterface(pythoncom.IID_IDispatch)
except pythoncom.com_error:
raise Exception(scode=winerror.E_NOINTERFACE, desc="Object has no dispatch interface available.")
newItem = self.subItems[name] = self.GetNamedItemClass()(self, name, dispatch, flags)
if newItem.IsGlobal():
newItem.CreateConnections()
def GetScriptDispatch(self, name):
# Base classes should override.
raise Exception(scode=winerror.E_NOTIMPL)
def GetCurrentScriptThreadID(self):
return self.baseThreadId
def GetScriptThreadID(self, win32ThreadId):
if self.baseThreadId == -1:
raise Exception(scode=winerror.E_UNEXPECTED)
if self.baseThreadId != win32ThreadId:
raise Exception(scode=winerror.E_INVALIDARG)
return self.baseThreadId
def GetScriptThreadState(self, scriptThreadId):
if self.baseThreadId == -1:
raise Exception(scode=winerror.E_UNEXPECTED)
if scriptThreadId != self.baseThreadId:
raise Exception(scode=winerror.E_INVALIDARG)
return self.threadState
def AddTypeLib(self, uuid, major, minor, flags):
# Get the win32com gencache to register this library.
from win32com.client import gencache
gencache.EnsureModule(uuid, self.lcid, major, minor, bForDemand = 1)
# This is never called by the C++ framework - it does magic.
# See PyGActiveScript.cpp
#def InterruptScriptThread(self, stidThread, exc_info, flags):
# raise Exception("Not Implemented", scode=winerror.E_NOTIMPL)
def Clone(self):
raise Exception("Not Implemented", scode=winerror.E_NOTIMPL)
#
# IObjectSafety
# Note that IE seems to insist we say we support all the flags, even tho
# we dont accept them all. If unknown flags come in, they are ignored, and never
# reflected in GetInterfaceSafetyOptions and the QIs obviously fail, but still IE
# allows our engine to initialize.
def SetInterfaceSafetyOptions(self, iid, optionsMask, enabledOptions):
# trace ("SetInterfaceSafetyOptions", iid, optionsMask, enabledOptions)
if optionsMask & enabledOptions == 0:
return
# See comments above.
# if (optionsMask & enabledOptions & \
# ~(axscript.INTERFACESAFE_FOR_UNTRUSTED_DATA | axscript.INTERFACESAFE_FOR_UNTRUSTED_CALLER)):
# # request for options we don't understand
# RaiseAssert(scode=winerror.E_FAIL, desc="Unknown safety options")
if iid in [pythoncom.IID_IPersist, pythoncom.IID_IPersistStream, pythoncom.IID_IPersistStreamInit,
axscript.IID_IActiveScript, axscript.IID_IActiveScriptParse]:
supported = self._GetSupportedInterfaceSafetyOptions()
self.safetyOptions = supported & optionsMask & enabledOptions
else:
raise Exception(scode=winerror.E_NOINTERFACE)
def _GetSupportedInterfaceSafetyOptions(self):
return 0
def GetInterfaceSafetyOptions(self, iid):
if iid in [pythoncom.IID_IPersist, pythoncom.IID_IPersistStream, pythoncom.IID_IPersistStreamInit,
axscript.IID_IActiveScript, axscript.IID_IActiveScriptParse]:
supported = self._GetSupportedInterfaceSafetyOptions()
return supported, self.safetyOptions
else:
raise Exception(scode=winerror.E_NOINTERFACE)
#
# Other helpers.
def ExecutePendingScripts(self):
self.RegisterNewNamedItems()
self.DoExecutePendingScripts()
def ProcessScriptItemEvent(self, item, event, lcid, wFlags, args):
# trace("ProcessScriptItemEvent", item, event, lcid, wFlags, args)
self.RegisterNewNamedItems()
return self.DoProcessScriptItemEvent(item, event, lcid, wFlags, args)
def _DumpNamedItems_(self):
for item in self.subItems.itervalues():
item._dump_(0)
def ResetNamedItems(self):
# Due to the way we work, we re-create persistent ones.
self.subItems = {}
for name, item in self.subItems.iteritems():
item.Close()
if item.flags & axscript.SCRIPTITEM_ISPERSISTENT:
self.AddNamedItem(item.name, item.flags)
def GetCurrentSafetyOptions(self):
return self.safetyOptions
def ProcessNewNamedItemsConnections(self):
# Process all sub-items.
for item in self.subItems.itervalues():
if not item.createdConnections: # Fast-track!
item.CreateConnections()
def RegisterNewNamedItems(self):
# Register all sub-items.
for item in self.subItems.itervalues():
if not item.isRegistered: # Fast-track!
self.RegisterNamedItem(item)
def RegisterNamedItem(self, item):
item.Register()
def CheckConnectedOrDisconnected(self):
if self.scriptState in [axscript.SCRIPTSTATE_CONNECTED, axscript.SCRIPTSTATE_DISCONNECTED]:
return
RaiseAssert(winerror.E_UNEXPECTED, "Not connected or disconnected - %d" % self.scriptState)
def Connect(self):
self.ProcessNewNamedItemsConnections()
self.RegisterNewNamedItems()
self.ConnectEventHandlers()
def Run(self):
# trace("AXScript running...")
if self.scriptState != axscript.SCRIPTSTATE_INITIALIZED and self.scriptState != axscript.SCRIPTSTATE_STARTED:
raise Exception(scode=winerror.E_UNEXPECTED)
# self._DumpNamedItems_()
self.ExecutePendingScripts()
self.DoRun()
def Stop(self):
# Stop all executing scripts, and disconnect.
if self.scriptState == axscript.SCRIPTSTATE_CONNECTED:
self.Disconnect()
# Reset back to initialized.
self.Reset()
def Disconnect(self):
self.CheckConnectedOrDisconnected()
try:
self.DisconnectEventHandlers()
except pythoncom.com_error:
# Ignore errors when disconnecting.
pass
self.ChangeScriptState(axscript.SCRIPTSTATE_DISCONNECTED)
def ConnectEventHandlers(self):
# trace ("Connecting to event handlers")
for item in self.subItems.itervalues():
item.Connect()
self.ChangeScriptState(axscript.SCRIPTSTATE_CONNECTED);
def DisconnectEventHandlers(self):
# trace ("Disconnecting from event handlers")
for item in self.subItems.itervalues():
item.Disconnect()
def Reset(self):
# Keeping persistent engine state, reset back an initialized state
self.ResetNamedItems()
self.ChangeScriptState(axscript.SCRIPTSTATE_INITIALIZED)
def ChangeScriptState(self, state):
#print " ChangeScriptState with %s - currentstate = %s" % (state_map.get(state),state_map.get(self.scriptState))
self.DisableInterrupts()
try:
self.scriptState = state
try:
if self.scriptSite: self.scriptSite.OnStateChange(state)
except pythoncom.com_error, (hr, desc, exc, arg):
pass # Ignore all errors here - E_NOTIMPL likely from scriptlets.
finally:
self.EnableInterrupts()
# This stack frame is debugged - therefore we do as little as possible in it.
def _ApplyInScriptedSection(self, fn, args):
if self.debugManager:
self.debugManager.OnEnterScript()
if self.debugManager.adb.appDebugger:
return self.debugManager.adb.runcall(fn, *args)
else:
return fn(*args)
else:
return fn(*args)
def ApplyInScriptedSection(self, codeBlock, fn, args):
self.BeginScriptedSection()
try:
try:
# print "ApplyInSS", codeBlock, fn, args
return self._ApplyInScriptedSection(fn, args)
finally:
if self.debugManager: self.debugManager.OnLeaveScript()
self.EndScriptedSection()
except:
self.HandleException(codeBlock)
# This stack frame is debugged - therefore we do as little as possible in it.
def _CompileInScriptedSection(self, code, name, type):
if self.debugManager: self.debugManager.OnEnterScript()
return compile(code, name, type)
def CompileInScriptedSection(self, codeBlock, type, realCode = None):
if codeBlock.codeObject is not None: # already compiled
return 1
if realCode is None:
code = codeBlock.codeText
else:
code = realCode
name = codeBlock.GetFileName()
self.BeginScriptedSection()
try:
try:
codeObject = self._CompileInScriptedSection(RemoveCR(code), name, type)
codeBlock.codeObject = codeObject
return 1
finally:
if self.debugManager: self.debugManager.OnLeaveScript()
self.EndScriptedSection()
except:
self.HandleException(codeBlock)
# This stack frame is debugged - therefore we do as little as possible in it.
def _ExecInScriptedSection(self, codeObject, globals, locals = None):
if self.debugManager:
self.debugManager.OnEnterScript()
if self.debugManager.adb.appDebugger:
return self.debugManager.adb.run(codeObject, globals, locals)
else:
exec codeObject in globals, locals
else:
exec codeObject in globals, locals
def ExecInScriptedSection(self, codeBlock, globals, locals = None):
if locals is None: locals = globals
assert not codeBlock.beenExecuted, "This code block should not have been executed"
codeBlock.beenExecuted = 1
self.BeginScriptedSection()
try:
try:
self._ExecInScriptedSection(codeBlock.codeObject, globals, locals)
finally:
if self.debugManager: self.debugManager.OnLeaveScript()
self.EndScriptedSection()
except:
self.HandleException(codeBlock)
def _EvalInScriptedSection(self, codeBlock, globals, locals = None):
if self.debugManager:
self.debugManager.OnEnterScript()
if self.debugManager.adb.appDebugger:
return self.debugManager.adb.runeval(codeBlock, globals, locals)
else:
return eval(codeBlock, globals, locals)
else:
return eval(codeBlock, globals, locals)
def EvalInScriptedSection(self, codeBlock, globals, locals = None):
if locals is None: locals = globals
assert not codeBlock.beenExecuted, "This code block should not have been executed"
codeBlock.beenExecuted = 1
self.BeginScriptedSection()
try:
try:
return self._EvalInScriptedSection(codeBlock.codeObject, globals, locals)
finally:
if self.debugManager: self.debugManager.OnLeaveScript()
self.EndScriptedSection()
except:
self.HandleException(codeBlock)
def HandleException(self, codeBlock):
# NOTE - Never returns - raises a ComException
exc_type, exc_value, exc_traceback = sys.exc_info()
# If a SERVER exception, re-raise it. If a client side COM error, it is
# likely to have originated from the script code itself, and therefore
# needs to be reported like any other exception.
if IsCOMServerException(exc_type):
# Ensure the traceback doesnt cause a cycle.
exc_traceback = None
raise
# It could be an error by another script.
if issubclass(pythoncom.com_error, exc_type) and exc_value[0]==axscript.SCRIPT_E_REPORTED:
# Ensure the traceback doesnt cause a cycle.
exc_traceback = None
raise Exception(scode=exc_value[0])
exception = error.AXScriptException(self, \
codeBlock, exc_type, exc_value, exc_traceback)
# Ensure the traceback doesnt cause a cycle.
exc_traceback = None
result_exception = error.ProcessAXScriptException(self.scriptSite, self.debugManager, exception)
if result_exception is not None:
try:
self.scriptSite.OnScriptTerminate(None, result_exception)
except pythoncom.com_error:
pass # Ignore errors telling engine we stopped.
# reset ourselves to 'connected' so further events continue to fire.
self.SetScriptState(axscript.SCRIPTSTATE_CONNECTED)
raise result_exception
# I think that in some cases this should just return - but the code
# that could return None above is disabled, so it never happens.
RaiseAssert(winerror.E_UNEXPECTED, "Don't have an exception to raise to the caller!")
def BeginScriptedSection(self):
if self.scriptSite is None:
raise Exception(E_UNEXPECTED)
self.scriptSite.OnEnterScript()
def EndScriptedSection(self):
if self.scriptSite is None:
raise Exception(E_UNEXPECTED)
self.scriptSite.OnLeaveScript()
def DisableInterrupts(self):
pass
def EnableInterrupts(self):
pass
def GetNamedItem(self, name):
try:
return self.subItems[name]
except KeyError:
raise Exception(scode=winerror.E_INVALIDARG)
def GetNamedItemClass(self):
return ScriptItem
def _AddScriptCodeBlock(self, codeBlock):
self.scriptCodeBlocks[codeBlock.GetFileName()] = codeBlock
if self.debugManager:
self.debugManager.AddScriptBlock(codeBlock)
if __name__=='__main__':
print "This is a framework class - please use pyscript.py etc"
def dumptypeinfo(typeinfo):
return
attr = typeinfo.GetTypeAttr()
# Loop over all methods
print "Methods"
for j in xrange(attr[6]):
fdesc = list(typeinfo.GetFuncDesc(j))
id = fdesc[0]
try:
names = typeinfo.GetNames(id)
except pythoncom.ole_error:
names = None
doc = typeinfo.GetDocumentation(id)
print " ", names, "has attr", fdesc
# Loop over all variables (ie, properties)
print "Variables"
for j in xrange(attr[7]):
fdesc = list(typeinfo.GetVarDesc(j))
names = typeinfo.GetNames(id)
print " ", names, "has attr", fdesc
|