1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
|
#!/usr/bin/python3
# -*- 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 3, 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 sys
from optparse import OptionParser
from launchpadlib.errors import HTTPError
from lptools import config
def die(message):
print("Fatal: " + message, file=sys.stderr)
sys.exit(1)
def main():
usage = "Usage: %prog [-f] <new main bug> <bug to dup> [<bug to dup>...]"
opt_parser = OptionParser(usage)
opt_parser.add_option("-f",
help="Skip confirmation prompt",
dest="force", default=False, action="store_true")
opt_parser.add_option("-l", "--lpinstance", metavar="INSTANCE",
help="Launchpad instance to connect to "
"(default: production)",
dest="lpinstance", default=None)
opt_parser.add_option("--no-conf",
help="Don't read config files or "
"environment variables.",
dest="no_conf", default=False, action="store_true")
(options, args) = opt_parser.parse_args()
if len(args) < 2:
opt_parser.error("Need at least a new main bug and a bug to dup")
launchpad = config.get_launchpad("set-dup")
# check that the new main bug isn't a duplicate
try:
new_main_bug = launchpad.bugs[args[0]]
except HTTPError as error:
if error.response.status == 401:
print(("E: Don't have enough permissions to access "
"bug %s") % (args[0]), file=sys.stderr)
die(error.content)
else:
raise
new_main_dup_of = new_main_bug.duplicate_of
if new_main_dup_of is not None:
answer = None
try:
answer = 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 answer.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 bug_number in args[1:]:
print("Processing %s" % (bug_number))
try:
bug = launchpad.bugs[bug_number]
except HTTPError as error:
if error.response.status == 401:
print(("W: Don't have enough permissions to "
"access bug %s") % (bug_number), file=sys.stderr)
print("W: %s" % (error.content), file=sys.stderr)
continue
else:
raise
dups = bug.duplicates
if dups is not None:
bugs_to_process.extend(dups)
print("Found %i dups for %s" % (len(dups), bug_number))
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:
answer = None
try:
answer = input("Proceed? [y/N]")
except:
die("Aborted")
if answer.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()
if __name__ == '__main__':
main()
|
