ubuntu-dev-tools/lp-set-dup

#!/usr/bin/python
# -*- coding: UTF-8 -*-
"""Sets the "duplicate of" bug of a bug and its dups."""

# Copyright (c) 2009 Canonical Ltd.
#
# lp-set-dup is free software; you can redistribute it and/or modify it
# under the terms of the GNU General Public License as published by the
# Free Software Foundation; either version 2, or (at your option) any
# later version.
#
# lp-set-dup is distributed in the hope that it will be useful, but
# WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
# General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with lp-set-dup; see the file COPYING.  If not, write to the Free
# Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
# 02110-1301, USA.
#
# Authors:
#  Loïc Minier <lool@dooz.org>

import os, sys
from optparse import OptionParser

from ubuntutools.lp.libsupport as lp_libsupport

def die(message):
    print >> sys.stderr, "Fatal: " + message
    sys.exit(1)

if __name__ == '__main__':
    usage = "Usage: %prog [-f] <new main bug> <bug to dup> [<bug to dup>...]"
    optParser = OptionParser(usage)
    optParser.add_option("-f", action = "store_true",
        dest = "force", default = False,
        help = "Skip confirmation prompt")

    (options, args) = optParser.parse_args()

    if len(args) < 2:
        optParser.error("Need at least a new main bug and a bug to dup")

    launchpad = None
    try:
        print "Setting up Launchpad"
        launchpad = lp_libsupport.get_launchpad("ubuntu-dev-tools")
        print "Launchpad setup complete"
    except ImportError:
        suggestion = "check whether python-launchpadlib is installed"
    except IOError:
        suggestion = "you might want to \"manage-credentials create --consumer ubuntu-dev-tools --level 2\""
    if launchpad is None:
        die("Couldn't setup Launchpad for the ubuntu-dev-tools consumer; %s" (suggestion, ))

    # check that the new main bug isn't a duplicate
    new_main_bug = launchpad.bugs[args[0]]
    new_main_dup_of = new_main_bug.duplicate_of
    if new_main_dup_of is not None:
        s = None
        try:
            s = raw_input("Bug %s is a duplicate of %s; would you like to use %s as the new main bug instead? [y/N]" % (new_main_bug.id, new_main_dup_of.id, new_main_dup_of.id))
        except:
            die("Aborted")
        if s.lower() not in ("y", "yes"):
            die("User aborted")
        new_main_bug = new_main_dup_of

    # build list of bugs to process, first the dups then the bug
    bugs_to_process = []
    for b in args[1:]:
        print "Processing %s" % (b)
        bug = launchpad.bugs[b]
        dups = bug.duplicates
        if dups is not None:
            bugs_to_process.extend(dups)
            print "Found %i dups for %s" % (len(dups), b)
        bugs_to_process.append(bug)

    # process dups first, then their main bug
    print "Would set the following bugs as duplicates of %s: %s" % (new_main_bug.id, " ".join([str(b.id) for b in bugs_to_process]))

    if not options.force:
        s = None
        try:
            s = raw_input("Proceed? [y/N]")
        except:
            die("Aborted")
        if s.lower() not in ("y", "yes"):
            die("User aborted")

    for bug in bugs_to_process:
        print "Marking bug %s as a duplicate of %s" % (bug.id, new_main_bug.id)
        bug.duplicate_of = new_main_bug
        bug.lp_save()