From f7f0d9c959aee752298fdfe7a73939cf4c09fda5 Mon Sep 17 00:00:00 2001 From: "W. Trevor King" Date: Sat, 5 Dec 2009 03:32:06 -0500 Subject: [PATCH] Adjusted diff.Subscription.__init__() to guess type_root if required. --- libbe/diff.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) 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) [, , ] """ 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 -- 2.26.2