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']