mirror of
https://github.com/python/cpython.git
synced 2024-12-23 00:34:40 +08:00
475 lines
17 KiB
Python
475 lines
17 KiB
Python
# Copyright (C) 2005 Martin v. Löwis
|
|
# Licensed to PSF under a Contributor Agreement.
|
|
from _msi import *
|
|
import os, string, re, sys
|
|
|
|
AMD64 = "AMD64" in sys.version
|
|
Itanium = "Itanium" in sys.version
|
|
Win64 = AMD64 or Itanium
|
|
|
|
# Partially taken from Wine
|
|
datasizemask= 0x00ff
|
|
type_valid= 0x0100
|
|
type_localizable= 0x0200
|
|
|
|
typemask= 0x0c00
|
|
type_long= 0x0000
|
|
type_short= 0x0400
|
|
type_string= 0x0c00
|
|
type_binary= 0x0800
|
|
|
|
type_nullable= 0x1000
|
|
type_key= 0x2000
|
|
# XXX temporary, localizable?
|
|
knownbits = datasizemask | type_valid | type_localizable | \
|
|
typemask | type_nullable | type_key
|
|
|
|
class Table:
|
|
def __init__(self, name):
|
|
self.name = name
|
|
self.fields = []
|
|
|
|
def add_field(self, index, name, type):
|
|
self.fields.append((index,name,type))
|
|
|
|
def sql(self):
|
|
fields = []
|
|
keys = []
|
|
self.fields.sort()
|
|
fields = [None]*len(self.fields)
|
|
for index, name, type in self.fields:
|
|
index -= 1
|
|
unk = type & ~knownbits
|
|
if unk:
|
|
print("%s.%s unknown bits %x" % (self.name, name, unk))
|
|
size = type & datasizemask
|
|
dtype = type & typemask
|
|
if dtype == type_string:
|
|
if size:
|
|
tname="CHAR(%d)" % size
|
|
else:
|
|
tname="CHAR"
|
|
elif dtype == type_short:
|
|
assert size==2
|
|
tname = "SHORT"
|
|
elif dtype == type_long:
|
|
assert size==4
|
|
tname="LONG"
|
|
elif dtype == type_binary:
|
|
assert size==0
|
|
tname="OBJECT"
|
|
else:
|
|
tname="unknown"
|
|
print("%s.%sunknown integer type %d" % (self.name, name, size))
|
|
if type & type_nullable:
|
|
flags = ""
|
|
else:
|
|
flags = " NOT NULL"
|
|
if type & type_localizable:
|
|
flags += " LOCALIZABLE"
|
|
fields[index] = "`%s` %s%s" % (name, tname, flags)
|
|
if type & type_key:
|
|
keys.append("`%s`" % name)
|
|
fields = ", ".join(fields)
|
|
keys = ", ".join(keys)
|
|
return "CREATE TABLE %s (%s PRIMARY KEY %s)" % (self.name, fields, keys)
|
|
|
|
def create(self, db):
|
|
v = db.OpenView(self.sql())
|
|
v.Execute(None)
|
|
v.Close()
|
|
|
|
class _Unspecified:pass
|
|
def change_sequence(seq, action, seqno=_Unspecified, cond = _Unspecified):
|
|
"Change the sequence number of an action in a sequence list"
|
|
for i in range(len(seq)):
|
|
if seq[i][0] == action:
|
|
if cond is _Unspecified:
|
|
cond = seq[i][1]
|
|
if seqno is _Unspecified:
|
|
seqno = seq[i][2]
|
|
seq[i] = (action, cond, seqno)
|
|
return
|
|
raise ValueError("Action not found in sequence")
|
|
|
|
def add_data(db, table, values):
|
|
v = db.OpenView("SELECT * FROM `%s`" % table)
|
|
count = v.GetColumnInfo(MSICOLINFO_NAMES).GetFieldCount()
|
|
r = CreateRecord(count)
|
|
for value in values:
|
|
assert len(value) == count, value
|
|
for i in range(count):
|
|
field = value[i]
|
|
if isinstance(field, int):
|
|
r.SetInteger(i+1,field)
|
|
elif isinstance(field, str):
|
|
r.SetString(i+1,field)
|
|
elif field is None:
|
|
pass
|
|
elif isinstance(field, Binary):
|
|
r.SetStream(i+1, field.name)
|
|
else:
|
|
raise TypeError("Unsupported type %s" % field.__class__.__name__)
|
|
try:
|
|
v.Modify(MSIMODIFY_INSERT, r)
|
|
except Exception as e:
|
|
raise MSIError("Could not insert "+repr(values)+" into "+table)
|
|
|
|
r.ClearData()
|
|
v.Close()
|
|
|
|
|
|
def add_stream(db, name, path):
|
|
v = db.OpenView("INSERT INTO _Streams (Name, Data) VALUES ('%s', ?)" % name)
|
|
r = CreateRecord(1)
|
|
r.SetStream(1, path)
|
|
v.Execute(r)
|
|
v.Close()
|
|
|
|
def init_database(name, schema,
|
|
ProductName, ProductCode, ProductVersion,
|
|
Manufacturer):
|
|
try:
|
|
os.unlink(name)
|
|
except OSError:
|
|
pass
|
|
ProductCode = ProductCode.upper()
|
|
# Create the database
|
|
db = OpenDatabase(name, MSIDBOPEN_CREATE)
|
|
# Create the tables
|
|
for t in schema.tables:
|
|
t.create(db)
|
|
# Fill the validation table
|
|
add_data(db, "_Validation", schema._Validation_records)
|
|
# Initialize the summary information, allowing atmost 20 properties
|
|
si = db.GetSummaryInformation(20)
|
|
si.SetProperty(PID_TITLE, "Installation Database")
|
|
si.SetProperty(PID_SUBJECT, ProductName)
|
|
si.SetProperty(PID_AUTHOR, Manufacturer)
|
|
if Itanium:
|
|
si.SetProperty(PID_TEMPLATE, "Intel64;1033")
|
|
elif AMD64:
|
|
si.SetProperty(PID_TEMPLATE, "x64;1033")
|
|
else:
|
|
si.SetProperty(PID_TEMPLATE, "Intel;1033")
|
|
si.SetProperty(PID_REVNUMBER, gen_uuid())
|
|
si.SetProperty(PID_WORDCOUNT, 2) # long file names, compressed, original media
|
|
si.SetProperty(PID_PAGECOUNT, 200)
|
|
si.SetProperty(PID_APPNAME, "Python MSI Library")
|
|
# XXX more properties
|
|
si.Persist()
|
|
add_data(db, "Property", [
|
|
("ProductName", ProductName),
|
|
("ProductCode", ProductCode),
|
|
("ProductVersion", ProductVersion),
|
|
("Manufacturer", Manufacturer),
|
|
("ProductLanguage", "1033")])
|
|
db.Commit()
|
|
return db
|
|
|
|
def add_tables(db, module):
|
|
for table in module.tables:
|
|
add_data(db, table, getattr(module, table))
|
|
|
|
def make_id(str):
|
|
identifier_chars = string.ascii_letters + string.digits + "._"
|
|
str = "".join([c if c in identifier_chars else "_" for c in str])
|
|
if str[0] in (string.digits + "."):
|
|
str = "_" + str
|
|
assert re.match("^[A-Za-z_][A-Za-z0-9_.]*$", str), "FILE"+str
|
|
return str
|
|
|
|
def gen_uuid():
|
|
return "{"+UuidCreate().upper()+"}"
|
|
|
|
class CAB:
|
|
def __init__(self, name):
|
|
self.name = name
|
|
self.files = []
|
|
self.filenames = set()
|
|
self.index = 0
|
|
|
|
def gen_id(self, file):
|
|
logical = _logical = make_id(file)
|
|
pos = 1
|
|
while logical in self.filenames:
|
|
logical = "%s.%d" % (_logical, pos)
|
|
pos += 1
|
|
self.filenames.add(logical)
|
|
return logical
|
|
|
|
def append(self, full, file, logical):
|
|
if os.path.isdir(full):
|
|
return
|
|
if not logical:
|
|
logical = self.gen_id(file)
|
|
self.index += 1
|
|
self.files.append((full, logical))
|
|
return self.index, logical
|
|
|
|
def commit(self, db):
|
|
from tempfile import mktemp
|
|
filename = mktemp()
|
|
FCICreate(filename, self.files)
|
|
add_data(db, "Media",
|
|
[(1, self.index, None, "#"+self.name, None, None)])
|
|
add_stream(db, self.name, filename)
|
|
os.unlink(filename)
|
|
db.Commit()
|
|
|
|
_directories = set()
|
|
class Directory:
|
|
def __init__(self, db, cab, basedir, physical, _logical, default, componentflags=None):
|
|
"""Create a new directory in the Directory table. There is a current component
|
|
at each point in time for the directory, which is either explicitly created
|
|
through start_component, or implicitly when files are added for the first
|
|
time. Files are added into the current component, and into the cab file.
|
|
To create a directory, a base directory object needs to be specified (can be
|
|
None), the path to the physical directory, and a logical directory name.
|
|
Default specifies the DefaultDir slot in the directory table. componentflags
|
|
specifies the default flags that new components get."""
|
|
index = 1
|
|
_logical = make_id(_logical)
|
|
logical = _logical
|
|
while logical in _directories:
|
|
logical = "%s%d" % (_logical, index)
|
|
index += 1
|
|
_directories.add(logical)
|
|
self.db = db
|
|
self.cab = cab
|
|
self.basedir = basedir
|
|
self.physical = physical
|
|
self.logical = logical
|
|
self.component = None
|
|
self.short_names = set()
|
|
self.ids = set()
|
|
self.keyfiles = {}
|
|
self.componentflags = componentflags
|
|
if basedir:
|
|
self.absolute = os.path.join(basedir.absolute, physical)
|
|
blogical = basedir.logical
|
|
else:
|
|
self.absolute = physical
|
|
blogical = None
|
|
add_data(db, "Directory", [(logical, blogical, default)])
|
|
|
|
def start_component(self, component = None, feature = None, flags = None, keyfile = None, uuid=None):
|
|
"""Add an entry to the Component table, and make this component the current for this
|
|
directory. If no component name is given, the directory name is used. If no feature
|
|
is given, the current feature is used. If no flags are given, the directory's default
|
|
flags are used. If no keyfile is given, the KeyPath is left null in the Component
|
|
table."""
|
|
if flags is None:
|
|
flags = self.componentflags
|
|
if uuid is None:
|
|
uuid = gen_uuid()
|
|
else:
|
|
uuid = uuid.upper()
|
|
if component is None:
|
|
component = self.logical
|
|
self.component = component
|
|
if Win64:
|
|
flags |= 256
|
|
if keyfile:
|
|
keyid = self.cab.gen_id(self.absolute, keyfile)
|
|
self.keyfiles[keyfile] = keyid
|
|
else:
|
|
keyid = None
|
|
add_data(self.db, "Component",
|
|
[(component, uuid, self.logical, flags, None, keyid)])
|
|
if feature is None:
|
|
feature = current_feature
|
|
add_data(self.db, "FeatureComponents",
|
|
[(feature.id, component)])
|
|
|
|
def make_short(self, file):
|
|
oldfile = file
|
|
file = file.replace('+', '_')
|
|
file = ''.join(c for c in file if not c in ' "/\[]:;=,')
|
|
parts = file.split(".")
|
|
if len(parts) > 1:
|
|
prefix = "".join(parts[:-1]).upper()
|
|
suffix = parts[-1].upper()
|
|
if not prefix:
|
|
prefix = suffix
|
|
suffix = None
|
|
else:
|
|
prefix = file.upper()
|
|
suffix = None
|
|
if len(parts) < 3 and len(prefix) <= 8 and file == oldfile and (
|
|
not suffix or len(suffix) <= 3):
|
|
if suffix:
|
|
file = prefix+"."+suffix
|
|
else:
|
|
file = prefix
|
|
else:
|
|
file = None
|
|
if file is None or file in self.short_names:
|
|
prefix = prefix[:6]
|
|
if suffix:
|
|
suffix = suffix[:3]
|
|
pos = 1
|
|
while 1:
|
|
if suffix:
|
|
file = "%s~%d.%s" % (prefix, pos, suffix)
|
|
else:
|
|
file = "%s~%d" % (prefix, pos)
|
|
if file not in self.short_names: break
|
|
pos += 1
|
|
assert pos < 10000
|
|
if pos in (10, 100, 1000):
|
|
prefix = prefix[:-1]
|
|
self.short_names.add(file)
|
|
assert not re.search(r'[\?|><:/*"+,;=\[\]]', file) # restrictions on short names
|
|
return file
|
|
|
|
def add_file(self, file, src=None, version=None, language=None):
|
|
"""Add a file to the current component of the directory, starting a new one
|
|
if there is no current component. By default, the file name in the source
|
|
and the file table will be identical. If the src file is specified, it is
|
|
interpreted relative to the current directory. Optionally, a version and a
|
|
language can be specified for the entry in the File table."""
|
|
if not self.component:
|
|
self.start_component(self.logical, current_feature, 0)
|
|
if not src:
|
|
# Allow relative paths for file if src is not specified
|
|
src = file
|
|
file = os.path.basename(file)
|
|
absolute = os.path.join(self.absolute, src)
|
|
assert not re.search(r'[\?|><:/*]"', file) # restrictions on long names
|
|
if file in self.keyfiles:
|
|
logical = self.keyfiles[file]
|
|
else:
|
|
logical = None
|
|
sequence, logical = self.cab.append(absolute, file, logical)
|
|
assert logical not in self.ids
|
|
self.ids.add(logical)
|
|
short = self.make_short(file)
|
|
full = "%s|%s" % (short, file)
|
|
filesize = os.stat(absolute).st_size
|
|
# constants.msidbFileAttributesVital
|
|
# Compressed omitted, since it is the database default
|
|
# could add r/o, system, hidden
|
|
attributes = 512
|
|
add_data(self.db, "File",
|
|
[(logical, self.component, full, filesize, version,
|
|
language, attributes, sequence)])
|
|
#if not version:
|
|
# # Add hash if the file is not versioned
|
|
# filehash = FileHash(absolute, 0)
|
|
# add_data(self.db, "MsiFileHash",
|
|
# [(logical, 0, filehash.IntegerData(1),
|
|
# filehash.IntegerData(2), filehash.IntegerData(3),
|
|
# filehash.IntegerData(4))])
|
|
# Automatically remove .pyc/.pyo files on uninstall (2)
|
|
# XXX: adding so many RemoveFile entries makes installer unbelievably
|
|
# slow. So instead, we have to use wildcard remove entries
|
|
if file.endswith(".py"):
|
|
add_data(self.db, "RemoveFile",
|
|
[(logical+"c", self.component, "%sC|%sc" % (short, file),
|
|
self.logical, 2),
|
|
(logical+"o", self.component, "%sO|%so" % (short, file),
|
|
self.logical, 2)])
|
|
return logical
|
|
|
|
def glob(self, pattern, exclude = None):
|
|
"""Add a list of files to the current component as specified in the
|
|
glob pattern. Individual files can be excluded in the exclude list."""
|
|
files = glob.glob1(self.absolute, pattern)
|
|
for f in files:
|
|
if exclude and f in exclude: continue
|
|
self.add_file(f)
|
|
return files
|
|
|
|
def remove_pyc(self):
|
|
"Remove .pyc/.pyo files on uninstall"
|
|
add_data(self.db, "RemoveFile",
|
|
[(self.component+"c", self.component, "*.pyc", self.logical, 2),
|
|
(self.component+"o", self.component, "*.pyo", self.logical, 2)])
|
|
|
|
class Binary:
|
|
def __init__(self, fname):
|
|
self.name = fname
|
|
def __repr__(self):
|
|
return 'msilib.Binary(os.path.join(dirname,"%s"))' % self.name
|
|
|
|
class Feature:
|
|
def __init__(self, db, id, title, desc, display, level = 1,
|
|
parent=None, directory = None, attributes=0):
|
|
self.id = id
|
|
if parent:
|
|
parent = parent.id
|
|
add_data(db, "Feature",
|
|
[(id, parent, title, desc, display,
|
|
level, directory, attributes)])
|
|
def set_current(self):
|
|
global current_feature
|
|
current_feature = self
|
|
|
|
class Control:
|
|
def __init__(self, dlg, name):
|
|
self.dlg = dlg
|
|
self.name = name
|
|
|
|
def event(self, event, argument, condition = "1", ordering = None):
|
|
add_data(self.dlg.db, "ControlEvent",
|
|
[(self.dlg.name, self.name, event, argument,
|
|
condition, ordering)])
|
|
|
|
def mapping(self, event, attribute):
|
|
add_data(self.dlg.db, "EventMapping",
|
|
[(self.dlg.name, self.name, event, attribute)])
|
|
|
|
def condition(self, action, condition):
|
|
add_data(self.dlg.db, "ControlCondition",
|
|
[(self.dlg.name, self.name, action, condition)])
|
|
|
|
class RadioButtonGroup(Control):
|
|
def __init__(self, dlg, name, property):
|
|
self.dlg = dlg
|
|
self.name = name
|
|
self.property = property
|
|
self.index = 1
|
|
|
|
def add(self, name, x, y, w, h, text, value = None):
|
|
if value is None:
|
|
value = name
|
|
add_data(self.dlg.db, "RadioButton",
|
|
[(self.property, self.index, value,
|
|
x, y, w, h, text, None)])
|
|
self.index += 1
|
|
|
|
class Dialog:
|
|
def __init__(self, db, name, x, y, w, h, attr, title, first, default, cancel):
|
|
self.db = db
|
|
self.name = name
|
|
self.x, self.y, self.w, self.h = x,y,w,h
|
|
add_data(db, "Dialog", [(name, x,y,w,h,attr,title,first,default,cancel)])
|
|
|
|
def control(self, name, type, x, y, w, h, attr, prop, text, next, help):
|
|
add_data(self.db, "Control",
|
|
[(self.name, name, type, x, y, w, h, attr, prop, text, next, help)])
|
|
return Control(self, name)
|
|
|
|
def text(self, name, x, y, w, h, attr, text):
|
|
return self.control(name, "Text", x, y, w, h, attr, None,
|
|
text, None, None)
|
|
|
|
def bitmap(self, name, x, y, w, h, text):
|
|
return self.control(name, "Bitmap", x, y, w, h, 1, None, text, None, None)
|
|
|
|
def line(self, name, x, y, w, h):
|
|
return self.control(name, "Line", x, y, w, h, 1, None, None, None, None)
|
|
|
|
def pushbutton(self, name, x, y, w, h, attr, text, next):
|
|
return self.control(name, "PushButton", x, y, w, h, attr, None, text, next, None)
|
|
|
|
def radiogroup(self, name, x, y, w, h, attr, prop, text, next):
|
|
add_data(self.db, "Control",
|
|
[(self.name, name, "RadioButtonGroup",
|
|
x, y, w, h, attr, prop, text, next, None)])
|
|
return RadioButtonGroup(self, name, prop)
|
|
|
|
def checkbox(self, name, x, y, w, h, attr, prop, text, next):
|
|
return self.control(name, "CheckBox", x, y, w, h, attr, prop, text, next, None)
|