Rewrite api from Flask to APIFlask

webservice/api_blueprint_portfolio.py

@@ -1,21 +1,23 @@
 import os
 
-from flask import Blueprint, jsonify
+from apiflask import APIBlueprint
+from flask import jsonify
 
 from db import db
-from helper_functions import get_username_from_token_data, extract_token_data, get_token, get_user_id_from_username, return_401
+from helper_functions import get_user_id_from_username, get_username_or_abort_401
 from models import Transaction
+from auth import auth
 
-portfolio_blueprint = Blueprint('portfolio', __name__, url_prefix='/api')
+portfolio_blueprint = APIBlueprint('portfolio', __name__, url_prefix='/api')
 __location__ = os.path.realpath(os.path.join(os.getcwd(), os.path.dirname(__file__)))
 
 
 @portfolio_blueprint.route('/portfolio', methods=['GET'])
+@portfolio_blueprint.output(200)
+@portfolio_blueprint.auth_required(auth)
+@portfolio_blueprint.doc(summary="Returns portfolio", description="Returns all shares of current user")
 def get_portfolio():
-    # get username from jwt token
-    username = get_username_from_token_data(extract_token_data(get_token()))
-    if username is None:  # If token not provided or invalid -> return 401 code
-        return return_401()
+    username = get_username_or_abort_401()
 
     return_portfolio = {}
     transactions = db.session.query(Transaction).filter_by(user_id=get_user_id_from_username(username)).all()