Profanity check

apis/users_api.py

@@ -19,6 +19,7 @@
 from pymysql.cursors import DictCursor
 
 from wikispeedruns.auth import passwords
+from wikispeedruns.profanity_check import profanity_check_func
 
 user_api = Blueprint("users", __name__, url_prefix="/api/users")
 
@@ -144,6 +145,9 @@ def create_user():
     # Validate username
     if (not _valid_username(username)):
         return "Invalid username", 400
+    # profanity check
+    if profanity_check_func(username):
+        return ("Please use clean language in username", 406)
 
     # Use SHA256 to allow for arbitrary length passwords
     hash = passwords.hash_password(password)
@@ -306,6 +310,9 @@ def change_username():
 
     id = session["user_id"]
     new_username = request.json["new_username"]
+    # profanity check
+    if profanity_check_func(new_username):
+        return ("Please use clean language in username", 406)
 
     db = get_db()
     with db.cursor(cursor=DictCursor) as cursor:

frontend/static/js/pages/account.js

@@ -82,6 +82,9 @@ var app = new Vue({
                 if (response.status === 200) {
                     window.location.reload();
                 }
+                else if (response.status === 406) {
+                    alert(await response.text());
+                }
                 this.feedbackMsg = await response.text()
             } catch (e) {
                 console.log(e);

requirements.txt

@@ -5,6 +5,7 @@ bcrypt>=3.2.0
 Flask-Dance==7.0.1
 itsdangerous==2.1.2
 pytest==6.2.5
+alt-profanity-check==1.3.2
 
 celery[redis]
 

wikispeedruns/profanity_check.py

@@ -0,0 +1,12 @@
+from profanity_check import predict
+
+
+def profanity_check_func(word):
+    '''
+    Profanity check for a word
+    '''
+    profanity = predict([word])
+    if profanity[0]:
+        return 1
+    else:
+        return 0