utility for install in zipped distutils packages

[Craig D]

I got really tired of unpacking a distutils package, installing then
cleaning up. So I wrote distinstall.py - it reads a zip file, unpacks
it, installs the bits and cleans up. I've only used it on windows and
Linux and for pure Python packages.
Windows: distinstall.py Photronics.zip
Linux:   python distinstall.py Photronics.zip

Hopefully somebody else will find it useful.

#======================== distinstall.py =====================
#!/usr/bin/env python
'''Install a zipped up distutils package'''

import zipfile
import sys, os, os.path

def usage():
   print "distinstall.py pack.zip: Install a zipped distutil python
package"
   print "One arg: name of the zip package"
   print "THIS PROGRAM ONLY WORKS WITH ZIPPED PURE PYTHON PACKAGES!"
   sys.exit(1)

def nuke(top):
    # Delete everything reachable from the directory named in 'top'.
    # CAUTION:  This is dangerous!  For example, if top == '/', it
    # could delete all your disk files.
    for root, dirs, files in os.walk(top, topdown=False):
       for name in files:
	  os.remove(os.path.join(root, name))
       for name in dirs:
	  os.rmdir(os.path.join(root, name))
    os.rmdir(top)
			

def explodeZipFile(zippedPackageFileName):
   if not zipfile.is_zipfile(zippedPackageFileName):
      print '"%s" is not a zip file!' % zippedPackageFileName
      usage()
   zip = zipfile.ZipFile(zippedPackageFileName,'r')
   if not filter(lambda n:"setup.py" in n, zip.namelist()):
      print "setup.py isn't in the zip file! Not a distutil package?"
      usage()

   tmp = os.getenv('TMP') or os.getenv('TMPDIR')
   if (not tmp) and os.path.isdir('/tmp'): tmp = '/tmp'
   if not tmp:
      print 'Can\'t find the tmp directory!'
      sys.exit(1)

   path,fileName = os.path.split(zippedPackageFileName)
   tmpDir = os.path.join(tmp,os.path.splitext(fileName)[0])

   print "unzip %s to %s" % (zippedPackageFileName,tmpDir)
   for z in zip.namelist():
#      print z
      data = zip.read(z)
      try:
	 os.makedirs(os.path.join(tmp,os.path.split(z)[0]),0777)
      except: pass
      f = file(os.path.join(tmp,z),'w')	#?? what's up with wb?
      f.writelines(data.split('\r'))	# this really sucks
      f.close()

   zip.close()
   setup  = os.path.join(tmpDir,"setup.py")
   return tmpDir,setup


if len(sys.argv) < 2: usage()

cwd = os.getcwd()
zippedPackageFileName = sys.argv[1]
tmpDir,setup = explodeZipFile(zippedPackageFileName)

print "====================Install================="
os.chdir(tmpDir)
install = "%s setup.py install --force" % sys.executable
print install
os.system(install)

os.chdir(cwd)
print "\nClean up %s" % tmpDir
nuke(tmpDir)	# this will clean up the build directory
#======================== distinstall.py =====================