Activity stream representations for user and favorites

This commit is contained in:
Eliot Berriot 2018-02-25 14:44:00 +01:00
parent 93e4a4f123
commit 691665e3cf
No known key found for this signature in database
GPG key ID: DD6965E2476E5C27
12 changed files with 159 additions and 0 deletions

View file

@ -1,8 +1,10 @@
from django.conf import settings
from django.db import models
from django.utils import timezone
from funkwhale_api.music.models import Track
class TrackFavorite(models.Model):
creation_date = models.DateTimeField(default=timezone.now)
user = models.ForeignKey(
@ -18,3 +20,7 @@ class TrackFavorite(models.Model):
def add(cls, track, user):
favorite, created = cls.objects.get_or_create(user=user, track=track)
return favorite
def get_activity_url(self):
return '{}/favorites/tracks/{}'.format(
self.user.get_activity_url(), self.pk)