| import msilib,os,win32com,tempfile,sys |
| PCBUILD="PCBuild" |
| certname = None |
| from config import * |
| |
| Win64 = "amd64" in PCBUILD |
| |
| mod_dir = os.path.join(os.environ["ProgramFiles"], "Common Files", "Merge Modules") |
| msi = None |
| if len(sys.argv)==2: |
| msi = sys.argv[1] |
| if Win64: |
| modules = ["Microsoft_VC90_CRT_x86_x64.msm", "policy_9_0_Microsoft_VC90_CRT_x86_x64.msm"] |
| if not msi: msi = "python-%s.amd64.msi" % full_current_version |
| else: |
| modules = ["Microsoft_VC90_CRT_x86.msm","policy_9_0_Microsoft_VC90_CRT_x86.msm"] |
| if not msi: msi = "python-%s.msi" % full_current_version |
| for i, n in enumerate(modules): |
| modules[i] = os.path.join(mod_dir, n) |
| |
| def merge(msi, feature, rootdir, modules): |
| cab_and_filecount = [] |
| # Step 1: Merge databases, extract cabfiles |
| m = msilib.MakeMerge2() |
| m.OpenLog("merge.log") |
| print "Opened Log" |
| m.OpenDatabase(msi) |
| print "Opened DB" |
| for module in modules: |
| print module |
| m.OpenModule(module,0) |
| print "Opened Module",module |
| m.Merge(feature, rootdir) |
| print "Errors:" |
| for e in m.Errors: |
| print e.Type, e.ModuleTable, e.DatabaseTable |
| print " Modkeys:", |
| for s in e.ModuleKeys: print s, |
| print |
| print " DBKeys:", |
| for s in e.DatabaseKeys: print s, |
| print |
| cabname = tempfile.mktemp(suffix=".cab") |
| m.ExtractCAB(cabname) |
| cab_and_filecount.append((cabname, len(m.ModuleFiles))) |
| m.CloseModule() |
| m.CloseDatabase(True) |
| m.CloseLog() |
| |
| # Step 2: Add CAB files |
| i = msilib.MakeInstaller() |
| db = i.OpenDatabase(msi, win32com.client.constants.msiOpenDatabaseModeTransact) |
| |
| v = db.OpenView("SELECT LastSequence FROM Media") |
| v.Execute(None) |
| maxmedia = -1 |
| while 1: |
| r = v.Fetch() |
| if not r: break |
| seq = r.IntegerData(1) |
| if seq > maxmedia: |
| maxmedia = seq |
| print "Start of Media", maxmedia |
| |
| for cabname, count in cab_and_filecount: |
| stream = "merged%d" % maxmedia |
| msilib.add_data(db, "Media", |
| [(maxmedia+1, maxmedia+count, None, "#"+stream, None, None)]) |
| msilib.add_stream(db, stream, cabname) |
| os.unlink(cabname) |
| maxmedia += count |
| # The merge module sets ALLUSERS to 1 in the property table. |
| # This is undesired; delete that |
| v = db.OpenView("DELETE FROM Property WHERE Property='ALLUSERS'") |
| v.Execute(None) |
| v.Close() |
| db.Commit() |
| |
| merge(msi, "SharedCRT", "TARGETDIR", modules) |
| |
| # certname (from config.py) should be (a substring of) |
| # the certificate subject, e.g. "Python Software Foundation" |
| if certname: |
| os.system('signtool sign /n "%s" /t http://timestamp.verisign.com/scripts/timestamp.dll %s' % (certname, msi)) |