X-Git-Url: https://git.p6c8.net/pcdenotes.git/blobdiff_plain/3fb7b9dea69252f64ee82967dfca97baf1b02df7..5603ef1d534888f3e85b93c98fb2623d4154fead:/notes/models.py diff --git a/notes/models.py b/notes/models.py index 3584c26..ed86192 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() @@ -27,3 +27,8 @@ class Note(models.Model): 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