"""Python.Dictionary COM Server.
This module implements a simple COM server that acts much like a Python
dictionary or as a standard string-keyed VB Collection. The keys of
the dictionary are strings and are case-insensitive.
It uses a highly customized policy to fine-tune the behavior exposed to
the COM client.
The object exposes the following properties:
int Count (readonly)
VARIANT Item(BSTR key) (propget for Item)
Item(BSTR key, VARIANT value) (propput for Item)
Note that 'Item' is the default property, so the following forms of
VB code are acceptable:
set ob = CreateObject("Python.Dictionary")
ob("hello") = "there"
ob.Item("hi") = ob("HELLO")
All keys are defined, returning VT_NULL (None) if a value has not been
stored. To delete a key, simply assign VT_NULL to the key.
The object responds to the _NewEnum method by returning an enumerator over
the dictionary's keys. This allows for the following type of VB code:
for each name in ob
debug.print name, ob(name)
next
"""
import pythoncom
from win32com.server import util,policy
from win32com.server.exception import COMException
import winerror
import types
import pywintypes
from pythoncom import DISPATCH_METHOD,DISPATCH_PROPERTYGET
from winerror import S_OK
class DictionaryPolicy(policy.BasicWrapPolicy):
### BasicWrapPolicy looks for this
_com_interfaces_ = [ ]
### BasicWrapPolicy looks for this
_name_to_dispid_ = {
'item' : pythoncom.DISPID_VALUE,
'_newenum' : pythoncom.DISPID_NEWENUM,
'count' : 1,
}
### Auto-Registration process looks for these...
_reg_desc_ = 'Python Dictionary'
_reg_clsid_ = '{39b61048-c755-11d0-86fa-00c04fc2e03e}'
_reg_progid_ = 'Python.Dictionary'
_reg_verprogid_ = 'Python.Dictionary.1'
_reg_policy_spec_ = 'win32com.servers.dictionary.DictionaryPolicy'
def _CreateInstance_(self, clsid, reqIID):
self._wrap_({ })
return pythoncom.WrapObject(self, reqIID)
def _wrap_(self, ob):
self._obj_ = ob # ob should be a dictionary
def _invokeex_(self, dispid, lcid, wFlags, args, kwargs, serviceProvider):
if dispid == 0: # item
l = len(args)
if l < 1:
raise COMException(desc="not enough parameters", scode=winerror.DISP_E_BADPARAMCOUNT)
key = args[0]
if type(key) not in [str, unicode]:
### the nArgErr thing should be 0-based, not reversed... sigh
raise COMException(desc="Key must be a string", scode=winerror.DISP_E_TYPEMISMATCH)
key = key.lower()
if wFlags & (DISPATCH_METHOD | DISPATCH_PROPERTYGET):
if l > 1:
raise COMException(scode=winerror.DISP_E_BADPARAMCOUNT)
try:
return self._obj_[key]
except KeyError:
return None # unknown keys return None (VT_NULL)
if l != 2:
raise COMException(scode=winerror.DISP_E_BADPARAMCOUNT)
if args[1] is None:
# delete a key when None is assigned to it
try:
del self._obj_[key]
except KeyError:
pass
else:
self._obj_[key] = args[1]
return S_OK
if dispid == 1: # count
if not wFlags & DISPATCH_PROPERTYGET:
raise COMException(scode=winerror.DISP_E_MEMBERNOTFOUND) # not found
if len(args) != 0:
raise COMException(scode=winerror.DISP_E_BADPARAMCOUNT)
return len(self._obj_)
if dispid == pythoncom.DISPID_NEWENUM:
return util.NewEnum(self._obj_.keys())
raise COMException(scode=winerror.DISP_E_MEMBERNOTFOUND)
def _getidsofnames_(self, names, lcid):
### this is a copy of MappedWrapPolicy._getidsofnames_ ...
name = names[0].lower()
try:
return (self._name_to_dispid_[name],)
except KeyError:
raise COMException(scode=winerror.DISP_E_MEMBERNOTFOUND,
desc="Member not found")
def Register():
from win32com.server.register import UseCommandLine
return UseCommandLine(DictionaryPolicy)
if __name__ == '__main__':
Register()
|