root/packaging/centuryegg/build-rpms

#!/usr/bin/env python

import logging
import os
import re
import shutil
import glob
try:
    import subprocess
except ImportError:
    from fake_subprocess import subprocess
import sys
import ConfigParser

successful_run = True
script = sys.argv[0]
script = os.path.abspath(script)
src_dir = os.path.dirname(script)
os.chdir(src_dir)

logging.getLogger().setLevel(logging.INFO)
log_dir = os.path.join(src_dir, "logs")
packages_dir = os.path.join(src_dir, "packages")
rpm_dir = os.path.join(src_dir, "rpms")
tmp_dir = os.path.join(src_dir, "tmp")

for required_dir in [log_dir, packages_dir, rpm_dir, tmp_dir]:
    if not os.path.exists(required_dir):
        os.makedirs(required_dir)

os.chdir(tmp_dir)
if sys.version_info[0] == 2 and sys.version_info[1] < os.environ["MIN_PYTHON_MINOR_VERSION"]:
    easyinstall = "easy_install-"+os.environ["PYTHON_VERSION"]
    python = "python"+os.environ["PYTHON_VERSION"]
else:
    easyinstall = "easy_install"
    python = "python"

requirements = sys.argv[1:]
if not requirements:
    requirements = [line.strip() for line in open(os.path.join(src_dir, "requirements")) if line.strip() and not line.startswith("#")]

packages = []

for requirement in requirements[:]:
    package_name = re.compile("^(?P<name>[A-Za-z0-9_-]+) *(?P<versionmatch>[<=>]+.*)? *").match(requirement).group("name")
    existing_packages = glob.glob("../rpms/%s-*.rpm" % package_name)
    if existing_packages:
        logging.info("Not building %s due to existing packages %r - remove these if a rebuild is required" % (requirement, existing_packages))
        requirements.remove(requirement)
        continue
    package_name = package_name.lower()
    logging.info("Easy-install downloading %s" % (requirement,))
    package_build_dir = os.path.join(packages_dir, package_name)
    if os.path.exists(package_build_dir):
        shutil.rmtree(package_build_dir)
    packages.append(package_name)
if not requirements:
    sys.exit()
log_filename = os.path.join(log_dir, "fetch-easyinstall.log")
log_file = open(log_filename, "ab")
retcode = subprocess.call([easyinstall, "-eUNb", packages_dir] + requirements, stdout=log_file, stderr=log_file)
if retcode:
    logging.warning("Error fetching source packages for %s: tail of log file %s follows" % (" ".join(requirements), log_filename))
    print "".join(open(log_filename, "rb").readlines()[-10:])
    successful_run = False

# for d in [d_name for d_name in os.listdir(packages_dir) if os.path.isdir(os.path.join(packages_dir, d_name)]:
for package_name in packages:
    logging.info("building rpm for %s" % (package_name,))
    package_subdir = os.path.join(packages_dir, package_name)
    if not os.path.exists(package_subdir):
        logging.warning("could not find directory for building: %s" % (package_subdir,))
        continue
    os.chdir(package_subdir)
    if os.path.exists("build"):
        shutil.rmtree("build")
    build_config = ConfigParser.ConfigParser()
    build_config.read('setup.cfg')
    build_config.has_section('install') or build_config.add_section('install')
    build_config.set('install', 'compile', '1')
    build_config.set('install', 'optimize', '1')
    build_config.write(open('setup.cfg', 'w'))
    subprocess.call([python, "setup.py", "clean"])
    if os.path.exists("MANIFEST"):
        if "setup.cfg\n" not in open("MANIFEST", "r").read():
            open("MANIFEST", "a").write("\nsetup.cfg\n")
    # TODO: add a way of customizing particular packages
    if package_name == "cherrypy":
        open("MANIFEST.in", "w").write("graft cherrypy/\nsetup.cfg\nsetup.py\n")
    if package_name == "oursql":
        setup_code = open("setup.py", "rb").read()
        setup_code = setup_code.replace("Extension(Extension)", "Extension(Extension, object)")
        open("setup.py", "wb").write(setup_code)
        open("MANIFEST.in", "w").write("include README\ninclude setup.cfg\ninclude setup.py\nglobal-include *.c *.h *.pyx *.pxi\ninclude MANIFEST.in\n")
    log_filename = os.path.join(log_dir, "%s-build-rpm.log" % (package_name,))
    log_file = open(log_filename, "wb")
    ret_code = subprocess.call([python, "setup.py", "bdist_rpm", "--python=%s" % python], stdout=log_file, stderr=log_file)
    if ret_code:
        logging.warning("build unsuccessful; tailing log file")
        print "".join(open(log_filename, "rb").readlines()[-10:])
        successful_run = False
    if os.path.exists("dist"):
        for rpm_filename in [filename for filename in os.listdir("dist") if filename.endswith(".rpm")]:
            shutil.copyfile(os.path.join("dist", rpm_filename), os.path.join(rpm_dir, rpm_filename))

if not successful_run:
    sys.exit(1)