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.

1 files changed, 118 insertions, 0 deletions

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.
+# <abentley@panoramicfeedback.com>
+#
+#    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