62 lines
3.3 KiB
Python
62 lines
3.3 KiB
Python
|
from win32com import storagecon
|
||
|
|
import pythoncom, os, win32api
|
||
|
|
import win32com.test.util
|
||
|
|
|
||
|
|
import unittest
|
||
|
|
|
||
|
|
class TestEnum(win32com.test.util.TestCase):
|
||
|
|
def testit(self):
|
||
|
|
fname, tmp = win32api.GetTempFileName(win32api.GetTempPath(),'stg')
|
||
|
|
m=storagecon.STGM_READWRITE | storagecon.STGM_SHARE_EXCLUSIVE
|
||
|
|
## file, mode, format, attrs (always 0), IID (IStorage or IPropertySetStorage, storage options(only used with STGFMT_DOCFILE)
|
||
|
|
pss=pythoncom.StgOpenStorageEx(fname, m, storagecon.STGFMT_FILE, 0 , pythoncom.IID_IPropertySetStorage)
|
||
|
|
### {"Version":2,"reserved":0,"SectorSize":512,"TemplateFile":u'somefilename'})
|
||
|
|
|
||
|
|
## FMTID_SummaryInformation FMTID_DocSummaryInformation FMTID_UserDefinedProperties
|
||
|
|
psuser=pss.Create(pythoncom.FMTID_UserDefinedProperties,
|
||
|
|
pythoncom.IID_IPropertySetStorage,
|
||
|
|
storagecon.PROPSETFLAG_DEFAULT,
|
||
|
|
storagecon.STGM_READWRITE|storagecon.STGM_CREATE|storagecon.STGM_SHARE_EXCLUSIVE) ## its very picky about flag combinations!
|
||
|
|
psuser.WriteMultiple((3,4),('hey','bubba'))
|
||
|
|
psuser.WritePropertyNames((3,4),('property3','property4'))
|
||
|
|
expected_summaries = []
|
||
|
|
expected_summaries.append( ('property3', 3, pythoncom.VT_BSTR))
|
||
|
|
expected_summaries.append( ('property4', 4, pythoncom.VT_BSTR))
|
||
|
|
psuser=None
|
||
|
|
|
||
|
|
pssum=pss.Create(pythoncom.FMTID_SummaryInformation,
|
||
|
|
pythoncom.IID_IPropertySetStorage,
|
||
|
|
storagecon.PROPSETFLAG_DEFAULT,
|
||
|
|
storagecon.STGM_READWRITE|storagecon.STGM_CREATE|storagecon.STGM_SHARE_EXCLUSIVE)
|
||
|
|
pssum.WriteMultiple((storagecon.PIDSI_AUTHOR,storagecon.PIDSI_COMMENTS),('me', 'comment'))
|
||
|
|
|
||
|
|
pssum=None
|
||
|
|
pss=None ## doesn't seem to be a close or release method, and you can't even reopen it from the same process until previous object is gone
|
||
|
|
|
||
|
|
pssread=pythoncom.StgOpenStorageEx(fname, storagecon.STGM_READ|storagecon.STGM_SHARE_EXCLUSIVE, storagecon.STGFMT_FILE, 0 , pythoncom.IID_IPropertySetStorage)
|
||
|
|
found_summaries = []
|
||
|
|
for psstat in pssread:
|
||
|
|
ps=pssread.Open(psstat[0],storagecon.STGM_READ|storagecon.STGM_SHARE_EXCLUSIVE)
|
||
|
|
for p in ps:
|
||
|
|
p_val = ps.ReadMultiple((p[1],))[0]
|
||
|
|
if (p[1]==storagecon.PIDSI_AUTHOR and p_val=='me') or \
|
||
|
|
(p[1]==storagecon.PIDSI_COMMENTS and p_val=='comment'):
|
||
|
|
pass
|
||
|
|
else:
|
||
|
|
self.fail("Uxexpected property %s/%s" % (p, p_val))
|
||
|
|
ps=None
|
||
|
|
## FMTID_UserDefinedProperties can't exist without FMTID_DocSummaryInformation, and isn't returned independently from Enum
|
||
|
|
## also can't be open at same time
|
||
|
|
if psstat[0]==pythoncom.FMTID_DocSummaryInformation:
|
||
|
|
ps=pssread.Open(pythoncom.FMTID_UserDefinedProperties,storagecon.STGM_READ|storagecon.STGM_SHARE_EXCLUSIVE)
|
||
|
|
for p in ps:
|
||
|
|
found_summaries.append(p)
|
||
|
|
ps=None
|
||
|
|
psread=None
|
||
|
|
expected_summaries.sort()
|
||
|
|
found_summaries.sort()
|
||
|
|
self.assertEqual(expected_summaries, found_summaries)
|
||
|
|
|
||
|
|
if __name__=='__main__':
|
||
|
|
unittest.main()
|
||
|
|
|