|
| 1 | +from django.db import models |
| 2 | +from django.conf import settings |
| 3 | +from django.urls.base import reverse |
| 4 | + |
| 5 | +from modules.utils import generate_upload_path, create_slug |
| 6 | + |
| 7 | + |
| 8 | +class BaseAppModel(models.Model): |
| 9 | + """This is the base model for other models. This will provide fields for other models. Hence abstract = |
| 10 | + True""" |
| 11 | + |
| 12 | + name = models.CharField('name', max_length=255, db_index=True, null=True, blank=True, |
| 13 | + help_text='this field will be used as name for models inheriting from this model') |
| 14 | + created_on = models.DateTimeField('created on', auto_now_add=True, auto_now=False) |
| 15 | + updated_on = models.DateTimeField('updated on', auto_now_add=False, auto_now=True) |
| 16 | + |
| 17 | + class Meta: |
| 18 | + abstract = True |
| 19 | + |
| 20 | + def __str__(self): |
| 21 | + return '%s-%s' % (self.id, self.name) |
| 22 | + |
| 23 | + |
| 24 | +class ArticleCategory(BaseAppModel): |
| 25 | + |
| 26 | + slug = models.SlugField('slug', max_length=50, blank=True, db_index=True) |
| 27 | + photo = models.ImageField('photo', null=True, blank=True) |
| 28 | + created_by = models.ForeignKey(settings.AUTH_USER_MODEL, default=1, db_index=True, on_delete=models.CASCADE) |
| 29 | + |
| 30 | + class Meta: |
| 31 | + verbose_name = 'Category' |
| 32 | + verbose_name_plural = 'Categories' |
| 33 | + db_table = 'article_category' |
| 34 | + unique_together = ('name', 'slug',) |
| 35 | + |
| 36 | + def __str__(self): |
| 37 | + return '%s - %s' % (self.id, self.name) |
| 38 | + |
| 39 | + @property |
| 40 | + def get_absolute_url(self): |
| 41 | + return reverse('blogs:category', kwargs={'slug': self.slug}) |
| 42 | + |
| 43 | + def save(self, force_insert=False, force_update=False, using=None, update_fields=None): |
| 44 | + self.slug = create_slug(self.name) |
| 45 | + super().save(force_insert=False, force_update=False, using=None, |
| 46 | + update_fields=None) |
| 47 | + |
| 48 | + |
| 49 | +class Article(BaseAppModel): |
| 50 | + |
| 51 | + category = models.ManyToManyField(ArticleCategory, related_name='category', db_index=True, |
| 52 | + help_text="Categories mentioned in the article.") |
| 53 | + |
| 54 | + slug = models.CharField('slug', max_length=500, db_index=True, unique=True, |
| 55 | + help_text="slug of an article. It should be unique.") |
| 56 | + |
| 57 | + blog = models.TextField(null=True, blank=True) |
| 58 | + intro = models.TextField('intro', max_length=500, help_text='Brief of your article in 50-60 words.') |
| 59 | + |
| 60 | + wallpaper = models.ImageField('wallpaper', max_length=300, null=True, blank=True, upload_to=generate_upload_path, |
| 61 | + help_text="wallpaper for the article. This image and its thumbnail " |
| 62 | + "will be used everywhere.") |
| 63 | + |
| 64 | + published_on = models.DateTimeField('published on', null=True, blank=True, |
| 65 | + help_text="date when the article was published.") |
| 66 | + published_by = models.ForeignKey(settings.AUTH_USER_MODEL, related_name="user", db_index=True, |
| 67 | + help_text="User who published the article.", on_delete=models.CASCADE) |
| 68 | + |
| 69 | + published = models.BooleanField('published', default=False, db_index=True, |
| 70 | + help_text="If an article is published or not. " |
| 71 | + "By default an article will be considered as draft.") |
| 72 | + |
| 73 | + class Meta: |
| 74 | + verbose_name_plural = "Blog" |
| 75 | + verbose_name = "Blogs" |
| 76 | + db_table = "articles" |
| 77 | + ordering = ['-id'] |
| 78 | + |
| 79 | + def __str__(self): |
| 80 | + return '%s-%s' % (self.id, self.name) |
| 81 | + |
| 82 | + @property |
| 83 | + def get_absolute_url(self): |
| 84 | + return reverse('blogs:blog', kwargs={'slug': self.slug}) |
| 85 | + |
| 86 | + |
| 87 | +# Comments model |
| 88 | +class Comment(BaseAppModel): |
| 89 | + STATUS = ( |
| 90 | + (1, 'Allowed'), |
| 91 | + (2, 'Deleted'), |
| 92 | + (3, 'Spam'), |
| 93 | + ) |
| 94 | + comment = models.TextField('comment') |
| 95 | + # image = models.FileField(null=True, blank=True, upload_to=generate_comment_upload_path, |
| 96 | + # help_text="image for the comment") |
| 97 | + |
| 98 | + article = models.ForeignKey(Article, db_index=True, related_name="comment", on_delete=models.CASCADE) |
| 99 | + comment_by = models.ForeignKey(settings.AUTH_USER_MODEL, db_index=True, related_name='comment_by', |
| 100 | + on_delete=models.CASCADE) |
| 101 | + |
| 102 | + status = models.PositiveIntegerField('status', choices=STATUS, default=1) |
| 103 | + is_edited = models.BooleanField('is_edited', default=False) |
| 104 | + |
| 105 | + class Meta: |
| 106 | + verbose_name_plural = "Comments" |
| 107 | + verbose_name = "Comment" |
| 108 | + db_table = "comment" |
| 109 | + ordering = ['-created_on'] |
| 110 | + |
| 111 | + def __str__(self): |
| 112 | + return '%s' % self.comment[0:30] |
0 commit comments