blob: 12cb9d0f93d402b0e03ff3776bc493fd10543a07 [file] [log] [blame]
Martin v. Löwis8ffe9ab2004-08-22 13:34:34 +00001# Microsoft Installer Library
2# (C) 2003 Martin v. Loewis
3
4import win32com.client.gencache
5import win32com.client
6import pythoncom, pywintypes
7from win32com.client import constants
Martin v. Löwis49561ad2010-08-28 13:39:09 +00008import re, string, os, sets, glob, subprocess, sys, _winreg, struct, _msi
Martin v. Löwis3b6cc0952008-07-18 18:40:42 +00009
10try:
11 basestring
12except NameError:
13 basestring = (str, unicode)
Martin v. Löwis8ffe9ab2004-08-22 13:34:34 +000014
Martin v. Löwis8ffe9ab2004-08-22 13:34:34 +000015# Partially taken from Wine
16datasizemask= 0x00ff
17type_valid= 0x0100
18type_localizable= 0x0200
19
20typemask= 0x0c00
21type_long= 0x0000
22type_short= 0x0400
23type_string= 0x0c00
24type_binary= 0x0800
25
26type_nullable= 0x1000
27type_key= 0x2000
28# XXX temporary, localizable?
29knownbits = datasizemask | type_valid | type_localizable | \
30 typemask | type_nullable | type_key
31
32# Summary Info Property IDs
33PID_CODEPAGE=1
34PID_TITLE=2
35PID_SUBJECT=3
36PID_AUTHOR=4
37PID_KEYWORDS=5
38PID_COMMENTS=6
39PID_TEMPLATE=7
40PID_LASTAUTHOR=8
41PID_REVNUMBER=9
42PID_LASTPRINTED=11
43PID_CREATE_DTM=12
44PID_LASTSAVE_DTM=13
45PID_PAGECOUNT=14
46PID_WORDCOUNT=15
47PID_CHARCOUNT=16
48PID_APPNAME=18
49PID_SECURITY=19
50
51def reset():
52 global _directories
53 _directories = sets.Set()
54
55def EnsureMSI():
56 win32com.client.gencache.EnsureModule('{000C1092-0000-0000-C000-000000000046}', 1033, 1, 0)
57
58def EnsureMSM():
59 try:
60 win32com.client.gencache.EnsureModule('{0ADDA82F-2C26-11D2-AD65-00A0C9AF11A6}', 0, 1, 0)
61 except pywintypes.com_error:
62 win32com.client.gencache.EnsureModule('{0ADDA82F-2C26-11D2-AD65-00A0C9AF11A6}', 0, 2, 0)
63
64_Installer=None
65def MakeInstaller():
66 global _Installer
67 if _Installer is None:
68 EnsureMSI()
69 _Installer = win32com.client.Dispatch('WindowsInstaller.Installer',
70 resultCLSID='{000C1090-0000-0000-C000-000000000046}')
71 return _Installer
72
73_Merge=None
74def MakeMerge2():
75 global _Merge
76 if _Merge is None:
77 EnsureMSM()
78 _Merge = win32com.client.Dispatch("Msm.Merge2.1")
79 return _Merge
80
81class Table:
82 def __init__(self, name):
83 self.name = name
84 self.fields = []
85
86 def add_field(self, index, name, type):
87 self.fields.append((index,name,type))
88
89 def sql(self):
90 fields = []
91 keys = []
92 self.fields.sort()
93 fields = [None]*len(self.fields)
94 for index, name, type in self.fields:
95 index -= 1
96 unk = type & ~knownbits
97 if unk:
Martin v. Löwis3b6cc0952008-07-18 18:40:42 +000098 print "%s.%s unknown bits %x" % (self.name, name, unk)
Martin v. Löwis8ffe9ab2004-08-22 13:34:34 +000099 size = type & datasizemask
100 dtype = type & typemask
101 if dtype == type_string:
102 if size:
103 tname="CHAR(%d)" % size
104 else:
105 tname="CHAR"
106 elif dtype == type_short:
107 assert size==2
108 tname = "SHORT"
109 elif dtype == type_long:
110 assert size==4
111 tname="LONG"
112 elif dtype == type_binary:
113 assert size==0
114 tname="OBJECT"
115 else:
116 tname="unknown"
Martin v. Löwis3b6cc0952008-07-18 18:40:42 +0000117 print "%s.%sunknown integer type %d" % (self.name, name, size)
Martin v. Löwis8ffe9ab2004-08-22 13:34:34 +0000118 if type & type_nullable:
119 flags = ""
120 else:
121 flags = " NOT NULL"
122 if type & type_localizable:
123 flags += " LOCALIZABLE"
124 fields[index] = "`%s` %s%s" % (name, tname, flags)
125 if type & type_key:
126 keys.append("`%s`" % name)
127 fields = ", ".join(fields)
128 keys = ", ".join(keys)
129 return "CREATE TABLE %s (%s PRIMARY KEY %s)" % (self.name, fields, keys)
130
131 def create(self, db):
132 v = db.OpenView(self.sql())
133 v.Execute(None)
134 v.Close()
135
136class Binary:
137 def __init__(self, fname):
138 self.name = fname
139 def __repr__(self):
140 return 'msilib.Binary(os.path.join(dirname,"%s"))' % self.name
141
142def gen_schema(destpath, schemapath):
143 d = MakeInstaller()
144 schema = d.OpenDatabase(schemapath,
145 win32com.client.constants.msiOpenDatabaseModeReadOnly)
146
147 # XXX ORBER BY
148 v=schema.OpenView("SELECT * FROM _Columns")
149 curtable=None
150 tables = []
151 v.Execute(None)
152 f = open(destpath, "wt")
153 f.write("from msilib import Table\n")
154 while 1:
155 r=v.Fetch()
156 if not r:break
157 name=r.StringData(1)
158 if curtable != name:
159 f.write("\n%s = Table('%s')\n" % (name,name))
160 curtable = name
161 tables.append(name)
162 f.write("%s.add_field(%d,'%s',%d)\n" %
163 (name, r.IntegerData(2), r.StringData(3), r.IntegerData(4)))
164 v.Close()
165
166 f.write("\ntables=[%s]\n\n" % (", ".join(tables)))
167
168 # Fill the _Validation table
169 f.write("_Validation_records = [\n")
170 v = schema.OpenView("SELECT * FROM _Validation")
171 v.Execute(None)
172 while 1:
173 r = v.Fetch()
174 if not r:break
175 # Table, Column, Nullable
Martin v. Löwis3b6cc0952008-07-18 18:40:42 +0000176 f.write("(%s,%s,%s," %
177 (`r.StringData(1)`, `r.StringData(2)`, `r.StringData(3)`))
Martin v. Löwis8ffe9ab2004-08-22 13:34:34 +0000178 def put_int(i):
179 if r.IsNull(i):f.write("None, ")
180 else:f.write("%d," % r.IntegerData(i))
181 def put_str(i):
182 if r.IsNull(i):f.write("None, ")
Martin v. Löwis3b6cc0952008-07-18 18:40:42 +0000183 else:f.write("%s," % `r.StringData(i)`)
Martin v. Löwis8ffe9ab2004-08-22 13:34:34 +0000184 put_int(4) # MinValue
185 put_int(5) # MaxValue
186 put_str(6) # KeyTable
187 put_int(7) # KeyColumn
188 put_str(8) # Category
189 put_str(9) # Set
190 put_str(10)# Description
191 f.write("),\n")
192 f.write("]\n\n")
193
Tim Peters94607dd2004-08-22 19:42:56 +0000194 f.close()
Martin v. Löwis8ffe9ab2004-08-22 13:34:34 +0000195
196def gen_sequence(destpath, msipath):
197 dir = os.path.dirname(destpath)
198 d = MakeInstaller()
199 seqmsi = d.OpenDatabase(msipath,
200 win32com.client.constants.msiOpenDatabaseModeReadOnly)
201
202 v = seqmsi.OpenView("SELECT * FROM _Tables");
203 v.Execute(None)
204 f = open(destpath, "w")
Martin v. Löwis3b6cc0952008-07-18 18:40:42 +0000205 print >>f, "import msilib,os;dirname=os.path.dirname(__file__)"
Martin v. Löwis8ffe9ab2004-08-22 13:34:34 +0000206 tables = []
207 while 1:
208 r = v.Fetch()
209 if not r:break
210 table = r.StringData(1)
211 tables.append(table)
212 f.write("%s = [\n" % table)
213 v1 = seqmsi.OpenView("SELECT * FROM `%s`" % table)
214 v1.Execute(None)
215 info = v1.ColumnInfo(constants.msiColumnInfoTypes)
216 while 1:
217 r = v1.Fetch()
218 if not r:break
219 rec = []
220 for i in range(1,r.FieldCount+1):
221 if r.IsNull(i):
222 rec.append(None)
223 elif info.StringData(i)[0] in "iI":
224 rec.append(r.IntegerData(i))
225 elif info.StringData(i)[0] in "slSL":
226 rec.append(r.StringData(i))
227 elif info.StringData(i)[0]=="v":
228 size = r.DataSize(i)
229 bytes = r.ReadStream(i, size, constants.msiReadStreamBytes)
230 bytes = bytes.encode("latin-1") # binary data represented "as-is"
231 if table == "Binary":
232 fname = rec[0]+".bin"
233 open(os.path.join(dir,fname),"wb").write(bytes)
234 rec.append(Binary(fname))
235 else:
236 rec.append(bytes)
237 else:
Martin v. Löwis3b6cc0952008-07-18 18:40:42 +0000238 raise "Unsupported column type", info.StringData(i)
Martin v. Löwis8ffe9ab2004-08-22 13:34:34 +0000239 f.write(repr(tuple(rec))+",\n")
240 v1.Close()
241 f.write("]\n\n")
242 v.Close()
Martin v. Löwis3b6cc0952008-07-18 18:40:42 +0000243 f.write("tables=%s\n" % repr(map(str,tables)))
Martin v. Löwis8ffe9ab2004-08-22 13:34:34 +0000244 f.close()
245
246class _Unspecified:pass
247def change_sequence(seq, action, seqno=_Unspecified, cond = _Unspecified):
248 "Change the sequence number of an action in a sequence list"
249 for i in range(len(seq)):
250 if seq[i][0] == action:
251 if cond is _Unspecified:
252 cond = seq[i][1]
253 if seqno is _Unspecified:
254 seqno = seq[i][2]
255 seq[i] = (action, cond, seqno)
256 return
Martin v. Löwis3b6cc0952008-07-18 18:40:42 +0000257 raise ValueError, "Action not found in sequence"
Martin v. Löwis8ffe9ab2004-08-22 13:34:34 +0000258
259def add_data(db, table, values):
260 d = MakeInstaller()
261 v = db.OpenView("SELECT * FROM `%s`" % table)
262 count = v.ColumnInfo(0).FieldCount
263 r = d.CreateRecord(count)
264 for value in values:
265 assert len(value) == count, value
266 for i in range(count):
267 field = value[i]
Martin v. Löwis3b6cc0952008-07-18 18:40:42 +0000268 if isinstance(field, (int, long)):
Martin v. Löwis8ffe9ab2004-08-22 13:34:34 +0000269 r.SetIntegerData(i+1,field)
Martin v. Löwis3b6cc0952008-07-18 18:40:42 +0000270 elif isinstance(field, basestring):
Martin v. Löwis8ffe9ab2004-08-22 13:34:34 +0000271 r.SetStringData(i+1,field)
272 elif field is None:
273 pass
274 elif isinstance(field, Binary):
275 r.SetStream(i+1, field.name)
276 else:
Martin v. Löwis3b6cc0952008-07-18 18:40:42 +0000277 raise TypeError, "Unsupported type %s" % field.__class__.__name__
Martin v. Löwis8ffe9ab2004-08-22 13:34:34 +0000278 v.Modify(win32com.client.constants.msiViewModifyInsert, r)
279 r.ClearData()
280 v.Close()
281
282def add_stream(db, name, path):
283 d = MakeInstaller()
284 v = db.OpenView("INSERT INTO _Streams (Name, Data) VALUES ('%s', ?)" % name)
285 r = d.CreateRecord(1)
286 r.SetStream(1, path)
287 v.Execute(r)
288 v.Close()
289
290def init_database(name, schema,
291 ProductName, ProductCode, ProductVersion,
Amaury Forgeot d'Arc35c86582008-06-17 21:11:29 +0000292 Manufacturer,
293 request_uac = False):
Martin v. Löwis8ffe9ab2004-08-22 13:34:34 +0000294 try:
295 os.unlink(name)
296 except OSError:
297 pass
298 ProductCode = ProductCode.upper()
299 d = MakeInstaller()
300 # Create the database
301 db = d.OpenDatabase(name,
302 win32com.client.constants.msiOpenDatabaseModeCreate)
303 # Create the tables
304 for t in schema.tables:
305 t.create(db)
306 # Fill the validation table
307 add_data(db, "_Validation", schema._Validation_records)
308 # Initialize the summary information, allowing atmost 20 properties
309 si = db.GetSummaryInformation(20)
310 si.SetProperty(PID_TITLE, "Installation Database")
311 si.SetProperty(PID_SUBJECT, ProductName)
312 si.SetProperty(PID_AUTHOR, Manufacturer)
Martin v. Löwis856bf9a2006-02-14 20:42:55 +0000313 si.SetProperty(PID_TEMPLATE, msi_type)
Martin v. Löwis1e3a2642004-09-10 11:55:32 +0000314 si.SetProperty(PID_REVNUMBER, gen_uuid())
Amaury Forgeot d'Arc35c86582008-06-17 21:11:29 +0000315 if request_uac:
316 wc = 2 # long file names, compressed, original media
317 else:
318 wc = 2 | 8 # +never invoke UAC
319 si.SetProperty(PID_WORDCOUNT, wc)
Martin v. Löwis8ffe9ab2004-08-22 13:34:34 +0000320 si.SetProperty(PID_PAGECOUNT, 200)
321 si.SetProperty(PID_APPNAME, "Python MSI Library")
322 # XXX more properties
323 si.Persist()
324 add_data(db, "Property", [
325 ("ProductName", ProductName),
326 ("ProductCode", ProductCode),
327 ("ProductVersion", ProductVersion),
328 ("Manufacturer", Manufacturer),
329 ("ProductLanguage", "1033")])
330 db.Commit()
331 return db
332
333def add_tables(db, module):
334 for table in module.tables:
335 add_data(db, table, getattr(module, table))
336
337def make_id(str):
338 #str = str.replace(".", "_") # colons are allowed
339 str = str.replace(" ", "_")
340 str = str.replace("-", "_")
Martin v. Löwisf38e0d02008-06-13 14:11:59 +0000341 str = str.replace("+", "_")
Martin v. Löwis8ffe9ab2004-08-22 13:34:34 +0000342 if str[0] in string.digits:
343 str = "_"+str
344 assert re.match("^[A-Za-z_][A-Za-z0-9_.]*$", str), "FILE"+str
345 return str
346
347def gen_uuid():
348 return str(pythoncom.CreateGuid())
349
350class CAB:
351 def __init__(self, name):
352 self.name = name
Martin v. Löwis49561ad2010-08-28 13:39:09 +0000353 self.files = []
Martin v. Löwis8ffe9ab2004-08-22 13:34:34 +0000354 self.filenames = sets.Set()
355 self.index = 0
356
357 def gen_id(self, dir, file):
358 logical = _logical = make_id(file)
359 pos = 1
360 while logical in self.filenames:
361 logical = "%s.%d" % (_logical, pos)
362 pos += 1
363 self.filenames.add(logical)
364 return logical
365
366 def append(self, full, file, logical = None):
367 if os.path.isdir(full):
368 return
369 if not logical:
370 logical = self.gen_id(dir, file)
371 self.index += 1
Martin v. Löwis49561ad2010-08-28 13:39:09 +0000372 self.files.append((full, logical))
Martin v. Löwis8ffe9ab2004-08-22 13:34:34 +0000373 return self.index, logical
374
375 def commit(self, db):
Martin v. Löwis8ffe9ab2004-08-22 13:34:34 +0000376 try:
377 os.unlink(self.name+".cab")
378 except OSError:
379 pass
Martin v. Löwis49561ad2010-08-28 13:39:09 +0000380 _msi.FCICreate(self.name+".cab", self.files)
Martin v. Löwis8ffe9ab2004-08-22 13:34:34 +0000381 add_data(db, "Media",
382 [(1, self.index, None, "#"+self.name, None, None)])
383 add_stream(db, self.name, self.name+".cab")
384 os.unlink(self.name+".txt")
385 os.unlink(self.name+".cab")
386 db.Commit()
387
388_directories = sets.Set()
389class Directory:
390 def __init__(self, db, cab, basedir, physical, _logical, default, componentflags=None):
391 """Create a new directory in the Directory table. There is a current component
392 at each point in time for the directory, which is either explicitly created
393 through start_component, or implicitly when files are added for the first
394 time. Files are added into the current component, and into the cab file.
395 To create a directory, a base directory object needs to be specified (can be
396 None), the path to the physical directory, and a logical directory name.
397 Default specifies the DefaultDir slot in the directory table. componentflags
398 specifies the default flags that new components get."""
399 index = 1
400 _logical = make_id(_logical)
401 logical = _logical
402 while logical in _directories:
403 logical = "%s%d" % (_logical, index)
404 index += 1
405 _directories.add(logical)
406 self.db = db
407 self.cab = cab
408 self.basedir = basedir
409 self.physical = physical
410 self.logical = logical
411 self.component = None
412 self.short_names = sets.Set()
413 self.ids = sets.Set()
414 self.keyfiles = {}
415 self.componentflags = componentflags
416 if basedir:
417 self.absolute = os.path.join(basedir.absolute, physical)
418 blogical = basedir.logical
419 else:
420 self.absolute = physical
421 blogical = None
Martin v. Löwis797721b2010-08-03 18:35:55 +0000422 # initially assume that all files in this directory are unpackaged
423 # as files from self.absolute get added, this set is reduced
424 self.unpackaged_files = set()
425 for f in os.listdir(self.absolute):
426 if os.path.isfile(os.path.join(self.absolute, f)):
427 self.unpackaged_files.add(f)
Martin v. Löwis8ffe9ab2004-08-22 13:34:34 +0000428 add_data(db, "Directory", [(logical, blogical, default)])
429
Martin v. Löwis141f41a2005-03-15 00:39:40 +0000430 def start_component(self, component = None, feature = None, flags = None, keyfile = None, uuid=None):
Martin v. Löwis8ffe9ab2004-08-22 13:34:34 +0000431 """Add an entry to the Component table, and make this component the current for this
432 directory. If no component name is given, the directory name is used. If no feature
433 is given, the current feature is used. If no flags are given, the directory's default
434 flags are used. If no keyfile is given, the KeyPath is left null in the Component
435 table."""
436 if flags is None:
437 flags = self.componentflags
Martin v. Löwis141f41a2005-03-15 00:39:40 +0000438 if uuid is None:
439 uuid = gen_uuid()
440 else:
441 uuid = uuid.upper()
Martin v. Löwis8ffe9ab2004-08-22 13:34:34 +0000442 if component is None:
443 component = self.logical
444 self.component = component
445 if Win64:
446 flags |= 256
447 if keyfile:
448 keyid = self.cab.gen_id(self.absolute, keyfile)
449 self.keyfiles[keyfile] = keyid
450 else:
451 keyid = None
452 add_data(self.db, "Component",
453 [(component, uuid, self.logical, flags, None, keyid)])
454 if feature is None:
455 feature = current_feature
456 add_data(self.db, "FeatureComponents",
457 [(feature.id, component)])
458
459 def make_short(self, file):
Martin v. Löwisf38e0d02008-06-13 14:11:59 +0000460 file = re.sub(r'[\?|><:/*"+,;=\[\]]', '_', file) # restrictions on short names
Martin v. Löwis8ffe9ab2004-08-22 13:34:34 +0000461 parts = file.split(".")
462 if len(parts)>1:
463 suffix = parts[-1].upper()
464 else:
465 suffix = None
466 prefix = parts[0].upper()
467 if len(prefix) <= 8 and (not suffix or len(suffix)<=3):
468 if suffix:
469 file = prefix+"."+suffix
470 else:
471 file = prefix
472 assert file not in self.short_names
473 else:
474 prefix = prefix[:6]
475 if suffix:
476 suffix = suffix[:3]
477 pos = 1
478 while 1:
479 if suffix:
480 file = "%s~%d.%s" % (prefix, pos, suffix)
481 else:
482 file = "%s~%d" % (prefix, pos)
483 if file not in self.short_names: break
484 pos += 1
485 assert pos < 10000
486 if pos in (10, 100, 1000):
487 prefix = prefix[:-1]
488 self.short_names.add(file)
Martin v. Löwis8ffe9ab2004-08-22 13:34:34 +0000489 return file
490
491 def add_file(self, file, src=None, version=None, language=None):
492 """Add a file to the current component of the directory, starting a new one
493 one if there is no current component. By default, the file name in the source
494 and the file table will be identical. If the src file is specified, it is
495 interpreted relative to the current directory. Optionally, a version and a
496 language can be specified for the entry in the File table."""
497 if not self.component:
498 self.start_component(self.logical, current_feature)
499 if not src:
500 # Allow relative paths for file if src is not specified
501 src = file
502 file = os.path.basename(file)
503 absolute = os.path.join(self.absolute, src)
Martin v. Löwis797721b2010-08-03 18:35:55 +0000504 if absolute.startswith(self.absolute):
505 # mark file as packaged
506 relative = absolute[len(self.absolute)+1:]
507 if relative in self.unpackaged_files:
508 self.unpackaged_files.remove(relative)
Martin v. Löwis8ffe9ab2004-08-22 13:34:34 +0000509 assert not re.search(r'[\?|><:/*]"', file) # restrictions on long names
Martin v. Löwis3b6cc0952008-07-18 18:40:42 +0000510 if self.keyfiles.has_key(file):
Martin v. Löwis8ffe9ab2004-08-22 13:34:34 +0000511 logical = self.keyfiles[file]
512 else:
513 logical = None
514 sequence, logical = self.cab.append(absolute, file, logical)
515 assert logical not in self.ids
516 self.ids.add(logical)
517 short = self.make_short(file)
518 full = "%s|%s" % (short, file)
519 filesize = os.stat(absolute).st_size
520 # constants.msidbFileAttributesVital
521 # Compressed omitted, since it is the database default
522 # could add r/o, system, hidden
Tim Peters94607dd2004-08-22 19:42:56 +0000523 attributes = 512
Martin v. Löwis8ffe9ab2004-08-22 13:34:34 +0000524 add_data(self.db, "File",
525 [(logical, self.component, full, filesize, version,
526 language, attributes, sequence)])
527 if not version:
528 # Add hash if the file is not versioned
529 filehash = MakeInstaller().FileHash(absolute, 0)
530 add_data(self.db, "MsiFileHash",
531 [(logical, 0, filehash.IntegerData(1),
532 filehash.IntegerData(2), filehash.IntegerData(3),
533 filehash.IntegerData(4))])
534 # Automatically remove .pyc/.pyo files on uninstall (2)
535 # XXX: adding so many RemoveFile entries makes installer unbelievably
536 # slow. So instead, we have to use wildcard remove entries
537 # if file.endswith(".py"):
538 # add_data(self.db, "RemoveFile",
539 # [(logical+"c", self.component, "%sC|%sc" % (short, file),
540 # self.logical, 2),
541 # (logical+"o", self.component, "%sO|%so" % (short, file),
542 # self.logical, 2)])
543
544 def glob(self, pattern, exclude = None):
545 """Add a list of files to the current component as specified in the
546 glob pattern. Individual files can be excluded in the exclude list."""
547 files = glob.glob1(self.absolute, pattern)
548 for f in files:
549 if exclude and f in exclude: continue
550 self.add_file(f)
551 return files
552
553 def remove_pyc(self):
Martin v. Löwis2e3d5392010-08-06 10:43:31 +0000554 "Remove .pyc/.pyo files from __pycache__ on uninstall"
555 directory = self.logical + "_pycache"
556 add_data(self.db, "Directory", [(directory, self.logical, "__PYCA~1|__pycache__")])
557 flags = 256 if Win64 else 0
558 add_data(self.db, "Component",
559 [(directory, gen_uuid(), directory, flags, None, None)])
560 add_data(self.db, "FeatureComponents", [(current_feature.id, directory)])
561 add_data(self.db, "CreateFolder", [(directory, directory)])
Martin v. Löwis8ffe9ab2004-08-22 13:34:34 +0000562 add_data(self.db, "RemoveFile",
Martin v. Löwis2e3d5392010-08-06 10:43:31 +0000563 [(self.component, self.component, "*.*", directory, 2),
564 ])
Martin v. Löwis8ffe9ab2004-08-22 13:34:34 +0000565
Martin v. Löwis90cc5ab2008-06-02 10:04:16 +0000566 def removefile(self, key, pattern):
567 "Add a RemoveFile entry"
568 add_data(self.db, "RemoveFile", [(self.component+key, self.component, pattern, self.logical, 2)])
569
570
Martin v. Löwis8ffe9ab2004-08-22 13:34:34 +0000571class Feature:
572 def __init__(self, db, id, title, desc, display, level = 1,
573 parent=None, directory = None, attributes=0):
574 self.id = id
575 if parent:
576 parent = parent.id
577 add_data(db, "Feature",
578 [(id, parent, title, desc, display,
579 level, directory, attributes)])
580 def set_current(self):
581 global current_feature
582 current_feature = self
583
584class Control:
585 def __init__(self, dlg, name):
586 self.dlg = dlg
587 self.name = name
588
589 def event(self, ev, arg, cond = "1", order = None):
590 add_data(self.dlg.db, "ControlEvent",
591 [(self.dlg.name, self.name, ev, arg, cond, order)])
592
593 def mapping(self, ev, attr):
594 add_data(self.dlg.db, "EventMapping",
595 [(self.dlg.name, self.name, ev, attr)])
596
597 def condition(self, action, condition):
598 add_data(self.dlg.db, "ControlCondition",
599 [(self.dlg.name, self.name, action, condition)])
600
601class RadioButtonGroup(Control):
602 def __init__(self, dlg, name, property):
603 self.dlg = dlg
604 self.name = name
605 self.property = property
606 self.index = 1
607
608 def add(self, name, x, y, w, h, text, value = None):
609 if value is None:
610 value = name
611 add_data(self.dlg.db, "RadioButton",
612 [(self.property, self.index, value,
613 x, y, w, h, text, None)])
614 self.index += 1
615
616class Dialog:
617 def __init__(self, db, name, x, y, w, h, attr, title, first, default, cancel):
618 self.db = db
619 self.name = name
620 self.x, self.y, self.w, self.h = x,y,w,h
621 add_data(db, "Dialog", [(name, x,y,w,h,attr,title,first,default,cancel)])
622
623 def control(self, name, type, x, y, w, h, attr, prop, text, next, help):
624 add_data(self.db, "Control",
625 [(self.name, name, type, x, y, w, h, attr, prop, text, next, help)])
626 return Control(self, name)
627
628 def text(self, name, x, y, w, h, attr, text):
629 return self.control(name, "Text", x, y, w, h, attr, None,
630 text, None, None)
631
632 def bitmap(self, name, x, y, w, h, text):
633 return self.control(name, "Bitmap", x, y, w, h, 1, None, text, None, None)
Tim Peters94607dd2004-08-22 19:42:56 +0000634
Martin v. Löwis8ffe9ab2004-08-22 13:34:34 +0000635 def line(self, name, x, y, w, h):
636 return self.control(name, "Line", x, y, w, h, 1, None, None, None, None)
637
638 def pushbutton(self, name, x, y, w, h, attr, text, next):
639 return self.control(name, "PushButton", x, y, w, h, attr, None, text, next, None)
640
641 def radiogroup(self, name, x, y, w, h, attr, prop, text, next):
642 add_data(self.db, "Control",
643 [(self.name, name, "RadioButtonGroup",
644 x, y, w, h, attr, prop, text, next, None)])
645 return RadioButtonGroup(self, name, prop)
646
647 def checkbox(self, name, x, y, w, h, attr, prop, text, next):
Tim Peters94607dd2004-08-22 19:42:56 +0000648 return self.control(name, "CheckBox", x, y, w, h, attr, prop, text, next, None)
Martin v. Löwis856bf9a2006-02-14 20:42:55 +0000649
650def pe_type(path):
Martin v. Löwisd149c212006-03-05 13:52:20 +0000651 header = open(path, "rb").read(1000)
652 # offset of PE header is at offset 0x3c
653 pe_offset = struct.unpack("<i", header[0x3c:0x40])[0]
Martin v. Löwis856bf9a2006-02-14 20:42:55 +0000654 assert header[pe_offset:pe_offset+4] == "PE\0\0"
655 machine = struct.unpack("<H", header[pe_offset+4:pe_offset+6])[0]
656 return machine
657
658def set_arch_from_file(path):
659 global msi_type, Win64, arch_ext
660 machine = pe_type(path)
661 if machine == 0x14c:
662 # i386
663 msi_type = "Intel"
664 Win64 = 0
665 arch_ext = ''
666 elif machine == 0x200:
667 # Itanium
668 msi_type = "Intel64"
669 Win64 = 1
670 arch_ext = '.ia64'
671 elif machine == 0x8664:
672 # AMD64
673 msi_type = "x64"
674 Win64 = 1
675 arch_ext = '.amd64'
676 else:
Martin v. Löwis3b6cc0952008-07-18 18:40:42 +0000677 raise ValueError, "Unsupported architecture"
Martin v. Löwis856bf9a2006-02-14 20:42:55 +0000678 msi_type += ";1033"