lemoncurry/entries/models.py

from django.contrib.auth import get_user_model
from django.db import models
from django.urls import reverse
from slugify import slugify

from . import kinds
ENTRY_KINDS = [(k.id, k.__name__) for k in kinds.all]


class Entry(models.Model):
    kind = models.CharField(
        max_length=30,
        choices=ENTRY_KINDS,
        default=ENTRY_KINDS[0][0]
    )

    name = models.CharField(max_length=100, blank=True)
    summary = models.TextField(blank=True)
    content = models.TextField()

    author = models.ForeignKey(get_user_model(), on_delete=models.CASCADE)

    published = models.DateTimeField()
    updated = models.DateTimeField()

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