"""Modification of bdist_wininst to include numeric version in filename
"""
import sys, os, string, struct
from distutils.command.bdist_wininst import bdist_wininst
if hasattr( bdist_wininst, 'get_installer_filename' ):
# Python 2.4a2 and beyond...
BdistWinInstaller = bdist_wininst
else:
class BdistWinInstaller(bdist_wininst):
"""Version of bdist_wininst with customization point for filename
This class should operate identically to the built-in
class for Python 2.3 and below, it exists solely to
provide the customization point which exists in
Python 2.4a2 and beyond
"""
def get_installer_filename(self, fullname ):
"""Calculate the final installer filename"""
if self.target_version:
return os.path.join(
self.dist_dir,
"%s.win32-py%s.exe" % (
fullname,
self.target_version
)
)
else:
return os.path.join(
self.dist_dir,
"%s.win32.exe" % fullname
)
def create_exe (self, arcname, fullname, bitmap=None):
"""Do the actual creation of the executable file
The base command, unfortunately, does not break down
this command into sub-commands, so this method is
almost entirely a duplication of the base-class method,
with the only noticeable difference being the
call to get_installer_filename(...) instead of
calculating the filename inline.
"""
import struct
self.mkpath(self.dist_dir)
cfgdata = self.get_inidata()
installer_name = self.get_installer_filename(
fullname,
)
self.announce("creating %s" % installer_name)
if bitmap:
bitmapdata = open(bitmap, "rb").read()
bitmaplen = len(bitmapdata)
else:
bitmaplen = 0
file = open(installer_name, "wb")
file.write(self.get_exe_bytes())
if bitmap:
file.write(bitmapdata)
file.write(cfgdata)
header = struct.pack("<iii",
0x1234567A, # tag
len(cfgdata), # length
bitmaplen, # number of bytes in bitmap
)
file.write(header)
file.write(open(arcname, "rb").read())
class NumericWinInstaller(BdistWinInstaller):
"""Version of BdistWininst which includes Numpy version in name"""
def get_installer_filename(self, fullname ):
"""Calculate the final installer filename"""
try:
import Numeric
numeric_version = 'numpy%s'%( string.split(Numeric.__version__, '.')[0], )
except ImportError:
numeric_version = 'nonum'
if self.target_version:
return os.path.join(
self.dist_dir,
"%s.py%s-%s.exe" % (
fullname,
self.target_version,
numeric_version,
)
)
else:
return os.path.join(
self.dist_dir,
"%s.%s.exe" %(
fullname,
numeric_version,
)
)
|