Capitalize each Word in a String in Heroe name

2 files changed, 29 insertions, 3 deletions

diff --git a/routes/heroes.py b/routes/heroes.py
index 9b631a5..ab1b0ba 100644
--- a/routes/heroes.py
+++ b/routes/heroes.py
@@ -36,6 +36,7 @@ from utils.errors import (
     not_data_found
 )
 from utils.decorators import superuser
+from utils.strings import capitalize_each
 from bson.objectid import ObjectId
 
 heroes = Blueprint("heroes", __name__)
@@ -86,7 +87,7 @@ def create_heroe():
             url=request.json['image']['url']
         )
         heroe = HeroeModel(
-            name=request.json['name'].capitalize(),
+            name=capitalize_each(request.json['name']),
             powerstats=powerstats,
             biography=biography,
             appearance=appearance,
@@ -318,7 +319,7 @@ def get_image_heroe(id):
 def get_some_heroe_name(name):
     """Search heroe by name"""
     try:
-        name = name.capitalize()
+        name = capitalize_each(name)
         heroe = HeroeModel.objects.filter(name=name)
         if len(heroe) > 0:
             data = heroe.exclude('added_by').to_json()
@@ -407,7 +408,7 @@ def put_heroe(id):
             url=request.json['image']['url']
         )
         heroe.update(
-            name=request.json['name'].capitalize(),
+            name=capitalize_each(request.json['name']),
             powerstats=powerstats,
             biography=biography,
             appearance=appearance,
diff --git a/utils/strings.py b/utils/strings.py
new file mode 100644
index 0000000..e28912b
--- /dev/null
+++ b/utils/strings.py
@@ -0,0 +1,25 @@
+"""
+Capitalize each Word in a String
+"""
+
+
+def capitalize_each(a):
+    textLen = len(a)
+    for i in range(textLen):
+        ch = a[i]
+        if i == 0:
+            ascVal = ord(ch)
+            if ascVal >= 97 and ascVal <= 122:
+                ascVal = ascVal-32
+                ascVal = chr(ascVal)
+                index = 0
+                a = a[:index] + ascVal + a[index+1:]
+        if ch == " ":
+            ch = a[i+1]
+            ascVal = ord(ch)
+            if ascVal >= 97 and ascVal <= 122:
+                ascVal = ascVal-32
+                ascVal = chr(ascVal)
+                index = i+1
+                a = a[:index] + ascVal + a[index+1:]
+    return a