From 1370240f86f4e63b72b9b5f767d7405e2385481f Mon Sep 17 00:00:00 2001
From: Laurent <laurent@staxriver.com>
Date: Thu, 9 Apr 2020 16:22:15 +0200
Subject: [PATCH] Capitalize words for the Post title

---
 news/models.py                 | 4 ++++
 news/templates/news/index.html | 2 +-
 news/templates/news/post.html  | 2 +-
 3 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/news/models.py b/news/models.py
index 2c20162..4b2b513 100644
--- a/news/models.py
+++ b/news/models.py
@@ -4,6 +4,7 @@ from django.conf import settings
 from django.db.models import Q
 from django.utils.text import slugify
 from enum import Enum
+import string
 
 # Create your models here.
 class PostState(Enum):
@@ -61,6 +62,9 @@ class Post(models.Model):
             return self.title
         return 'Poker Rumble - Amazing content'
 
+    def formatted_title(self):
+        return string.capwords(self.title)
+
 class Comment(models.Model):
     author = models.ForeignKey(settings.AUTH_USER_MODEL, on_delete=models.CASCADE, null=True)
     post = models.ForeignKey(Post, on_delete=models.CASCADE)
diff --git a/news/templates/news/index.html b/news/templates/news/index.html
index 0e5afbf..5501efa 100644
--- a/news/templates/news/index.html
+++ b/news/templates/news/index.html
@@ -31,7 +31,7 @@
             <img src="{% static 'media/' %}{{ post.image_url }}"/>
             <div class="inside_image_top">
               {% if post.url %}<a href="{{ post.url }}">{% endif %}
-                <h1><span>{{ post.title }}</span></h1>
+                <h1><span>{{ post.formatted_title }}</span></h1>
               {% if post.url %}</a>{% endif %}
 
               {% if post.body %}
diff --git a/news/templates/news/post.html b/news/templates/news/post.html
index b151e76..22b9c61 100644
--- a/news/templates/news/post.html
+++ b/news/templates/news/post.html
@@ -20,7 +20,7 @@
 <div class="quote">
 {% endif %}
 {% if post.style != 2 and post.style != 3 %}
-  <a href="{{ post.url }}"><h1>{{ post.title }}</h1></a>
+  <a href="{{ post.url }}"><h1>{{ post.formatted_title }}</h1></a>
 {% endif %}
 {% if post.style == 1 %}
 </div>