Add slug to Game model

Pipfile

@@ -23,6 +23,7 @@ python-dotenv = "*"
 flask-migrate = "*"
 gunicorn = "*"
 requests = "*"
+python-slugify = "*"
 
 [requires]
 python_version = "3.8"

api/v1/games.py

@@ -3,7 +3,7 @@
 from flask_restx import Namespace, Resource, Api, fields
 from extensions import guard, db, upload_img, delete_img
 from marshmallow import ValidationError, INCLUDE
-from utilities import get_auto_increment, base64_to_pillow_img, pillow_img_to_bytes
+from utilities import get_auto_increment, base64_to_pillow_img, pillow_img_to_bytes, slugify_text
 from sqlalchemy.exc import SQLAlchemyError
 from sqlalchemy import or_, and_
 
@@ -99,7 +99,8 @@ def post(self):
             # Set Authenticated User
             current_user = flask_praetorian.current_user()
             req['user'] = user_schema.dump(current_user)
-
+            req['slug'] = slugify_text(req['title'])
+
             # Validate        
             try:
                 new_game = game_schema.load(req)
@@ -153,7 +154,7 @@ def get(self, id):
             db.session.commit()
         except SQLAlchemyError:
             return { 'error': 'Unable to update view count' }, 500
-
+        
         return game_schema.dump(game)
 
     @flask_praetorian.roles_required('admin')
@@ -171,6 +172,7 @@ def put(self, id):
 
             icon_name = game.icon
             banner_name = game.banner
+            req['slug'] = slugify_text(req['title'])
 
             # Validate        
             try:

api/v1/schemas.py

@@ -136,11 +136,11 @@ class GameSchema(ma.ModelSchema):
     class Meta:
         model = Game
         sqla_session = db.session
-        fields = ("id", "title", "synopsis", "icon", "banner", "developer", "genres", "releases", "score", "library_count", "score_rank", "popularity_rank", "trending_page_views", "user")
+        fields = ("id", "title", "slug", "synopsis", "icon", "banner", "developer", "genres", "releases", "score", "library_count", "score_rank", "popularity_rank", "trending_page_views", "user")
 
 game_schema = GameSchema(exclude=['user'])
 games_schema = GameSchema(many=True, exclude=['releases'])
-nested_game_schema = GameSchema(only=("id", "title", "synopsis", "icon", "banner", "developer"))
+nested_game_schema = GameSchema(only=("id", "title", "slug", "synopsis", "icon", "banner", "developer"))
 
 # LibraryEntrySchema
 class LibraryEntrySchema(ma.ModelSchema):

models/game.py

@@ -15,6 +15,7 @@ class Game(db.Model):
 
     id = db.Column(db.Integer, primary_key=True)
     title = db.Column(db.String, unique=True, nullable=False)
+    slug = db.Column(db.String, unique=True, nullable=False)
     synopsis = db.Column(db.Text, nullable=False)
     icon = db.Column(db.String, unique=True, nullable=False)
     banner = db.Column(db.String, unique=True, nullable=False)

utilities.py

@@ -3,6 +3,7 @@
 from PIL import Image
 from io import BytesIO
 import base64
+from slugify import slugify
 
 def get_auto_increment(table_name):
     """
@@ -91,4 +92,7 @@ def base64_validation(base64_string, max_file_size):
     if file_type != 'jpeg' and file_type != 'png':
         return False
 
-    return True
+    return True
+
+def slugify_text(text):
+    return slugify(text, replacements=([['\'', '']]))