X-Git-Url: https://git.p6c8.net/pcdenotes.git/blobdiff_plain/3fb7b9dea69252f64ee82967dfca97baf1b02df7..e38cf469ca320b6f1e019e02befd173b5c223056:/notes/models.py diff --git a/notes/models.py b/notes/models.py index 3584c26..65069db 100644 --- a/notes/models.py +++ b/notes/models.py @@ -8,7 +8,7 @@ NOTE_STATUS = ((0, "Draft"), (1, "Published")) class Note(models.Model): - title = models.CharField(max_length=250, unique=True) + title = models.CharField(max_length=250) slug = models.SlugField(max_length=250, unique=True) author = models.ForeignKey(User, on_delete=models.CASCADE, related_name='notes_posted') content = models.TextField() @@ -25,5 +25,9 @@ class Note(models.Model): def get_absolute_url(self): return reverse("notes:note_detail", kwargs={"note_slug": self.slug}) - #return "/notes/%s" % (self.slug) + def is_draft(self): + return self.status == 0 + + def is_published(self): + return self.status == 1