bh@287: #! /usr/bin/python
bh@243: # Copyright (C) 2007, 2008, 2009 by Intevation GmbH
bh@177: # Authors:
bh@177: # Bernhard Herzog <bh@intevation.de>
bh@177: #
bh@177: # This program is free software under the GPL (>=v2)
bh@177: # Read the file COPYING coming with the software for details.
bh@177:
bh@177: """Usage: treepkgbuilder COMMAND [options]
bh@177:
bh@177: Manage the pbuilder environment for a tree packager installation. The
bh@177: pbuilder settings are taken from the configuration file, either the
bh@177: default treepkg.cfg or the one given with the --config-file option.
bh@177: Also, this script assumes that there is only one pbuilder setting for
bh@177: all packagers. Use treepkgbuilder COMMAND --help for more details about
bh@177: the commands.
bh@177: """
bh@177:
bh@177: import sys
bh@177:
bh@177: import treepkgcmd
bh@177: from treepkg.options import create_parser
bh@177: from treepkg.packager import create_package_track, PackagerGroup
bh@177: from treepkg.readconfig import read_config
bh@177:
bh@177:
bh@177: class Command(object):
bh@177:
bh@177: names = ()
bh@177:
bh@177: def __init__(self, arguments):
bh@177: parser = self.create_parser()
bh@177: self.opts, self.rest = parser.parse_args(arguments)
bh@177:
bh@177: def create_parser(self):
bh@177: return create_parser()
bh@177:
bh@177: def read_config(self):
bh@177: self.treepkg_opts, self.packager_opts \
bh@177: = read_config(self.opts.config_file)
bh@177:
bh@177: def get_builder(self):
bh@177: self.read_config()
bh@177: group = PackagerGroup([create_package_track(**opts)
bh@177: for opts in self.packager_opts],
bh@177: **self.treepkg_opts)
bh@177: track = group.get_package_tracks()[0]
bh@177: return track.builder
bh@177:
bh@177:
bh@177:
bh@177: class InitCommand(Command):
bh@177:
bh@177: names = ("init",)
bh@177:
bh@177: def __init__(self, arguments):
bh@177: super(InitCommand, self).__init__(arguments)
bh@177: if self.opts.mirrorsite is None:
bh@177: print >>sys.stderr, "Missing required option --mirrorsite"
bh@177: sys.exit(1)
bh@177:
bh@177: def create_parser(self):
bh@177: parser = super(InitCommand, self).create_parser()
bh@177: parser.set_defaults(distribution="etch")
bh@177: parser.add_option("--mirrorsite",
bh@177: help=("The debian mirror site"
bh@177: " (pbuilder MIRRORSITE setting). Required."))
bh@177: parser.add_option("--othermirror",
bh@177: help=("Extra contents of the OTHERMIRROR setting."
bh@177: " See the pbuilder documentation for the"
bh@177: " format."))
bh@177: parser.add_option("--distribution",
bh@177: help=("The debian distribution for the pbuilder"
bh@177: " chroot. Default is etch."))
bh@177: return parser
bh@177:
bh@177: def run(self):
bh@177: builder = self.get_builder()
bricks@344: builder.init_builder(distribution=self.opts.distribution,
bh@177: mirrorsite=self.opts.mirrorsite,
bh@177: extramirrors=self.opts.othermirror)
bh@177:
bh@177:
bh@192: class AddBinariesCommand(Command):
bh@192:
bh@192: names = ("add-binaries",)
bh@192:
bh@296: def create_parser(self):
bh@296: parser = super(AddBinariesCommand, self).create_parser()
bh@296: parser.set_defaults(subdir="manual")
bh@296: parser.add_option("--subdir",
bh@296: help=("The subdirectory of extra-pkg into which"
bh@296: " the packages are to be copied"
bh@296: " (default 'manual')."))
bh@296: return parser
bh@296:
bh@296:
bh@192: def run(self):
bh@192: builder = self.get_builder()
bh@296: builder.add_binaries_to_extra_pkg(self.rest,
bh@296: subdirectory=self.opts.subdir)
bh@192:
bh@192:
bh@177: class UpdateCommand(Command):
bh@177:
bh@177: names = ("update",)
bh@177:
bh@177: def run(self):
bh@177: builder = self.get_builder()
bh@177: builder.update(suppress_output=False, log_info=False)
bh@177:
bh@177:
bh@184: class AddKeyCommand(Command):
bh@184:
bh@184: names = ("addkey", "add-key")
bh@184:
bh@184: def __init__(self, arguments):
bh@184: super(AddKeyCommand, self).__init__(arguments)
bh@184: if not self.opts.key_id:
bh@184: print >>sys.stderr, "No key id given"
bh@184: sys.exit(1)
bh@184:
bh@184: def create_parser(self):
bh@184: parser = super(AddKeyCommand, self).create_parser()
bh@184: parser.add_option("--key-id",
bh@184: help=("The id of the key to add. Required."))
bh@184: return parser
bh@184:
bh@184: def run(self):
bh@184: builder = self.get_builder()
bh@184: builder.add_apt_key(self.opts.key_id)
bh@184:
bh@184:
bh@201: class LoginCommand(Command):
bh@201:
bh@201: names = ("login",)
bh@201:
bh@213: def create_parser(self):
bh@213: parser = super(LoginCommand, self).create_parser()
bh@213: parser.set_defaults(bind_mount=[])
bh@213: parser.add_option("--bind-mount", action="append",
bh@213: help=("Directories to bind-mount in the chroot"
bh@213: " environment."))
bh@245: parser.add_option("--save-after-login", action="store_true",
bh@245: help="Copy changes back to the tarball.")
bh@213: return parser
bh@213:
bh@213:
bh@201: def run(self):
bh@245: self.get_builder().login(bindmounts=self.opts.bind_mount,
bh@245: save_after_login=self.opts.save_after_login)
bh@201:
bh@201:
bh@210: class BuildCommand(Command):
bh@210:
bh@210: names = ("build",)
bh@210:
bh@210: def create_parser(self):
bh@210: parser = super(BuildCommand, self).create_parser()
bh@210: parser.add_option("--logfile", help=("The file for the build log."))
bh@210: parser.add_option("--binary-dir",
bh@210: help=("Directory for the binary packages."))
bh@210: return parser
bh@210:
bh@210: def run(self):
bh@210: if len(self.rest) != 1:
bh@210: print >>sys.stderr, "Exactly one .dsc file must be specified"
bh@210: sys.exit(1)
bh@210: dsc_file = self.rest[0]
bh@210: self.get_builder().build(dsc_file, binary_dir=self.opts.binary_dir,
bh@210: logfile=self.opts.logfile)
bh@210:
bh@210:
bh@243: class ExecuteCommand(Command):
bh@243:
bh@243: names = ("execute",)
bh@243:
bh@243: def create_parser(self):
bh@243: parser = super(ExecuteCommand, self).create_parser()
bh@243: parser.add_option("--logfile", help=("Log file."))
bh@243: parser.add_option("--bind-mount", action="append",
bh@243: help=("Directories to bind-mount in the chroot"
bh@243: " environment."))
bh@243: return parser
bh@243:
bh@243: def run(self):
bh@243: if not self.rest:
bh@243: print >>sys.stderr, "At least the script file must be specified"
bh@243: sys.exit(1)
bh@243: script = self.rest
bh@243: self.get_builder().run_script(script, bindmounts=self.opts.bind_mount,
bh@243: logfile=self.opts.logfile)
bh@243:
bh@243:
bh@177: class HelpCommand(Command):
bh@177:
bh@177: names = ("help", "--help", "-h")
bh@177:
bh@177: def run(self):
bh@177: print __doc__.rstrip()
bh@177: print "Supported commands:"
bh@177: commands = sorted((cmd for cmd in iter_commands() if cmd.names),
bh@177: key=lambda cmd: cmd.names[0])
bh@177: for cmd in commands:
bh@177: print " ", " ".join(cmd.names)
bh@177:
bh@177:
bh@177: def iter_commands():
bh@177: for key, obj in globals().items():
bh@177: if key.endswith("Command"):
bh@177: yield obj
bh@177:
bh@177: def get_command_class(name):
bh@177: for obj in iter_commands():
bh@177: if name in obj.names:
bh@177: return obj
bh@177: return None
bh@177:
bh@177:
bh@177: def main():
bh@177: arguments = sys.argv[1:]
bh@177:
bh@177: if len(arguments) < 1:
bh@177: print >>sys.stderr, "Missing command"
bh@177: sys.exit(1)
bh@177:
bh@177: command_name = arguments[0]
bh@177: arguments = arguments[1:]
bh@177:
bh@177: command_class = get_command_class(command_name)
bh@177: if command_class is None:
bh@177: print >>sys.stderr, "Unknown command %r" % command_name
bh@177: sys.exit(1)
bh@177:
bh@177: command = command_class(arguments)
bh@177: command.run()
bh@177:
bh@177:
bh@177: main()