lemoncurry/entries/models.py

50 lines
1.2 KiB
Python
Raw Normal View History

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