152 lines
4.8 KiB
Python
152 lines
4.8 KiB
Python
"""
|
|
Models for Social app
|
|
|
|
DataSignalManager captures create/update/(soft/hard)delete/add/remove from Journal app, and generate Activity objects,
|
|
ActivityManager generates chronological view for user and, in future, ActivityStreams
|
|
|
|
"""
|
|
|
|
from django.db import models
|
|
from users.models import User
|
|
from catalog.common.models import Item
|
|
from journal.models import *
|
|
import logging
|
|
from functools import cached_property
|
|
from django.db.models.signals import post_save, post_delete, pre_delete
|
|
from django.db.models import Q
|
|
from django.conf import settings
|
|
|
|
|
|
_logger = logging.getLogger(__name__)
|
|
|
|
|
|
class ActivityTemplate(models.TextChoices):
|
|
"""
|
|
"""
|
|
MarkItem = 'mark_item'
|
|
ReviewItem = 'review_item'
|
|
CreateCollection = 'create_collection'
|
|
LikeCollection = 'like_collection'
|
|
|
|
|
|
class LocalActivity(models.Model, UserOwnedObjectMixin):
|
|
owner = models.ForeignKey(User, on_delete=models.CASCADE)
|
|
visibility = models.PositiveSmallIntegerField(default=0) # 0: Public / 1: Follower only / 2: Self only
|
|
template = models.CharField(blank=False, choices=ActivityTemplate.choices, max_length=50)
|
|
action_object = models.ForeignKey(Piece, on_delete=models.CASCADE)
|
|
created_time = models.DateTimeField(default=timezone.now, db_index=True)
|
|
|
|
def __str__(self):
|
|
return f'Activity [{self.owner}:{self.template}:{self.action_object}]'
|
|
|
|
|
|
class ActivityManager:
|
|
def __init__(self, user):
|
|
self.owner = user
|
|
|
|
def get_timeline(self, before_time=None):
|
|
q = Q(owner_id__in=self.owner.following, visibility__lt=2) | Q(owner=self.owner)
|
|
if before_time:
|
|
q = q & Q(created_time__lt=before_time)
|
|
return LocalActivity.objects.filter(q).order_by('-created_time') # .select_related() https://github.com/django-polymorphic/django-polymorphic/pull/531
|
|
|
|
@staticmethod
|
|
def get_manager_for_user(user):
|
|
return ActivityManager(user)
|
|
|
|
|
|
User.activity_manager = cached_property(ActivityManager.get_manager_for_user)
|
|
User.activity_manager.__set_name__(User, 'activity_manager')
|
|
|
|
|
|
class DataSignalManager:
|
|
processors = {}
|
|
|
|
@staticmethod
|
|
def save_handler(sender, instance, created, **kwargs):
|
|
processor_class = DataSignalManager.processors.get(instance.__class__)
|
|
if processor_class:
|
|
processor = processor_class(instance)
|
|
if created:
|
|
if hasattr(processor, 'created'):
|
|
processor.created()
|
|
elif hasattr(processor, 'updated'):
|
|
processor.updated()
|
|
|
|
@staticmethod
|
|
def delete_handler(sender, instance, **kwargs):
|
|
processor_class = DataSignalManager.processors.get(instance.__class__)
|
|
if processor_class:
|
|
processor = processor_class(instance)
|
|
if hasattr(processor, 'deleted'):
|
|
processor.deleted()
|
|
|
|
@staticmethod
|
|
def add_handler_for_model(model):
|
|
if not settings.DISABLE_SOCIAL:
|
|
post_save.connect(DataSignalManager.save_handler, sender=model)
|
|
pre_delete.connect(DataSignalManager.delete_handler, sender=model)
|
|
|
|
@staticmethod
|
|
def register(processor):
|
|
DataSignalManager.add_handler_for_model(processor.model)
|
|
DataSignalManager.processors[processor.model] = processor
|
|
return processor
|
|
|
|
|
|
class DefaultActivityProcessor:
|
|
model = None
|
|
template = None
|
|
|
|
def __init__(self, action_object):
|
|
self.action_object = action_object
|
|
|
|
def created(self):
|
|
params = {
|
|
'owner': self.action_object.owner,
|
|
'visibility': self.action_object.visibility,
|
|
'template': self.template,
|
|
'action_object': self.action_object,
|
|
}
|
|
print(params)
|
|
LocalActivity.objects.create(**params)
|
|
|
|
def updated(self):
|
|
activity = LocalActivity.objects.filter(action_object=self.action_object).first()
|
|
if not activity:
|
|
self.created()
|
|
elif activity.visibility != self.action_object.visibility:
|
|
activity.visibility = self.action_object.visibility
|
|
activity.save()
|
|
|
|
|
|
@DataSignalManager.register
|
|
class MarkProcessor(DefaultActivityProcessor):
|
|
model = ShelfMember
|
|
template = ActivityTemplate.MarkItem
|
|
|
|
|
|
@DataSignalManager.register
|
|
class ReviewProcessor(DefaultActivityProcessor):
|
|
model = Review
|
|
template = ActivityTemplate.ReviewItem
|
|
|
|
|
|
@DataSignalManager.register
|
|
class CollectionProcessor(DefaultActivityProcessor):
|
|
model = Collection
|
|
template = ActivityTemplate.CreateCollection
|
|
|
|
|
|
@DataSignalManager.register
|
|
class LikeCollectionProcessor(DefaultActivityProcessor):
|
|
model = Like
|
|
template = ActivityTemplate.LikeCollection
|
|
|
|
def created(self):
|
|
if isinstance(self.action_object.target, Collection):
|
|
super().created()
|
|
|
|
def updated(self):
|
|
if isinstance(self.action_object.target, Collection):
|
|
super().update()
|