PC World 2002 May/PCWorld_2002-05_cd.bin/Software/TemaCD/activepython/ActivePython-2.1.1.msi/Python21_win32com_test_testPersist.py

Python Source | 2001-07-26 | 5.8 KB | 207 lines

import pythoncom
import win32com.server.util
import util
import time

import win32com, sys, string, win32api, traceback
import win32com.client.dynamic
import win32com.client
import pythoncom
from win32com.axcontrol import axcontrol
from pywintypes import Unicode
from win32com import storagecon
from win32com.test.util import CheckClean

import pywintypes
import win32ui
import win32api, os

S_OK = 0

class LockBytes:
  _public_methods_ = [ 'ReadAt', 'WriteAt', 'Flush', 'SetSize', 'LockRegion', 'UnlockRegion', 'Stat' ]
  _com_interfaces_ = [ pythoncom.IID_ILockBytes ]

  def __init__(self, data = ""):
    self.data = data
    now = pywintypes.Time(time.time())
    self.ctime = now
    self.mtime = now
    self.atime = now

  def ReadAt(self, offset, cb):
    print "ReadAt"
    result = self.data[offset:offset + cb]
    return result

  def WriteAt(self, offset, data):
    print "WriteAt " +str(offset)
    print "len " + str(len(data))
    print "data:"
    #print data
    if len(self.data) >= offset:
      newdata = self.data[0:offset] + data
    print len(newdata)
    if len(self.data) >= offset + len(data):
      newdata = newdata + self.data[offset +  len(data):]
    print len(newdata)
    self.data = newdata
    return len(data)

  def Flush(self, whatsthis=0):
    print "Flush" + str(whatsthis)
    fname = os.path.join(win32api.GetTempPath(), "persist.doc")
    open(fname, "wb").write(self.data)
    return S_OK

  def SetSize(self, size):
    print "Set Size" + str(size)
    if size > len(self.data):
      self.data = self.data +  "\000" * (size - len(self.data))
    else:
      self.data = self.data[0:size]
    return S_OK

  def LockRegion(self, offset, size, locktype):
    print "LockRegion"
    pass

  def UnlockRegion(self, offset, size, locktype):
    print "UnlockRegion"
    pass

  def Stat(self, statflag):
    print "returning Stat " + str(statflag)
    return (
      "PyMemBytes",
      storagecon.STGTY_LOCKBYTES,
      len(self.data),
      self.mtime,
      self.ctime,
      self.atime,
      storagecon.STGM_DIRECT|storagecon.STGM_READWRITE|storagecon.STGM_CREATE ,
      storagecon.STGM_SHARE_EXCLUSIVE,
      "{00020905-0000-0000-C000-000000000046}",
      0,   # statebits ?
      0
      )


class OleClientSite:
  _public_methods_ = [ 'SaveObject', 'GetMoniker', 'GetContainer', 'ShowObject', 'OnShowWindow', 'RequestNewObjectLayout' ]
  _com_interfaces_ = [ axcontrol.IID_IOleClientSite ]

  def __init__(self, data = ""):
    self.IPersistStorage = None
    self.IStorage = None

  def SetIPersistStorage(self, IPersistStorage):
    self.IPersistStorage = IPersistStorage

  def SetIStorage(self, IStorage):
    self.IStorage = IStorage

  def SaveObject(self):
    print "SaveObject"
    if self.IPersistStorage != None and self.IStorage != None:
      self.IPersistStorage.Save(self.IStorage,1)
      self.IStorage.Commit(0)
    return S_OK

  def GetMoniker(self, dwAssign, dwWhichMoniker):
    print "GetMoniker " + str(dwAssign) + " " + str(dwWhichMoniker)

  def GetContainer(self):
    print "GetContainer"

  def ShowObject(self):
    print "ShowObject"

  def OnShowWindow(self, fShow):
    print "ShowObject" + str(fShow)

  def RequestNewObjectLayout(self):
    print "RequestNewObjectLayout"


def test():
    # create a LockBytes object and
    #wrap it as a COM object
#    import win32com.server.dispatcher
    lbcom = win32com.server.util.wrap(LockBytes(), pythoncom.IID_ILockBytes) #, useDispatcher=win32com.server.dispatcher.DispatcherWin32trace)

    # create a structured storage on the ILockBytes object
    stcom = pythoncom.StgCreateDocfileOnILockBytes(lbcom, storagecon.STGM_DIRECT| storagecon.STGM_CREATE | storagecon.STGM_READWRITE | storagecon.STGM_SHARE_EXCLUSIVE, 0)

    # create our ClientSite
    ocs = OleClientSite()
    # wrap it as a COM object
    ocscom = win32com.server.util.wrap(ocs, axcontrol.IID_IOleClientSite)

    # create a Word OLE Document, connect it to our site and our storage
    oocom=axcontrol.OleCreate("{00020906-0000-0000-C000-000000000046}",
        axcontrol.IID_IOleObject,
        0,
        (0,),
        ocscom,
        stcom,
        )

    mf=win32ui.GetMainFrame()
    hwnd=mf.GetSafeHwnd()

    # Set the host and document name
    # for unknown reason document name becomes hostname, and document name
    # is not set, debugged it, but don't know where the problem is?
    oocom.SetHostNames("OTPython", "This is Cool")

    # activate the OLE document
    oocom.DoVerb( -1, ocscom, 0, hwnd, mf.GetWindowRect())

    # set the hostnames again
    oocom.SetHostNames("OTPython2", "ThisisCool2")

    # get IDispatch of Word
    doc=win32com.client.Dispatch(oocom.QueryInterface(pythoncom.IID_IDispatch))

    # get IPersistStorage of Word
    dpcom=oocom.QueryInterface(pythoncom.IID_IPersistStorage)

    # let our ClientSite know the interfaces
    ocs.SetIPersistStorage(dpcom)
    ocs.SetIStorage(stcom)

    # use IDispatch to do the Office Word test
    # pasted from TestOffice.py

    wrange = doc.Range()
    for i in range(10):
        wrange.InsertAfter("Hello from Python %d\n" % i)
    paras = doc.Paragraphs
    for i in range(len(paras)):
        paras[i]().Font.ColorIndex = i+1
        paras[i]().Font.Size = 12 + (4 * i)
    # XXX - note that
    # for para in paras:
    #     para().Font...
    # doesnt seem to work - no error, just doesnt work
    # Should check if it works for VB!


    dpcom.Save(stcom, 0)
    dpcom.HandsOffStorage()
#    oocom.Close(axcontrol.OLECLOSE_NOSAVE) # or OLECLOSE_SAVEIFDIRTY, but it fails???

    #Save the ILockBytes data to "persist2.doc"
    lbcom.Flush()

    #exiting Winword will automatically update the ILockBytes data
    #and flush it to "%TEMP%\persist.doc"
    doc.Application.Quit()

if __name__=='__main__':
    test()
    pythoncom.CoUninitialize()
    CheckClean()