ubuntu-dev-tools/bitesize

#!/usr/bin/python3
"""Add 'bitesize' tag to bugs and add a comment."""

# Copyright (c) 2011 Canonical Ltd.
#
# bitesize 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.
#
# bitesize 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 bitesize; see the file COPYING.  If not, write to the Free
# Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
# 02110-1301, USA.
#
# Authors:
#  Daniel Holbach <daniel.holbach@canonical.com>

import sys
from optparse import OptionParser

from launchpadlib.launchpad import Launchpad
from launchpadlib.errors import HTTPError

from ubuntutools.config import UDTConfig

from ubuntutools import getLogger

Logger = getLogger()


def error_out(msg):
    Logger.error(msg)
    sys.exit(1)


def save_entry(entry):
    try:
        entry.lp_save()
    except HTTPError as error:
        error_out(error.content)


def tag_bug(bug):
    bug.tags = bug.tags + ["bitesize"]  # LP: #254901 workaround
    save_entry(bug)


def main():
    usage = "Usage: %prog <bug number>"
    opt_parser = OptionParser(usage)
    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()
    config = UDTConfig(options.no_conf)
    if options.lpinstance is None:
        options.lpinstance = config.get_value("LPINSTANCE")
    if len(args) < 1:
        opt_parser.error("Need at least one bug number.")

    launchpad = Launchpad.login_with("ubuntu-dev-tools", options.lpinstance)
    if launchpad is None:
        error_out("Couldn't authenticate to Launchpad.")

    # check that the new main bug isn't a duplicate
    try:
        bug = launchpad.bugs[args[0]]
    except HTTPError as error:
        if error.response.status == 401:
            error_out(
                "Don't have enough permissions to access bug %s. %s" % (args[0], error.content)
            )
        else:
            raise
    if "bitesize" in bug.tags:
        error_out("Bug is already marked as 'bitesize'.")
    bug.newMessage(
        content="I'm marking this bug as 'bitesize' as it looks "
        "like an issue that is easy to fix and suitable "
        "for newcomers in Ubuntu development. If you need "
        "any help with fixing it, talk to me about it."
    )
    bug.subscribe(person=launchpad.me)
    tag_bug(launchpad.bugs[bug.id])  # fresh bug object, LP: #336866 workaround


if __name__ == "__main__":
    main()