-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathmodels.py
64 lines (53 loc) · 2.03 KB
/
models.py
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
import markdown
from django.db import models
from django.contrib.auth.models import User
from django.urls import reverse
from django.utils.html import strip_tags
# Model must inherit models.Model
# Aricle Category
class Category(models.Model):
name = models.CharField(max_length=100)
objects = models.Manager()
def __str__(self):
return self.name
# Aricle Tag
class Tag(models.Model):
name = models.CharField(max_length=100)
objects = models.Manager()
def __str__(self):
return self.name
# Aricle published
class Article(models.Model):
title = models.CharField(max_length=100)
body = models.TextField()
created_time = models.DateTimeField()
modified_time = models.DateTimeField()
summary = models.CharField(max_length=200, blank=True)
views = models.PositiveIntegerField(default=0)
# Relationships with other classes
category = models.ForeignKey(Category,on_delete=models.PROTECT)
tags = models.ManyToManyField(Tag,blank=True)
author = models.ForeignKey(User,on_delete=models.PROTECT)
objects = models.Manager()
def __str__(self):
return self.title
def get_absolute_url(self):
return reverse('blog:detail', args=[str(self.pk)])
def add_views(self):
self.views +=1
self.save(update_fields=['views'])
#override save() of Model for generating article summary
def save(self, *args, **kwargs):
if not self.summary:
# instantiate a markdown class
md = markdown.Markdown(extensions=[
'markdown.extensions.extra',
'markdown.extensions.codehilite',
])
# md converts article body into HTML
# stripe_tags clean up all the HTML tags
self.summary = strip_tags(md.convert(self.body))[:54]
# invoke save() in its parent class to save it into database
super(Article,self).save(*args, **kwargs)
class Meta:
ordering = ['-created_time']