ubuntu-dev-tools/massfile

#!/usr/bin/python
# -*- coding: utf-8 -*-
#
# Copyright (C) 2007 Canonical Ltd.
#
# Modified by Iain Lane <iain@orangesquash.org.uk>, taking some code written by
# Daniel Hahler <ubuntu@thequod.de>
#
# python-launchpadlib support was added by Markus Korn <thekorn@gmx.de>.
#
# ##################################################################
#
# 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.
#
# ##################################################################

import optparse
import os
import sys
import email

from launchpadlib.launchpad import Launchpad

from ubuntutools.config import UDTConfig
from ubuntutools.lp.libsupport import translate_web_api

def read_config():
    instructions_file = open("instructions")
    instructions = email.message_from_file(instructions_file)
    instructions_file.close()
    instr = dict()

    for field in "subject", "assignee", "subscribers", "tags", "text", \
                 "buglist-url", "status":
        instr[field] = instructions.get(field)

    return instr

def read_list():
    pack_list = set()

    listfile = open("list")
    for line in listfile.readlines():
        if line.strip()!="":
            pack_list.add(line.strip("\n"))

    listfile.close()
    return pack_list

def check_configfiles():
    result = True

    bin_path = os.path.dirname(os.path.abspath(__file__))
    if bin_path == "/usr/bin":
        example_dir = "/usr/share/doc/ubuntu-dev-tools/examples"
    else:
        example_dir = "%s/examples" % bin_path

    if not os.path.exists("instructions"):
        os.system("cp %s/massfile.instructions instructions" % example_dir)
        print >> sys.stderr, \
            "No 'instructions' file found. Copied template from %s." % \
            example_dir
        result = False

    if not os.path.exists("list"):
        os.system("cp %s/massfile.list list" % example_dir)
        print >> sys.stderr, \
            "No 'list' file found. Copied template from %s." % example_dir
        result = False

    return result


def file_bug(config, launchpad):
    try:
        summary = config["subject"].replace("$pack", config["sourcepackage"])
        description = config["text"].replace("$pack", config["sourcepackage"])

        product_url = "%subuntu/+source/%s" % \
                      (launchpad._root_uri, config["sourcepackage"])
        tags = [t for t in [t.strip("\n").strip() for t in
                            config["tags"].split(",")] if t]
        bug = launchpad.bugs.createBug(description=description, title=summary,
            target=product_url, tags=tags)

        print "Successfully filed bug %i: %s" % \
              (bug.id, bug.web_link)

        subscribers = [s for s in [s.strip("\n").strip() for s in
                                   config["subscribers"].split(",")] if s]
        for sub in subscribers:
            subscribe_url = "%s~%s" % (launchpad._root_uri, sub)
            bug.subscribe(person=subscribe_url)

        #newly created bugreports have one task
        task = bug.bug_tasks[0]

        if config["status"]:
            status = config["status"].capitalize()
        else:
            status = "Confirmed"
        task.status = status

        assignee = config["assignee"]
        if assignee:
            assignee_url = "%s~%s" % (launchpad._root_uri, assignee)
            task.assignee = assignee_url
        task.lp_save()
    except:
        print >> sys.stderr, "Bug for '%s' was not filed." % \
                             config["sourcepackage"]

def read_buglist(url, launchpad):
    if not url:
        return set()

    if len(url.split("?", 1)) == 2:
        # search options not supported, because there is no mapping
        # web ui options <-> API options
        print >> sys.stderr, "Options in url are not supported, url: %s" % url
        sys.exit(1)

    packages = set()

    api_url = translate_web_api(url, launchpad)
    # workaround LP #303414
    # if this is fixed it should simply be: buglist = launchpad.load(api_url)
    api_url = api_url.split("?", 1)[0]
    project = launchpad.load(api_url)
    buglist = project.searchTasks()

    for bug in buglist:
        packages.add(bug.bug_target_name)

    return packages

def main():
    description = ('Files bugs against multiple packages in Ubuntu. '
                   'Reads the bug from "instructions" and files them against '
                   'packages listed in "list". '
                   "If these files aren't preset they are created.")
    parser = optparse.OptionParser(description=description)
    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")
    options = parser.parse_args()[0]
    udtconfig = UDTConfig(options.no_conf)
    if options.lpinstance is None:
        options.lpinstance = udtconfig.get_value('LPINSTANCE')

    if not check_configfiles():
        sys.exit(1)

    launchpad = Launchpad.login_with('ubuntu-dev-tools', options.lpinstance)
    config = read_config()
    pack_list = read_list()
    buglist = read_buglist(config["buglist-url"], launchpad)

    for pack in pack_list:
        if pack not in buglist:
            config["sourcepackage"] = pack
            file_bug(config, launchpad)


if __name__ == '__main__':
    main()