144 lines
4.5 KiB
Python
Executable File
144 lines
4.5 KiB
Python
Executable File
#! /usr/bin/env python
|
|
|
|
"""
|
|
This command will build the contents.xml file at the top of a Panda3D
|
|
download hierarchy. This file lists all of the packages hosted here,
|
|
along with their current versions.
|
|
|
|
This program runs on a local copy of the hosting directory hierarchy;
|
|
it must be a complete copy to generate a complete contents.xml file.
|
|
|
|
make_contents.py [opts]
|
|
|
|
Options:
|
|
|
|
-d stage_dir
|
|
|
|
Specify the staging directory. This is a temporary directory on
|
|
the local machine that contains a copy of the web server
|
|
contents. The default is the current directory.
|
|
|
|
"""
|
|
|
|
import sys
|
|
import getopt
|
|
import os
|
|
import md5
|
|
|
|
class ArgumentError(AttributeError):
|
|
pass
|
|
|
|
class FileSpec:
|
|
""" Represents a single file in the directory, and its associated
|
|
timestamp, size, and md5 hash. """
|
|
|
|
def __init__(self, filename, pathname):
|
|
self.filename = filename
|
|
self.pathname = pathname
|
|
|
|
s = os.stat(pathname)
|
|
self.size = s.st_size
|
|
self.timestamp = int(s.st_mtime)
|
|
|
|
m = md5.new()
|
|
m.update(open(pathname, 'rb').read())
|
|
self.hash = m.hexdigest()
|
|
|
|
def getParams(self):
|
|
return 'filename="%s" size="%s" timestamp="%s" hash="%s"' % (
|
|
self.filename, self.size, self.timestamp, self.hash)
|
|
|
|
class ContentsMaker:
|
|
def __init__(self):
|
|
self.installDir = None
|
|
|
|
def build(self):
|
|
if not self.installDir:
|
|
raise ArgumentError, "Stage directory not specified."
|
|
|
|
self.packages = []
|
|
self.scanDirectory()
|
|
|
|
if not self.packages:
|
|
raise ArgumentError, "No packages found."
|
|
|
|
# Now write the contents.xml file.
|
|
contentsFileBasename = 'contents.xml'
|
|
contentsFilePathname = os.path.join(self.installDir, contentsFileBasename)
|
|
|
|
f = open(contentsFilePathname, 'w')
|
|
print >> f, '<?xml version="1.0" ?>'
|
|
print >> f, ''
|
|
print >> f, '<contents>'
|
|
for packageName, packagePlatform, packageVersion, file in self.packages:
|
|
print >> f, ' <package name="%s" platform="%s" version="%s" %s />' % (
|
|
packageName, packagePlatform or '', packageVersion, file.getParams())
|
|
print >> f, '</contents>'
|
|
f.close()
|
|
|
|
def scanDirectory(self):
|
|
""" Walks through all the files in the stage directory and
|
|
looks for the package directory xml files. """
|
|
|
|
startDir = self.installDir
|
|
if startDir.endswith(os.sep):
|
|
startDir = startDir[:-1]
|
|
prefix = startDir + os.sep
|
|
for dirpath, dirnames, filenames in os.walk(startDir):
|
|
if dirpath == startDir:
|
|
localpath = ''
|
|
xml = ''
|
|
else:
|
|
assert dirpath.startswith(prefix)
|
|
localpath = dirpath[len(prefix):].replace(os.sep, '/') + '/'
|
|
xml = dirpath[len(prefix):].replace(os.sep, '_') + '.xml'
|
|
|
|
# A special case: the "plugin" and "coreapi" directories
|
|
# don't have xml files, just dll's.
|
|
if xml.startswith('plugin_') or xml.startswith('coreapi_'):
|
|
if filenames:
|
|
assert len(filenames) == 1
|
|
xml = filenames[0]
|
|
|
|
if xml not in filenames:
|
|
continue
|
|
|
|
if localpath.count('/') == 2:
|
|
packageName, packageVersion, junk = localpath.split('/')
|
|
packagePlatform = None
|
|
file = FileSpec(localpath + xml,
|
|
os.path.join(self.installDir, localpath + xml))
|
|
print file.filename
|
|
self.packages.append((packageName, packagePlatform, packageVersion, file))
|
|
|
|
if localpath.count('/') == 3:
|
|
packageName, packagePlatform, packageVersion, junk = localpath.split('/')
|
|
file = FileSpec(localpath + xml,
|
|
os.path.join(self.installDir, localpath + xml))
|
|
print file.filename
|
|
self.packages.append((packageName, packagePlatform, packageVersion, file))
|
|
|
|
|
|
def makeContents(args):
|
|
opts, args = getopt.getopt(args, 'd:h')
|
|
|
|
cm = ContentsMaker()
|
|
cm.installDir = '.'
|
|
for option, value in opts:
|
|
if option == '-d':
|
|
cm.installDir = value
|
|
|
|
elif option == '-h':
|
|
print __doc__
|
|
sys.exit(1)
|
|
|
|
cm.build()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
try:
|
|
makeContents(sys.argv[1:])
|
|
except ArgumentError, e:
|
|
print e.args[0]
|
|
sys.exit(1)
|