Small fix

Depado · May 13, 2015 · 253fca5 · 253fca5
1 parent 870cd0f
commit 253fca5
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/app/models/post.py b/app/models/post.py
@@ -23,6 +23,9 @@ class Post(db.Model):
     def __init__(self, **kwargs):
         super(Post, self).__init__(**kwargs)
         self.pub_date = self.pub_date if self.pub_date is not None else datetime.utcnow()
+        self.set_title_slug()
+
+    def set_title_slug(self):
         self.title_slug = slugify("{date}-{title}".format(date=str(self.pub_date.date()), title=self.title))
 
     def save(self):

diff --git a/app/views/post.py b/app/views/post.py
@@ -44,6 +44,7 @@ def edit(post_id):
 
                 post.title = form.title.data
                 post.content = form.content.data
+                post.set_title_slug()
                 saved = post.save()
                 if saved:
                     flash("Successfully saved the post.")