"""Modification of standard bdist_wininst to include numeric version in filename

"""
import sys, os, string, struct
from distutils.command.bdist_wininst import bdist_wininst

class BdistWinInstaller(bdist_wininst):
	"""Version of bdist_wininst with customization point for filename

	This class should operate identically to the built-in
	class, it exists solely to provide the customization point.
	"""
	def get_installer_filename(self, directory, fullname, target_version= None):
		"""Calculate the final installer filename"""
		if target_version:
			return os.path.join(
				directory,
				"%s.win32-py%s.exe" % (
					fullname,
					target_version
				)
			)
		else:
			return os.path.join(
				directory,
				"%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(
			self.dist_dir,
			fullname,
			self.target_version,
		)
		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, directory, fullname, target_version= None):
		"""Calculate the final installer filename"""
		try:
			import Numeric
			numeric_version = 'numpy%s'%( string.split(Numeric.__version__, '.')[0], )
		except ImportError:
			numeric_version = 'nonum'
		if target_version:
			return os.path.join(
				directory,
				"%s.py%s-%s.exe" % (
					fullname,
					target_version,
					numeric_version,
				)
			)
		else:
			return os.path.join(
				directory,
				"%s.%s.exe" %(
					fullname,
					numeric_version,
				)
			)