Rewrite commands to use bugdirs instead of a single bugdir.

[be.git] / libbe / command / severity.py

diff --git a/libbe/command/severity.py b/libbe/command/severity.py
index 67e4c386176a5f2b2634bd061f9d431ea21ca96e..51096a799dbc18147cc2fddaa56301c4eca129c7 100644 (file)
--- a/libbe/command/severity.py
+++ b/libbe/command/severity.py
@@ -36,7 +36,7 @@ class Severity (libbe.command.Command):
     >>> io = libbe.command.StringInputOutput()
     >>> io.stdout = sys.stdout
     >>> ui = libbe.command.UserInterface(io=io)
-    >>> ui.storage_callbacks.set_bugdir(bd)
+    >>> ui.storage_callbacks.set_storage(bd.storage)
     >>> cmd = Severity(ui=ui)
 
     >>> bd.bug_from_uuid('a').severity
@@ -66,10 +66,11 @@ class Severity (libbe.command.Command):
                 ])
 
     def _run(self, **params):
-        bugdir = self._get_bugdir()
+        bugdirs = self._get_bugdirs()
         for bug_id in params['bug-id']:
-            bug,dummy_comment = \
-                libbe.command.util.bug_comment_from_user_id(bugdir, bug_id)
+            bugdir,bug,comment = (
+                libbe.command.util.bugdir_bug_comment_from_user_id(
+                    bugdirs, bug_id))
             if bug.severity != params['severity']:
                 try:
                     bug.severity = params['severity']
@@ -82,7 +83,7 @@ class Severity (libbe.command.Command):
 
     def _long_help(self):
         try: # See if there are any per-tree severity configurations
-            bd = self._get_bugdir()
+            bugdirs = self._get_bugdirs()
         except NotImplementedError:
             pass # No tree, just show the defaults
         longest_severity_len = max([len(s) for s in libbe.bug.severity_values])