Update autocomplete API to use marshmallow

Author: recrsn

Pipfile

@@ -21,6 +21,7 @@ markupsafe = "*"
 flask-sqlalchemy = "*"
 flask-marshmallow = "*"
 marshmallow-sqlalchemy = "*"
+marshmallow = "*"
 
 [requires]
 python_version = "3.8"

Pipfile.lock

@@ -1,9 +1,17 @@
 from flask import jsonify
 
 
-def bad_request(message):
-    return 400, jsonify({'message': message})
+def error_body(message):
+    return jsonify({'message': message})
 
 
-def not_found():
-    return 404, jsonify({'message': "Not found"})
+def bad_request(message="Bad request"):
+    return error_body(message), 400
+
+
+def not_found(message="Not found"):
+    return error_body(message), 404
+
+
+def unauthorized(message="Unauthorized"):
+    return jsonify(message), 403

snowflake/controllers/api/response.py

@@ -1,22 +1,21 @@
-from flask import Blueprint, request, jsonify
+from flask import Blueprint, request
 
-from snowflake.models import User
 from .response import bad_request
+from ...models import User
+from ...schemas.user import UserSchema
 
 blueprint = Blueprint('users', __name__)
 
+schema = UserSchema()
+
 
 @blueprint.route('/_autocomplete')
-def autocomplete_search():
+def autocomplete():
     term = request.args.get('q')
 
     if term is None:
         return bad_request("missing search term 'q'")
 
     users = User.find_by_name_prefix(term)
 
-    return jsonify([{
-        'name': user.name,
-        'username': user.username,
-        'avatar': user.profile_pic
-    } for user in users])
+    return schema.jsonify(users, many=True)