From: W. Trevor King <wking@drexel.edu>
Date: Sat, 5 Dec 2009 08:32:06 +0000 (-0500)
Subject: Adjusted diff.Subscription.__init__() to guess type_root if required.
X-Git-Tag: 1.0.0~59^2~64^2~8
X-Git-Url: http://git.tremily.us/?a=commitdiff_plain;h=f7f0d9c959aee752298fdfe7a73939cf4c09fda5;p=be.git

Adjusted diff.Subscription.__init__() to guess type_root if required.
---

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