ubuntu-dev-tools/grab-attachments
Brian Murray 146b1619f1 * grab-attachments, doc/grab-attachments.1:
- add in download attachments from duplicates
  - add in download attachments from all bugs about a package
  - document new options in the manpage
2011-06-16 12:02:04 -07:00

109 lines
3.4 KiB
Python
Executable File

#!/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 errno
import os
import sys
from launchpadlib.launchpad import Launchpad
from ubuntutools.config import UDTConfig
USAGE = "grab-attachments <bug numbers>"
def download_attachments(bug):
bug_folder_name = 'bug-%s' % bug.id
try:
os.mkdir(bug_folder_name)
except OSError, error:
if error.errno == errno.EEXIST:
return
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('-l', '--lpinstance', metavar='INSTANCE',
dest='lpinstance', default=None,
help='Launchpad instance to connect to '
'(default: production)')
parser.add_option('--no-conf',
dest='no_conf', default=False, action='store_true',
help="Don't read config files or environment variables")
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 a '
'task for this source package')
opts, args = parser.parse_args()
if len(args) < 1 and not opts.package:
parser.error('No bug numbers provided')
config = UDTConfig(opts.no_conf)
if opts.lpinstance is None:
opts.lpinstance = config.get_value('LPINSTANCE')
try:
launchpad = Launchpad.login_with("ubuntu-dev-tools", opts.lpinstance)
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)
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)
if opts.duplicates is True:
for bug in bug.duplicates:
download_attachments(bug)
# no LP credentials
except IOError, error:
print error
sys.exit(1)
if __name__ == '__main__':
main()