ubuntu-dev-tools/grab-attachments
2011-04-19 12:43:25 -07:00

83 lines
2.6 KiB
Python
Executable File

#!/usr/bin/python
#
# Copyright (C) 2007, Canonical Ltd.
# Written by Daniel Holbach,
# Stefano Rivera
#
# ##################################################################
#
# 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 os
import sys
from launchpadlib.launchpad import Launchpad
from ubuntutools.config import UDTConfig
USAGE = "grab-attachments <bug numbers>"
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")
opts, args = parser.parse_args()
if len(args) < 1:
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)
for arg in args:
try:
number = int(arg)
except ValueError:
parser.error("'%s' is not a valid bug number." % arg)
bug = launchpad.bugs[number]
bug_folder_name = 'bug-%s' % number
try:
os.mkdir(bug_folder_name)
except OSError, error:
if error.errno == 17:
continue
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()
# no LP credentials
except IOError, error:
print error
sys.exit(1)
if __name__ == '__main__':
main()