1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
|
from django.db import models
from django.utils.translation import ugettext_lazy as _
from django.db.models import permalink
from django.contrib.auth.models import User
from tagging.fields import TagField
from news.managers import PublicManager
from django.urls import reverse
import datetime
import settings
import tagging
def get_upload_name(inst, fn):
try:
extension = fn.split('.')[-1].lower()
except:
extension = 'png'
return 'news/img/%s.%s' % (inst.title, extension)
class Category(models.Model):
"""Category model."""
title = models.CharField(_('title'), max_length=100)
slug = models.SlugField(_('slug'), unique=True)
image = models.ImageField(upload_to=get_upload_name, max_length=100)
class Meta:
verbose_name = _('category')
verbose_name_plural = _('categories')
db_table = 'news_categories'
ordering = ('title',)
def __unicode__(self):
return u'%s' % self.title
def get_absolute_url(self):
return reverse('category_posts', args=(self.slug,))
class Post(models.Model):
"""Post model."""
STATUS_CHOICES = (
(1, _('Draft')),
(2, _('Public')),
)
title = models.CharField(_('title'), max_length=200)
slug = models.SlugField(_('slug'), unique_for_date='publish')
author = models.ForeignKey(User, null=True)
body = models.TextField(
_('body'), help_text='Text entered here will be rendered using Markdown')
tease = models.TextField(_('tease'), blank=True)
status = models.IntegerField(
_('status'), choices=STATUS_CHOICES, default=2)
allow_comments = models.BooleanField(_('allow comments'), default=True)
publish = models.DateTimeField(_('publish'))
created = models.DateTimeField(_('created'), auto_now_add=True)
modified = models.DateTimeField(_('modified'), auto_now=True)
categories = models.ManyToManyField(Category, blank=True)
tags = TagField()
objects = PublicManager()
class Meta:
verbose_name = _('post')
verbose_name_plural = _('posts')
db_table = 'news_posts'
ordering = ('-publish',)
get_latest_by = 'publish'
def __unicode__(self):
return u'%s' % self.title
#########
# IMAGE #
#########
# Currently this is only inherited from the category, but one
# day we might want to override the post image here
@property
def has_image(self):
if self.categories.count() == 0:
return False
return self.categories.all()[0].image != ''
@property
def image(self):
if self.categories.count() == 0:
return None
return self.categories.all()[0].image
@property
def image_alt(self):
"alt='' tag for <img>"
if self.categories.count() == 0:
return ''
return self.categories.all()[0].title
def get_absolute_url(self):
return reverse('news_detail', args=(self.publish.year, self.publish.strftime('%b'), self.publish.day, self.slug, ))
def get_category_slug(self):
try:
s = self.categories.all()[0].slug
except IndexError:
return 'none'
return s
def get_previous_post(self):
# get_previous_by_FOO(**kwargs) is a django model function
return self.get_previous_by_publish(status__gte=2)
def get_next_post(self):
# get_next_by_FOO(**kwargs) is a django model function
return self.get_next_by_publish(status__gte=2, publish__lte=datetime.datetime.now())
|