diff options
Diffstat (limited to 'becommands/subscribe.py')
| -rw-r--r-- | becommands/subscribe.py | 40 |
1 files changed, 29 insertions, 11 deletions
diff --git a/becommands/subscribe.py b/becommands/subscribe.py index b754937..2522f0c 100644 --- a/becommands/subscribe.py +++ b/becommands/subscribe.py @@ -135,21 +135,30 @@ def execute(args, manipulate_encodings=True): bug = bd.bug_from_shortname(args[0]) entity = bug entity_name = bug.uuid + if options.list_all == True: + entity_name = "anything in the bug directory" types = [type_from_name(name, type_root, default=INVALID_TYPE, default_ok=options.unsubscribe) for name in types] estrs = entity.extra_strings - if options.unsubscribe == True: - estrs = unsubscribe(estrs, subscriber, types, servers, type_root) - else: # add the tag - estrs = subscribe(estrs, subscriber, types, servers, type_root) - entity.extra_strings = estrs # reassign to notice change - - subscriptions = [] - for estr in entity.extra_strings: - if estr.startswith(TAG): - subscriptions.append(estr[len(TAG):]) + if options.list == True or options.list_all == True: + pass + else: # alter subscriptions + if options.unsubscribe == True: + estrs = unsubscribe(estrs, subscriber, types, servers, type_root) + else: # add the tag + estrs = subscribe(estrs, subscriber, types, servers, type_root) + entity.extra_strings = estrs # reassign to notice change + + if options.list_all == True: + bd.load_all_bugs() + subscriptions = get_bugdir_subscribers(bd, servers[0]) + else: + subscriptions = [] + for estr in entity.extra_strings: + if estr.startswith(TAG): + subscriptions.append(estr[len(TAG):]) if len(subscriptions) > 0: print "Subscriptions for %s:" % entity_name @@ -161,6 +170,12 @@ def get_parser(): parser.add_option("-u", "--unsubscribe", action="store_true", dest="unsubscribe", default=False, help="Unsubscribe instead of subscribing.") + parser.add_option("-a", "--list-all", action="store_true", + dest="list_all", default=False, + help="List all subscribers (no ID argument, read only action).") + parser.add_option("-l", "--list", action="store_true", + dest="list", default=False, + help="List subscribers (read only action).") parser.add_option("-s", "--subscriber", dest="subscriber", metavar="SUBSCRIBER", help="Email address of the subscriber (defaults to bugdir.user_id).") @@ -325,7 +340,7 @@ def get_subscribers(extra_strings, type, server, type_root, type_match = True break server_match = False - if server in servers or servers == ["*"]: + if server in servers or servers == ["*"] or server == "*": server_match = True if type_match == True and server_match == True: yield subscriber @@ -338,6 +353,9 @@ def get_bugdir_subscribers(bugdir, server): where id is either a bug.uuid (in the case of a bug subscription) or "DIR" (in the case of a bugdir subscription). + Only checks bugs that are currently in memory, so you might want + to call bugdir.load_all_bugs() first. + >>> bd = bugdir.SimpleBugDir(sync_with_disk=False) >>> a = bd.bug_from_shortname("a") >>> bd.extra_strings = subscribe(bd.extra_strings, "John Doe <j@doe.com>", [BUGDIR_TYPE_ALL], ["a.com"], BUGDIR_TYPE_ALL) |