import os
import setuptools
import shutil
import subprocess
import sys
import tarfile
import zipfile
import tempfile
class TarArchive:
def __init__(self, filename):
self.filename = filename
self.tgz = tarfile.TarFile.gzopen(filename, 'r')
def names(self):
return self.tgz.getnames()
def lines(self, name):
return self.tgz.extractfile(name).readlines()
def extract(self, name, tempdir):
return self.tgz.extract(name, tempdir)
def extractall(self, tempdir):
os.system('cd %s && tar xzf %s' % (tempdir,
os.path.abspath(self.filename)))
def close(self):
return self.tgz.close()
class ZipArchive:
def __init__(self, filename):
self.filename = filename
self.zipf = zipfile.ZipFile(filename, 'r')
def names(self):
return self.zipf.namelist()
def lines(self, name):
return self.zipf.read(name).split('\n')
def extract(self, name, tempdir):
data = self.zipf.read(name)
fn = name.split(os.sep)[-1]
fn = os.path.join(tempdir, fn)
f = open(fn, 'wb')
f.write(data)
def extractall(self, tempdir):
os.system('cd %s && unzip %s' % (tempdir,
os.path.abspath(self.filename)))
def close(self):
return self.zipf.close()
def _extractNameVersion(filename, tempdir):
print 'Parsing:', filename
if filename.endswith('.gz') or filename.endswith('.tgz'):
archive = TarArchive(filename)
elif filename.endswith('.egg') or filename.endswith('.zip'):
archive = ZipArchive(filename)
try:
for name in archive.names():
if name.endswith('PKG-INFO'):
project, version = None, None
lines = archive.lines(name)
for line in lines:
key, value = line.split(':', 1)
if key == 'Name':
project = value.strip()
elif key == 'Version':
version = value.strip()
if project is not None and version is not None:
return project, version
# no PKG-INFO found, do it the hard way.
archive.extractall(tempdir)
dirs = os.listdir(tempdir)
dir = os.path.join(tempdir, dirs[0])
if not os.path.isdir(dir):
dir = tempdir
command = ('cd %s && %s setup.py --name --version'
% (dir, sys.executable))
popen = subprocess.Popen(command,
stdout=subprocess.PIPE,
shell=True,
)
output = popen.communicate()[0]
return output.splitlines()[:2]
finally:
archive.close()
def main(argv=None):
if argv is None:
argv = sys.argv[1:]
projects = {}
for arg in argv:
if arg.startswith('*'):
continue
try:
tempdir = tempfile.mkdtemp()
project, revision = _extractNameVersion(arg, tempdir)
projects.setdefault(project, []).append((revision, arg))
finally:
shutil.rmtree(tempdir)
items = projects.items()
items.sort()
topname = 'index'
os.makedirs(topname)
top = open('%s/index.html' % topname, 'w')
top.writelines(['\n',
'\n',
'Package Index
\n',
'\n'])
for key, value in items:
print 'Project: %s' % key
dirname = '%s/%s' % (topname, key)
os.makedirs(dirname)
top.write('- %s\n' % (key, key))
sub = open('%s/%s/index.html' % (topname, key), 'w')
sub.writelines(['\n',
'\n',
'
%s Distributions
\n' % key,
'\n'])
for revision, archive in value:
print ' -> %s, %s' % (revision, archive)
sub.write('- %s\n' % (archive, archive))
sub.writelines(['
\n',
'\n',
'\n'])
top.writelines(['
\n',
'\n',
'\n'])
top.close()
if __name__ == '__main__':
main()