From 0940a0196db1e589a9c91652d92a284b28cd5629 Mon Sep 17 00:00:00 2001 From: "W. Trevor King" Date: Tue, 23 Jun 2009 11:44:06 -0400 Subject: [PATCH] Added becommands/tag.py Oops, I forgot to add becommands/tag.py with my last commit. Here it is now, with the added ability to remove tags. --- becommands/tag.py | 118 ++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 118 insertions(+) create mode 100644 becommands/tag.py diff --git a/becommands/tag.py b/becommands/tag.py new file mode 100644 index 0000000..0b19e91 --- /dev/null +++ b/becommands/tag.py @@ -0,0 +1,118 @@ +# Copyright (C) 2005 Aaron Bentley and Panometrics, Inc. +# +# +# 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; either version 2 of the License, or +# (at your option) any later version. +# +# 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. +# +# You should have received a copy of the GNU General Public License +# along with this program; if not, write to the Free Software +# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA +"""Tag a bug, or search bugs for tags.""" +from libbe import cmdutil, bugdir +import os, copy +__desc__ = __doc__ + +def execute(args, test=False): + """ + >>> from libbe import utility + >>> bd = bugdir.simple_bug_dir() + >>> os.chdir(bd.root) + >>> a = bd.bug_from_shortname("a") + >>> print a.extra_strings + [] + >>> execute(["a", "GUI"], test=True) + Tagging bug a: + GUI + >>> bd._clear_bugs() # resync our copy of bug + >>> a = bd.bug_from_shortname("a") + >>> print a.extra_strings + ['TAG:GUI'] + >>> execute(["a", "later"], test=True) + Tagging bug a: + GUI + later + >>> execute(["a"], test=True) + Tags for a: + GUI + later + >>> execute(["a", "Alphabetically first"], test=True) + Tagging bug a: + Alphabetically first + GUI + later + >>> bd._clear_bugs() # resync our copy of bug + >>> a = bd.bug_from_shortname("a") + >>> print a.extra_strings + ['TAG:Alphabetically first', 'TAG:GUI', 'TAG:later'] + >>> a.extra_strings = [] + >>> print a.extra_strings + [] + >>> a.save() + >>> execute(["a"], test=True) + Tags for a: + >>> bd._clear_bugs() # resync our copy of bug + >>> a = bd.bug_from_shortname("a") + >>> print a.extra_strings + [] + """ + parser = get_parser() + options, args = parser.parse_args(args) + cmdutil.default_complete(options, args, parser, + bugid_args={0: lambda bug : bug.active==True}) + + if len(args) < 1: + raise cmdutil.UsageError("Please specify a bug id.") + if len(args) > 2: + help() + raise cmdutil.UsageError("Too many arguments.") + + bd = bugdir.BugDir(from_disk=True, manipulate_encodings=not test) + bug = bd.bug_from_shortname(args[0]) + + new_tag = None + if len(args) == 2: + given_tag = args[1] + # reassign list so the change_hook realizes we've altered it. + tags = bug.extra_strings + tag_string = "TAG:%s" % given_tag + if options.remove == True: + tags.remove(tag_string) + else: # add the tag + new_tag = given_tag + tags.append(tag_string) + bug.extra_strings = tags + + tags = [] + for estr in bug.extra_strings: + if estr.startswith("TAG:"): + tags.append(estr[4:]) + + bd.save() + + if new_tag == None: + print "Tags for %s:" % bug.uuid + else: + print "Tagging bug %s:" % bug.uuid + for tag in tags: + print tag + +def get_parser(): + parser = cmdutil.CmdOptionParser("be tag BUG-ID [TAG]") + parser.add_option("-r", "--remove", action="store_true", dest="remove", + help="Remove TAG (instead of adding it)") + return parser + +longhelp=""" +If TAG is given, add TAG to BUG-ID. If it is not specified, just +print the tags for BUG-ID. +""" + +def help(): + return get_parser().help_str() + longhelp -- 2.26.2