diff options
Diffstat (limited to 'routes/heroes.py')
| -rw-r--r-- | routes/heroes.py | 7 |
1 files changed, 4 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, |