def test_subscriptiontype(self):
eq_(SubscriptionType.by_name(u'__test_comments'), CommentsSubscriptionType)
eq_(SubscriptionType.by_object_type(TestSubscriptionComment), [CommentsSubscriptionType])
eq_(SubscriptionType.by_subject_type(TestSubscriptionCategory), [CategorySubscriptionType])
eq_(sorted(SubscriptionType.by_action(NewEntrySubscriptionAction)),
sorted([CategorySubscriptionType, BlogSubscriptionType, TagSubscriptionType]))
eq_(SubscriptionType.by_action(u'__test_new_comment'), [CommentsSubscriptionType])
def test_subscriptiontype(self):
eq_(SubscriptionType.by_name(u'__test_comments'),
CommentsSubscriptionType)
eq_(SubscriptionType.by_object_type(TestSubscriptionComment),
[CommentsSubscriptionType])
eq_(SubscriptionType.by_subject_type(TestSubscriptionCategory),
[CategorySubscriptionType])
eq_(
sorted(SubscriptionType.by_action(NewEntrySubscriptionAction)),
sorted([
CategorySubscriptionType, BlogSubscriptionType,
TagSubscriptionType
]))
eq_(SubscriptionType.by_action(u'__test_new_comment'),
[CommentsSubscriptionType])
def new(object, action):
"""
Sends notifications for a new object, saves it as `first_unread_object`
or in `unread_objects` (depending on the type's mode) and increments
the unread count.
Must be called whenever a new object is created or moved to the scope
of a SubscriptionType (e.g. when a topic is moved to another forum).
:param object: The added object.
:param action: The action which lead to the creation of the object.
It is used to select the SubscriptionTypes that need to
be considered and the action's :meth:`notify` method is
called.
"""
#XXX: if `object` is removed before the task is run, subscriptions are
# marked as notified though the user didn't receive a notification
if isinstance(action, basestring):
action = SubscriptionAction.by_name(action)
#: store ids of Subscription objects for which we need to notify
subscriptions = []
for t in SubscriptionType.by_action(action):
subjects = t.get_subjects(object)
if not subjects:
continue
subject_ids = [getattr(s, 'id', s) for s in subjects]
base_cond = (Subscription.type_name == t.name)
if t.subject_type is not None:
base_cond &= Subscription.subject_id.in_(subject_ids)
if t.mode == 'sequent':
#: increment unread count where there already are unread objects
cond = base_cond & (Subscription.count > 0)
q = db.update(Subscription.__table__, cond,
{'count': Subscription.count + 1})
db.session.execute(q)
db.session.commit()
#: then set the first unread object and notify the user
#: where there were no new objects since the last visit
cond = base_cond & (Subscription.count == 0)
for s in Subscription.query.filter(cond):
subscriptions.append(s.id)
s.first_unread_object_id = object.id
s.count = 1
db.session.commit()
if t.mode == 'multiple':
for s in Subscription.query.filter(base_cond):
subscriptions.append(s.id)
s.unread_object_ids.add(object.id)
s.count = len(s.unread_object_ids)
db.session.commit()
from inyoka.core import tasks
tasks.send_notifications(object, action.name, subscriptions)
