notes/models.py

   1 from django.db import models
   2 from django.db.models import Count
   3 from django.db.models.functions import ExtractYear, ExtractMonth
   4 from django.db.models.signals import pre_save
   5
   6 from django.dispatch import receiver
   7 from django.contrib.auth.models import User
   8 from django.urls import reverse
   9
  10 from datetime import datetime
  11
  12 # Create your models here.
  13
  14 NOTE_STATUS = ((0, "Draft"),
  15                (1, "Published"))
  16
  17 class NoteQuerySet(models.QuerySet):
  18     pass
  19
  20 class NoteManager(models.Manager):
  21     def per_year(self, year):
  22         return super().get_queryset().filter(status=1, created_at__year=year)
  23
  24     def per_month(self, year, month):
  25         return super().get_queryset().filter(status=1, created_at__year=year, created_at__month=month)
  26
  27     def years(self):
  28         return super().get_queryset().filter(status=1).annotate(created_year=ExtractYear('created_at')).values_list('created_year', flat=True).distinct().order_by('created_year')
  29
  30     def years_with_total(self):
  31         return super().get_queryset().filter(status=1).annotate(created_year=ExtractYear('created_at')).values('created_year').annotate(total=Count('id')).order_by('created_year').values('created_year', 'total').distinct()
  32
  33     def months(self, year):
  34         return self.per_year(year).annotate(created_month=ExtractMonth('created_at')).values_list('created_month', flat=True).distinct().order_by('created_month')
  35
  36     def months_with_total(self, year):
  37         return self.per_year(year).annotate(created_month=ExtractMonth('created_at')).values('created_month').annotate(total=Count('id')).order_by('created_month').values('created_month', 'total').distinct()
  38
  39 class Note(models.Model):
  40     title = models.CharField(max_length=250)
  41     slug = models.SlugField(max_length=250, unique=True)
  42     author = models.ForeignKey(User, on_delete=models.CASCADE, related_name='notes_posted')
  43     content = models.TextField()
  44     status = models.IntegerField(choices=NOTE_STATUS, default=0)
  45
  46     created_at = models.DateTimeField(auto_now_add=True)
  47     updated_at = models.DateTimeField(auto_now=True)
  48     published_at = models.DateTimeField(null=True, editable=False)
  49
  50     objects = NoteManager()
  51
  52     class Meta:
  53         ordering = ['-created_at']
  54
  55     def __str__(self):
  56         return self.title
  57
  58     def get_absolute_url(self):
  59         return reverse("notes:note_detail", kwargs={"note_slug": self.slug})
  60
  61     def is_draft(self):
  62         return self.status == 0
  63
  64     def is_published(self):
  65         return self.status == 1
  66
  67     def publication_date(self):
  68         if self.published_at is None:
  69             return self.created_at
  70         else:
  71             return self.published_at
  72
  73 @receiver(pre_save, sender=Note)
  74 def note_pre_save(sender, instance, **kwargs):
  75     if instance.pk is None:
  76         if instance.status == 1:
  77             instance.published_at = datetime.now()
  78     else:
  79         old_instance = Note.objects.get(pk=instance.pk)
  80
  81         if old_instance.status == 0 and instance.status == 1:
  82             instance.published_at = datetime.now()