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
|
#!/usr/bin/python
#
# Copyright (C) 2007, Canonical Ltd.
# Written by Daniel Holbach,
# Stefano Rivera,
# Brian Murray
#
# ##################################################################
#
# This program 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; version 3.
#
# This program 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.
#
# See file /usr/share/common-licenses/GPL-3 for more details.
#
# ##################################################################
from optparse import OptionParser
import codecs
import errno
import os
from lptools import config
USAGE = "lp-grab-attachments <bug numbers>"
def download_attachments(bug, descriptions):
bug_folder_name = 'bug-%s' % bug.id
try:
os.mkdir(bug_folder_name)
except OSError, error:
if error.errno == errno.EEXIST:
return
if descriptions:
description = bug.description
filename = os.path.join(os.getcwd(), bug_folder_name,
'Description.txt')
local_file = codecs.open(filename, encoding="utf-8", mode="w")
local_file.write(description)
local_file.close()
for attachment in bug.attachments:
f = attachment.data.open()
filename = os.path.join(os.getcwd(), bug_folder_name, f.filename)
local_file = open(filename, "w")
local_file.write(f.read())
f.close()
local_file.close()
def main():
parser = OptionParser('Usage: %prog [options] <bug numbers>')
parser.add_option('-d', '--duplicates', default=False,
action='store_true',
help='Download attachments from duplicates too')
parser.add_option('-p', '--package',
help='Download attachments from all bugs with an '
'open task for this Ubuntu source package')
parser.add_option('-P', '--project',
help='Download attachments from all bugs with a '
'open task for this project')
parser.add_option('-D', '--descriptions', default=False,
action='store_true',
help='Also download bug descriptions as Description.txt')
opts, args = parser.parse_args()
if len(args) < 1 and not opts.package and not opts.project:
parser.error('No bug numbers provided')
launchpad = config.get_launchpad("grab-attachments")
if opts.package:
ubuntu = launchpad.projects['ubuntu']
src_package = ubuntu.getSourcePackage(name=opts.package)
if src_package is None:
parser.error('Unable to find package %s' % opts.package)
for task in src_package.searchTasks():
args.append(task.bug.id)
if opts.project:
try:
project = launchpad.projects['%s' % opts.project]
except KeyError:
parser.error('Unable to find project %s' % opts.project)
for task in project.searchTasks():
args.append(task.bug.id)
for arg in args:
try:
bug_number = int(arg)
except ValueError:
parser.error("'%s' is not a valid bug number." % arg)
bug = launchpad.bugs[bug_number]
download_attachments(bug, opts.descriptions)
if opts.duplicates is True:
for bug in bug.duplicates:
download_attachments(bug, opts.descriptions)
if __name__ == '__main__':
main()
|
