Skip to content

HTTPS clone URL

Subversion checkout URL

You can clone with
or
.
Download ZIP
Tree: 6126fc34a9
Fetching contributors…

Cannot retrieve contributors at this time

135 lines (107 sloc) 4.681 kB
from django import template
from django.conf import settings
from django.db import models
from django.db.models.signals import post_save
from django.utils.translation import ugettext as _
from stream.registry import actor_map, target_map, action_object_map
from stream.signals import action
DEFAULT_STREAM_VERBS = (
('default', _('Stream Item')),
)
STREAM_VERBS = getattr(settings, 'STREAM_VERBS', DEFAULT_STREAM_VERBS)
class ActionManager(models.Manager):
def create(self, actor, verb, target=None, action_object=None, **kwargs):
""" Create a new action object """
action = Action(actor=actor, verb=verb, target=target,
action_object=action_object, **kwargs)
action.save()
return action
def get_or_create(self, actor, verb, target=None, action_object=None, **kwargs):
result = self.filter(verb=verb) & self.get_for_actor(actor)
if target:
result = result & self.get_for_target(target)
if action_object:
result = result & self.get_for_action_object(action_object)
result = result.filter(**kwargs)
if result.count() == 0:
return self.create(actor, verb, target, action_object, **kwargs), True
if result.count() == 1:
return result[0], False
raise Action.MultipleObjectsReturned
def _multi_lookup(self, objects, method):
result = method(objects[0])
for obj in objects[1:]:
result = result | method(obj)
return result
def get_for_actor(self, actor):
""" Returns all objects involving `actor` """
_, f_name = actor_map[actor.__class__]
return self.filter(**{f_name:actor})
def get_for_actors(self, actors):
"""
Similar to `get_for_actor` - but does lookups for multiple actors.
"""
return self._multi_lookup(actors, self.get_for_actor)
def get_for_target(self, target):
""" Returns all objects involving `target` """
_, f_name = target_map[target.__class__]
return self.filter(**{f_name: target})
def get_for_targets(self, targets):
"""
Similar to `get_for_target` - but does lookups for multiple targets.
"""
return self._multi_lookup(targets, self.get_for_target)
def get_for_action_object(self, obj):
""" Returns all objects involving `obj` """
_, f_name = action_object_map[obj.__class__]
return self.filter(**{f_name: obj})
def get_for_action_objects(self, objects):
"""
Similar to `get_for_action_object` - but does lookups for multiple
action objects.
"""
return self._multi_lookup(objects, self.get_for_action_object)
class Action(models.Model):
"""
Base class for all actions. This model is dynamically extended through the
register function in `stream.util`
All the actor, target and action object fields are implemented as indexed
foreign keys to avoid having generic relations. They're really uncool.
"""
datetime = models.DateTimeField(auto_now_add=True)
description = models.TextField(blank=True, null=True)
verb = models.CharField(choices=STREAM_VERBS, max_length=255, db_index=True)
objects = ActionManager()
def __unicode__(self):
return u'%s: %s - %s' % (self.verb, self.actor, self.target)
def _get(self, map):
for _, fname in map.values():
if getattr(self, fname):
return getattr(self, fname)
return None
def _set(self, obj, map):
for _, fname in map.values():
setattr(self, fname, None)
if obj is None:
return
_, fname = map[obj.__class__]
setattr(self, fname, obj)
def _get_actor(self):
return self._get(actor_map)
def _set_actor(self, obj):
self._set(obj, actor_map)
def _get_target(self):
return self._get(target_map)
def _set_target(self, obj):
self._set(obj, target_map)
def _get_action_object(self):
return self._get(action_object_map)
def _set_action_object(self, obj):
return self._set(obj, action_object_map)
actor = property(fget=_get_actor, fset=_set_actor)
target = property(fget=_get_target, fset=_set_target)
action_object = property(fget=_get_action_object, fset=_set_action_object)
def action_dispatch(sender, instance, created=False, **kwargs):
if created:
action.send(sender, instance=instance)
post_save.connect(action_dispatch, dispatch_uid='stream.action_dispatch', sender=Action)
Jump to Line
Something went wrong with that request. Please try again.