forked from 00dani/lemoncurry
102 lines
2.5 KiB
Python
102 lines
2.5 KiB
Python
from django.contrib.auth import get_user_model
|
|
from django.db import models
|
|
from django.urls import reverse
|
|
from slugify import slugify
|
|
|
|
from meta.models import ModelMeta
|
|
from users.models import Profile
|
|
|
|
from . import kinds
|
|
ENTRY_KINDS = [(k.id, k.id) for k in kinds.all]
|
|
|
|
|
|
class EntryManager(models.Manager):
|
|
def get_queryset(self):
|
|
qs = super(EntryManager, self).get_queryset()
|
|
return qs.select_related('author').prefetch_related('syndications')
|
|
|
|
|
|
class Entry(ModelMeta, models.Model):
|
|
objects = EntryManager()
|
|
kind = models.CharField(
|
|
max_length=30,
|
|
choices=ENTRY_KINDS,
|
|
db_index=True,
|
|
default=ENTRY_KINDS[0][0]
|
|
)
|
|
|
|
name = models.CharField(max_length=100, blank=True)
|
|
content = models.TextField()
|
|
|
|
author = models.ForeignKey(
|
|
get_user_model(),
|
|
related_name='entries',
|
|
on_delete=models.CASCADE,
|
|
)
|
|
|
|
published = models.DateTimeField()
|
|
updated = models.DateTimeField()
|
|
|
|
_metadata = {
|
|
'description': 'content',
|
|
'image': 'image_url',
|
|
'twitter_creator': 'twitter_creator',
|
|
'og_profile_id': 'og_profile_id',
|
|
}
|
|
|
|
@property
|
|
def twitter_creator(self):
|
|
return self.author.twitter_username
|
|
|
|
@property
|
|
def og_profile_id(self):
|
|
return self.author.facebook_id
|
|
|
|
@property
|
|
def image_url(self):
|
|
return self.author.avatar_url
|
|
|
|
def __str__(self):
|
|
return '{kind} {id}: {content}'.format(
|
|
kind=self.kind,
|
|
id=self.id,
|
|
content=self.content
|
|
)
|
|
|
|
@property
|
|
def url(self):
|
|
kind = kinds.from_id[self.kind]
|
|
route = 'entries:{kind}_entry'.format(kind=kind.plural)
|
|
args = [self.id]
|
|
if kind.has('slug'):
|
|
route += '_slug'
|
|
args.append(self.slug)
|
|
return reverse(route, args=args)
|
|
|
|
@property
|
|
def slug(self):
|
|
return slugify(self.name)
|
|
|
|
class Meta:
|
|
verbose_name_plural = 'entries'
|
|
ordering = ['-published']
|
|
|
|
|
|
class SyndicationManager(models.Manager):
|
|
def get_queryset(self):
|
|
qs = super(SyndicationManager, self).get_queryset()
|
|
return qs.select_related('profile__site')
|
|
|
|
|
|
class Syndication(models.Model):
|
|
objects = SyndicationManager()
|
|
entry = models.ForeignKey(
|
|
Entry,
|
|
related_name='syndications',
|
|
on_delete=models.CASCADE
|
|
)
|
|
profile = models.ForeignKey(Profile, on_delete=models.CASCADE)
|
|
url = models.CharField(max_length=255)
|
|
|
|
class Meta:
|
|
ordering = ['profile']
|