89 lines
3.5 KiB
Python
89 lines
3.5 KiB
Python
import unittest
|
|
|
|
import pythoncom
|
|
import win32api
|
|
import win32com.test.util
|
|
from win32com import storagecon
|
|
|
|
|
|
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()
|