Coverage for message/models.py: 100.00%

1from django.contrib.auth import get_user_model (empty)

2from django.db import models (empty)

3from django.template.defaultfilters import slugify (empty)

4from django.urls import reverse (empty)

5from django.utils import timezone (empty)

6 

7 

8class PublishedManager(models.Manager): (empty)

9 def get_queryset(self): (empty)

10 return super().get_queryset().filter(status=Message.Status.PUBLISHED) 3 ctx1bfc

11 

12 

13""" (empty)

14class DraftManager(models.Manager): 

15 def get_queryset(self): 

16 return super().get_queryset().filter(status=Message.Status.DRAFT) 

17""" 

18 

19 

20class Message(models.Model): (empty)

21 class Status(models.TextChoices): (empty)

22 DRAFT = "DF", "Draft" (empty)

23 PUBLISHED = "PB", "Published" (empty)

24 

25 title = models.CharField(max_length=250) (empty)

26 slug = models.SlugField( (empty)

27 max_length=250, 

28 null=False, 

29 unique_for_date="publish", 

30 ) 

31 author = models.ForeignKey( (empty)

32 get_user_model(), 

33 on_delete=models.CASCADE, 

34 related_name="message_posts", 

35 ) 

36 body = models.TextField() (empty)

37 publish = models.DateTimeField(default=timezone.now) (empty)

38 created = models.DateTimeField(auto_now_add=True) (empty)

39 updated = models.DateTimeField(auto_now=True) (empty)

40 status = models.CharField( (empty)

41 max_length=2, 

42 choices=Status.choices, 

43 default=Status.PUBLISHED, 

44 ) 

45 

46 objects = models.Manager() (empty)

47 # draft = DraftManager() 

48 published = PublishedManager() (empty)

49 

50 class Meta: (empty)

51 ordering = ["-publish"] (empty)

52 indexes = [ (empty)

53 models.Index(fields=["-publish"]), 

54 ] 

55 

56 def __str__(self): (empty)

57 return self.title 1 ctx1d

58 

59 def save(self, *args, **kwargs): (empty)

60 if not self.slug: 5 ctx1aedbc

61 self.slug = slugify(self.title) 1 ctx1b

62 return super().save(*args, **kwargs) 5 ctx1aedbc

63 

64 def get_absolute_url(self): (empty)

65 return reverse( 1 ctx1e

66 "message_detail", 

67 args=[self.publish.year, self.publish.month, self.publish.day, self.slug], 

68 )