becommands/severity.py

   1 # Copyright (C) 2005-2009 Aaron Bentley and Panometrics, Inc.
   2 #                         Marien Zwart <marienz@gentoo.org>
   3 #                         Thomas Gerigk <tgerigk@gmx.de>
   4 #                         W. Trevor King <wking@drexel.edu>
   5 #
   6 # This program is free software; you can redistribute it and/or modify
   7 # it under the terms of the GNU General Public License as published by
   8 # the Free Software Foundation; either version 2 of the License, or
   9 # (at your option) any later version.
  10 #
  11 # This program is distributed in the hope that it will be useful,
  12 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14 # GNU General Public License for more details.
  15 #
  16 # You should have received a copy of the GNU General Public License along
  17 # with this program; if not, write to the Free Software Foundation, Inc.,
  18 # 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  19 """Show or change a bug's severity level"""
  20 from libbe import cmdutil, bugdir, bug
  21 __desc__ = __doc__
  22
  23 def execute(args, manipulate_encodings=True):
  24     """
  25     >>> import os
  26     >>> bd = bugdir.simple_bug_dir()
  27     >>> os.chdir(bd.root)
  28     >>> execute(["a"], manipulate_encodings=False)
  29     minor
  30     >>> execute(["a", "wishlist"], manipulate_encodings=False)
  31     >>> execute(["a"], manipulate_encodings=False)
  32     wishlist
  33     >>> execute(["a", "none"], manipulate_encodings=False)
  34     Traceback (most recent call last):
  35     UserError: Invalid severity level: none
  36     """
  37     parser = get_parser()
  38     options, args = parser.parse_args(args)
  39     complete(options, args, parser)
  40     if len(args) not in (1,2):
  41         raise cmdutil.UsageError
  42     bd = bugdir.BugDir(from_disk=True,
  43                        manipulate_encodings=manipulate_encodings)
  44     bug = bd.bug_from_shortname(args[0])
  45     if len(args) == 1:
  46         print bug.severity
  47     elif len(args) == 2:
  48         try:
  49             bug.severity = args[1]
  50         except ValueError, e:
  51             if e.name != "severity":
  52                 raise e
  53             raise cmdutil.UserError ("Invalid severity level: %s" % e.value)
  54         bd.save()
  55
  56 def get_parser():
  57     parser = cmdutil.CmdOptionParser("be severity BUG-ID [SEVERITY]")
  58     return parser
  59
  60 def help():
  61     longhelp=["""
  62 Show or change a bug's severity level.
  63
  64 If no severity is specified, the current value is printed.  If a severity level
  65 is specified, it will be assigned to the bug.
  66
  67 Severity levels are:
  68 """]
  69     try: # See if there are any per-tree severity configurations
  70         bd = bugdir.BugDir(from_disk=True, manipulate_encodings=False)
  71     except bugdir.NoBugDir, e:
  72         pass # No tree, just show the defaults
  73     longest_severity_len = max([len(s) for s in bug.severity_values])
  74     for severity in bug.severity_values :
  75         description = bug.severity_description[severity]
  76         s = "%*s : %s\n" % (longest_severity_len, severity, description)
  77         longhelp.append(s)
  78     longhelp = ''.join(longhelp)
  79     return get_parser().help_str() + longhelp
  80
  81 def complete(options, args, parser):
  82     for option,value in cmdutil.option_value_pairs(options, parser):
  83         if value == "--complete":
  84             # no argument-options at the moment, so this is future-proofing
  85             raise cmdutil.GetCompletions()
  86     for pos,value in enumerate(args):
  87         if value == "--complete":
  88             try: # See if there are any per-tree severity configurations
  89                 bd = bugdir.BugDir(from_disk=True,
  90                                    manipulate_encodings=False)
  91             except bugdir.NoBugDir:
  92                 bd = None
  93             if pos == 0: # fist positional argument is a bug id
  94                 ids = []
  95                 if bd != None:
  96                     bd.load_all_bugs()
  97                     filter = lambda bg : bg.active==True
  98                     bugs = [bg for bg in bd if filter(bg)==True]
  99                     ids = [bd.bug_shortname(bg) for bg in bugs]
 100                 raise cmdutil.GetCompletions(ids)
 101             elif pos == 1: # second positional argument is a severity
 102                 raise cmdutil.GetCompletions(bug.severity_values)
 103             raise cmdutil.GetCompletions()