diff options
| author | Barry Warsaw | 2016-09-08 19:49:00 +1200 |
|---|---|---|
| committer | Barry Warsaw | 2016-09-08 19:49:00 +1200 |
| commit | 447b9bd6df0113ca7e2332d6ad20c01691f3ce07 (patch) | |
| tree | 637eff5c4d0923ae801e225ef0ce09a9821e2642 /src/mailman/model/mailinglist.py | |
| parent | 54ae6bb8c78998c533b75d81c3b3e72151813d9e (diff) | |
| download | mailman-447b9bd6df0113ca7e2332d6ad20c01691f3ce07.tar.gz mailman-447b9bd6df0113ca7e2332d6ad20c01691f3ce07.tar.zst mailman-447b9bd6df0113ca7e2332d6ad20c01691f3ce07.zip | |
Diffstat (limited to 'src/mailman/model/mailinglist.py')
| -rw-r--r-- | src/mailman/model/mailinglist.py | 29 |
1 files changed, 21 insertions, 8 deletions
diff --git a/src/mailman/model/mailinglist.py b/src/mailman/model/mailinglist.py index 4e76e61c1..eec58ad04 100644 --- a/src/mailman/model/mailinglist.py +++ b/src/mailman/model/mailinglist.py @@ -441,32 +441,45 @@ class MailingList(Model): else: raise ValueError('Undefined MemberRole: {}'.format(role)) - @dbconnection - def is_subscribed(self, store, subscriber, role=MemberRole.member): - """Check if a user/address is subscribed to this list.""" + def _get_subscriber(self, store, subscriber, role): + """Get some information about a user/address. + + Returns a 2-tuple of (member, email) for the given subscriber. If the + subscriber is is not an ``IAddress`` or ``IUser``, then a 2-tuple of + (None, None) is returned. If the subscriber is not already + subscribed, then (None, email) is returned. If the subscriber is an + ``IUser`` and does not have a preferred address, (member, None) is + returned. + """ member = None + email = None if IAddress.providedBy(subscriber): member = store.query(Member).filter( Member.role == role, Member.list_id == self._list_id, Member._address == subscriber).first() + email = subscriber.email elif IUser.providedBy(subscriber): if subscriber.preferred_address is None: raise MissingPreferredAddressError(subscriber) + email = subscriber.preferred_address.email member = store.query(Member).filter( Member.role == role, Member.list_id == self._list_id, Member._user == subscriber).first() + return member, email + + @dbconnection + def is_subscribed(self, store, subscriber, role=MemberRole.member): + """See `IMailingList`.""" + member, email = self._get_subscriber(store, subscriber, role) return member is not None @dbconnection def subscribe(self, store, subscriber, role=MemberRole.member): """See `IMailingList`.""" - if IAddress.providedBy(subscriber): - email = subscriber.email - elif IUser.providedBy(subscriber): - email = subscriber.preferred_address.email - if self.is_subscribed(subscriber, role): + member, email = self._get_subscriber(store, subscriber, role) + if member is not None: raise AlreadySubscribedError(self.fqdn_listname, email, role) member = Member(role=role, list_id=self._list_id, |
