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, test=False):
  24     """
  25     >>> import os
  26     >>> bd = bugdir.simple_bug_dir()
  27     >>> os.chdir(bd.root)
  28     >>> execute(["a"], test=True)
  29     minor
  30     >>> execute(["a", "wishlist"], test=True)
  31     >>> execute(["a"], test=True)
  32     wishlist
  33     >>> execute(["a", "none"], test=True)
  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, manipulate_encodings=not test)
  43     bug = cmdutil.bug_from_shortname(bd, args[0])
  44     if len(args) == 1:
  45         print bug.severity
  46     elif len(args) == 2:
  47         try:
  48             bug.severity = args[1]
  49         except ValueError, e:
  50             if e.name != "severity":
  51                 raise e
  52             raise cmdutil.UserError ("Invalid severity level: %s" % e.value)
  53
  54 def get_parser():
  55     parser = cmdutil.CmdOptionParser("be severity BUG-ID [SEVERITY]")
  56     return parser
  57
  58 def help():
  59     longhelp=["""
  60 Show or change a bug's severity level.
  61
  62 If no severity is specified, the current value is printed.  If a severity level
  63 is specified, it will be assigned to the bug.
  64
  65 Severity levels are:
  66 """]
  67     try: # See if there are any per-tree severity configurations
  68         bd = bugdir.BugDir(from_disk=True, manipulate_encodings=False)
  69     except bugdir.NoBugDir, e:
  70         pass # No tree, just show the defaults
  71     longest_severity_len = max([len(s) for s in bug.severity_values])
  72     for severity in bug.severity_values :
  73         description = bug.severity_description[severity]
  74         s = "%*s : %s\n" % (longest_severity_len, severity, description)
  75         longhelp.append(s)
  76     longhelp = ''.join(longhelp)
  77     return get_parser().help_str() + longhelp
  78
  79 def complete(options, args, parser):
  80     for option,value in cmdutil.option_value_pairs(options, parser):
  81         if value == "--complete":
  82             # no argument-options at the moment, so this is future-proofing
  83             raise cmdutil.GetCompletions()
  84     for pos,value in enumerate(args):
  85         if value == "--complete":
  86             try: # See if there are any per-tree severity configurations
  87                 bd = bugdir.BugDir(from_disk=True,
  88                                    manipulate_encodings=False)
  89             except bugdir.NoBugDir:
  90                 bd = None
  91             if pos == 0: # fist positional argument is a bug id
  92                 ids = []
  93                 if bd != None:
  94                     bd.load_all_bugs()
  95                     filter = lambda bg : bg.active==True
  96                     bugs = [bg for bg in bd if filter(bg)==True]
  97                     ids = [bd.bug_shortname(bg) for bg in bugs]
  98                 raise cmdutil.GetCompletions(ids)
  99             elif pos == 1: # second positional argument is a severity
 100                 raise cmdutil.GetCompletions(bug.severity_values)
 101             raise cmdutil.GetCompletions()