From f7f0d9c959aee752298fdfe7a73939cf4c09fda5 Mon Sep 17 00:00:00 2001
From: "W. Trevor King" <wking@drexel.edu>
Date: Sat, 5 Dec 2009 03:32:06 -0500
Subject: Adjusted diff.Subscription.__init__() to guess type_root if required.

---
 libbe/diff.py | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

(limited to 'libbe/diff.py')

diff --git a/libbe/diff.py b/libbe/diff.py
index b5384a8..73db13d 100644
--- a/libbe/diff.py
+++ b/libbe/diff.py
@@ -75,13 +75,18 @@ def type_from_name(name, type_root, default=None, default_ok=False):
 
 class Subscription (object):
     """
-    >>> subscriptions = [Subscription('XYZ', 'all', type_root=BUG_TYPE_ALL),
-    ...                  Subscription('DIR', 'new', type_root=BUGDIR_TYPE_ALL),
+    >>> subscriptions = [Subscription('XYZ', 'all'),
+    ...                  Subscription('DIR', 'new'),
     ...                  Subscription('ABC', BUG_TYPE_ALL),]
     >>> print sorted(subscriptions)
     [<Subscription: DIR (new)>, <Subscription: ABC (all)>, <Subscription: XYZ (all)>]
     """
     def __init__(self, id, subscription_type, **kwargs):
+        if 'type_root' not in kwargs:
+            if id == BUGDIR_ID:
+                kwargs['type_root'] = BUGDIR_TYPE_ALL 
+            else:
+                kwargs['type_root'] = BUG_TYPE_ALL 
         if type(subscription_type) in types.StringTypes:
             subscription_type = type_from_name(subscription_type, **kwargs)
         self.id = id
-- 
cgit