freebsd-dev/release/scripts/package-split.py

175 lines
5.6 KiB
Python

#!/usr/local/bin/python
#
# This script generates a master INDEX file for the CD images built by the
# FreeBSD release engineers. Each disc is given a list of desired packages.
# Dependencies of these packages are placed on either the same disc or an
# earlier disc. The resulting master INDEX file is then written out.
#
# Usage: package-split.py <INDEX> <master INDEX>
#
# $FreeBSD$
import os
import sys
try:
arch = os.environ["PKG_ARCH"]
except:
arch = os.uname()[4]
print "Using arch %s" % (arch)
if 'PKG_VERBOSE' in os.environ:
verbose = 1
else:
verbose = 0
# List of packages for disc1. This just includes packages sysinstall can
# install as a distribution
def disc1_packages():
pkgs = ['lang/perl5.8']
pkgs.extend(['x11/xorg',
'devel/imake'])
if arch == 'i386':
pkgs.append('emulators/linux_base-fc4')
return pkgs
# List of packages for disc2. This includes packages that the X desktop
# menu depends on (if it still exists) and other "nice to have" packages.
# For architectures that use a separate livefs, this is actually disc3.
def disc2_packages():
# X Desktops
if arch == 'ia64':
pkgs = ['x11/gnome2-lite',
'x11/kde-lite']
else:
pkgs = ['x11/gnome2',
'x11/kde3']
pkgs.extend(['x11-wm/afterstep',
'x11-wm/windowmaker',
'x11-wm/fvwm2',
# "Nice to have"
'archivers/unzip',
'astro/xearth',
'devel/gmake',
'editors/emacs',
'editors/vim-lite',
'emulators/mtools',
'graphics/png',
'graphics/xv',
'irc/xchat',
'mail/exim',
'mail/fetchmail',
'mail/mutt',
'mail/pine4',
'mail/popd',
'mail/xfmail',
'mail/postfix',
'net/cvsup-without-gui',
'net/rsync',
'net/samba3',
'news/slrn',
'news/tin',
'ports-mgmt/portupgrade',
'print/a2ps-letter',
'print/apsfilter',
'print/ghostscript-gnu-nox11',
'print/gv',
'print/psutils-letter',
'shells/bash',
'shells/pdksh',
'shells/zsh',
'security/sudo',
'www/links',
'www/lynx',
'x11/rxvt',
# Formerly on disc3
'ports-mgmt/portaudit'])
return pkgs
# The list of desired packages
def desired_packages():
disc1 = disc1_packages()
disc2 = disc2_packages()
return [disc1, disc2]
# Suck the entire INDEX file into a two different dictionaries. The first
# dictionary maps port names (origins) to package names. The second
# dictionary maps a package name to a list of its dependent packages.
PACKAGE_COL=0
ORIGIN_COL=1
DEPENDS_COL=8
def load_index(index):
deps = {}
pkgs = {}
line_num = 1
for line in index:
fields = line.split('|')
name = fields[PACKAGE_COL]
if name in deps:
sys.stderr.write('%d: Duplicate package %s\n' % (line_num, name))
sys.exit(1)
origin = fields[ORIGIN_COL].replace('/usr/ports/', '', 1)
if origin in pkgs:
sys.stderr.write('%d: Duplicate port %s\n' % (line_num, origin))
sys.exit(1)
deps[name] = fields[DEPENDS_COL].split()
pkgs[origin] = name
line_num = line_num + 1
return (deps, pkgs)
# Layout the packages on the various CD images. Here's how it works. We walk
# each disc in the list of discs. Within each disc we walk the list of ports.
# For each port, we add the package name to a dictionary with the value being
# the current disc number. We also add all of the dependent packages. If
# a package is already in the dictionary when we go to add it, we just leave
# the dictionary as it is. This means that each package ends up on the first
# disc that either lists it or contains it as a dependency.
def layout_discs(discs, pkgs, deps):
disc_num = 1
layout = {}
for disc in discs:
for port in disc:
if port not in pkgs:
sys.stderr.write('Disc %d: Unable to find package for %s\n' %
(disc_num, port))
continue
pkg = pkgs[port]
pkg_list = [pkg] + deps[pkg]
for pkg in pkg_list:
if pkg not in layout:
if verbose:
print "--> Adding %s to Disc %d" % (pkg, disc_num)
layout[pkg] = disc_num
disc_num = disc_num + 1
return layout
# Generate a master INDEX file based on the generated layout. The way this
# works is that for each INDEX line, we check to see if the package is in the
# layout. If it is, we put that INDEX line into the master INDEX and append
# a new field with the disc number to the line.
def generate_index(index, layout, master_index):
for line in index:
pkg = line.split('|')[PACKAGE_COL]
if pkg in layout:
new_line = '%s|%d\n' % (line.splitlines()[0], layout[pkg])
master_index.write(new_line)
# Verify the command line arguments
if len(sys.argv) != 3:
sys.stderr.write('Invalid number of arguments\n')
sys.stderr.write('Usage: package-split.py <source INDEX> <master INDEX>\n')
sys.exit(1)
print "Loading %s..." % (sys.argv[1])
index = file(sys.argv[1])
(deps, pkgs) = load_index(index)
discs = desired_packages()
layout = layout_discs(discs, pkgs, deps)
index.seek(0)
print "Generating %s..." % (sys.argv[2])
master_index = file(sys.argv[2], 'w')
generate_index(index, layout, master_index)
index.close()
master_index.close()