Django ブログサイト models.py モデル

2020/07/04 （更新:2020/11/19）

モデル定義です。

投稿者

Django サイト管理機能のユーザー（スタッフ）に紐づけます。

プロパティ
is_commentコメント機能の有効・無効を返却します
Author 

class Author(models.Model):
    user = models.OneToOneField(settings.AUTH_USER_MODEL, on_delete=models.CASCADE)
    title_text = models.CharField(max_length=100)
    flags = models.IntegerField(default=0, help_text=_lazy('AUTHOR_FLAGS_HELP'))
    
    def __str__(self):
        return f'{self.user.username}:{self.title_text}'

    @property
    def is_comment(self):
        return self.flags & constants.AuthorFlag.COMMENT_FLAG.value > 0

投稿者関連共通モデル

投稿者に紐づくモデルの共通クラスです。

AplModel 

class AplModel(models.Model):
    author = models.ForeignKey(Author, on_delete=models.CASCADE)
    
    class Meta:
        abstract = True

記事属性

投稿者関連共通モデルを継承します。
statusはChoiceFieldを指定し列挙型で定義します。

プロパティ
is_commentコメント機能の有効・無効を返却します
is_comment_entryコメント投稿の可否を返却します
Post 

class PostStatus(models.IntegerChoices):
    DRAFT = 1
    PRIVATE = 2
    PUBLIC = 3
    DELETED = -1

class Post(AplModel):
    template_text = models.CharField(max_length=20)
    status = models.IntegerField(choices=PostStatus.choices, default=PostStatus.DRAFT)
    flags = models.IntegerField(default=0, help_text=_lazy('POST_FLAGS_HELP'))
    view_count = models.IntegerField(default=0)
    created_date = models.DateTimeField(default=timezone.now)
    updated_date = models.DateTimeField(null=False, blank=True, default=timezone.now, help_text=_lazy('POST_TITLE_HELP'))

    def __str__(self):
        return f'{self.template_text}'

    @property
    def is_comment(self):
        return self.author.is_comment and self.flags & constants.PostFlag.COMMENT_FLAG.value > 0
    
    def get_comment_count(self, gt_date):
        return Comment.objects.filter(post=self.id,created_date__gt=gt_date).count()
    
    def get_comment_day_count(self):
        return self.get_comment_count(timezone.now()-datetime.timedelta(days=1));
    
    @property
    def is_comment_entry(self):
        count = self.get_comment_day_count()
        return count <= 10
    
    def get_title_text(self):
        return PostContent.objects.get(post=self.id, language_code=settings.LANGUAGE_CODE).title_text

記事内容

投稿者関連共通モデルを継承します。
language_codeはChoiceFieldを指定し列挙型で定義します。

PostContent 

class LanguageCode(models.TextChoices):
    EN = 'en'
    JA = 'ja'

class PostContent(models.Model):
    post = models.ForeignKey(Post, on_delete=models.CASCADE)
    language_code = models.CharField(choices=LanguageCode.choices, default=LanguageCode.JA, max_length=2)
    title_text = models.CharField(max_length=100, null=False, blank=True, help_text=_lazy('POST_TITLE_HELP'))
    summary_text = models.CharField(max_length=1000, null=True, blank=True, help_text=_lazy('POST_TITLE_HELP'))
    search_text = models.CharField(max_length=5000, null=False, blank=True, help_text=_lazy('POST_TITLE_HELP'))

カテゴリ

投稿者関連共通モデルを継承します。

Category 

class Category(AplModel):
    category_text = models.CharField(max_length=20)
    order_number = models.IntegerField()

    def __str__(self):
        return self.category_text

記事カテゴリ関連

PostCategory 

class PostCategory(models.Model):
    post = models.ForeignKey(Post, on_delete=models.CASCADE)
    category = models.ForeignKey(Category, on_delete=models.CASCADE)

    def __str__(self):
        return f'{self.category}:{self.post}'

コメント

statusはChoiceFieldを指定し列挙型で定義します。
選択肢毎にラベルを設定します。

Comment 

from django.utils.translation import ugettext as _

class CommentStatus(models.IntegerChoices):
    UNAPPROVED = 1, 'Unapproved'
    APPROVED = 2, 'Approved'
    DELETED = -1, 'Deleted'

    def get_choices():
        return [(c.value, _(c.label)) for c in CommentStatus]

    def __str__(self):
        return _(self.label)

class Comment(models.Model):
    name_text = models.CharField(max_length=100, null=False, blank=True)
    comment_text = models.TextField(max_length=1000)
    status = models.IntegerField(choices=CommentStatus.choices, default=CommentStatus.UNAPPROVED)
    client_text = models.CharField(max_length=20)
    post = models.ForeignKey(Post, on_delete=models.CASCADE)
    parent = models.ForeignKey('self', null=True, blank=True, on_delete=models.CASCADE)
    created_date = models.DateTimeField()

    def save(self, *args, **kwargs):
        if not self.status:
            self.status = CommentStatus.UNAPPROVED
        self.created_date = timezone.now()
        super().save(args, kwargs)

    def __str__(self):
        return f'{str(self.post.id)}:{self.comment_text[:10]}'