Tools/msi/merge.py

   1 import msilib,os,win32com,tempfile,sys
   2 PCBUILD="PCBuild"
   3 certname = None
   4 from config import *
   5
   6 Win64 = "amd64" in PCBUILD
   7
   8 mod_dir = os.path.join(os.environ["ProgramFiles"], "Common Files", "Merge Modules")
   9 msi = None
  10 if len(sys.argv)==2:
  11     msi = sys.argv[1]
  12 if Win64:
  13     modules = ["Microsoft_VC90_CRT_x86_x64.msm", "policy_9_0_Microsoft_VC90_CRT_x86_x64.msm"]
  14     if not msi: msi = "python-%s.amd64.msi" % full_current_version
  15 else:
  16     modules = ["Microsoft_VC90_CRT_x86.msm","policy_9_0_Microsoft_VC90_CRT_x86.msm"]
  17     if not msi: msi = "python-%s.msi" % full_current_version
  18 for i, n in enumerate(modules):
  19     modules[i] = os.path.join(mod_dir, n)
  20
  21 def merge(msi, feature, rootdir, modules):
  22     cab_and_filecount = []
  23     # Step 1: Merge databases, extract cabfiles
  24     m = msilib.MakeMerge2()
  25     m.OpenLog("merge.log")
  26     print "Opened Log"
  27     m.OpenDatabase(msi)
  28     print "Opened DB"
  29     for module in modules:
  30         print module
  31         m.OpenModule(module,0)
  32         print "Opened Module",module
  33         m.Merge(feature, rootdir)
  34         print "Errors:"
  35         for e in m.Errors:
  36             print e.Type, e.ModuleTable, e.DatabaseTable
  37             print "   Modkeys:",
  38             for s in e.ModuleKeys: print s,
  39             print
  40             print "   DBKeys:",
  41             for s in e.DatabaseKeys: print s,
  42             print
  43         cabname = tempfile.mktemp(suffix=".cab")
  44         m.ExtractCAB(cabname)
  45         cab_and_filecount.append((cabname, len(m.ModuleFiles)))
  46         m.CloseModule()
  47     m.CloseDatabase(True)
  48     m.CloseLog()
  49
  50     # Step 2: Add CAB files
  51     i = msilib.MakeInstaller()
  52     db = i.OpenDatabase(msi, win32com.client.constants.msiOpenDatabaseModeTransact)
  53
  54     v = db.OpenView("SELECT LastSequence FROM Media")
  55     v.Execute(None)
  56     maxmedia = -1
  57     while 1:
  58         r = v.Fetch()
  59         if not r: break
  60         seq = r.IntegerData(1)
  61         if seq > maxmedia:
  62             maxmedia = seq
  63     print "Start of Media", maxmedia
  64
  65     for cabname, count in cab_and_filecount:
  66         stream = "merged%d" % maxmedia
  67         msilib.add_data(db, "Media",
  68                 [(maxmedia+1, maxmedia+count, None, "#"+stream, None, None)])
  69         msilib.add_stream(db, stream,  cabname)
  70         os.unlink(cabname)
  71         maxmedia += count
  72     # The merge module sets ALLUSERS to 1 in the property table.
  73     # This is undesired; delete that
  74     v = db.OpenView("DELETE FROM Property WHERE Property='ALLUSERS'")
  75     v.Execute(None)
  76     v.Close()
  77     db.Commit()
  78
  79 merge(msi, "SharedCRT", "TARGETDIR", modules)
  80
  81 # certname (from config.py) should be (a substring of)
  82 # the certificate subject, e.g. "Python Software Foundation"
  83 if certname:
  84     os.system('signtool sign /n "%s" /t http://timestamp.verisign.com/scripts/timestamp.dll %s' % (certname, msi))