Added portfolio endpoint

This commit is contained in:
Administrator 2022-03-14 23:40:36 +01:00
parent d03cd1d2f8
commit 859935cc01
3 changed files with 57 additions and 1 deletions

View File

@ -334,6 +334,29 @@
"response": []
}
]
},
{
"name": "Portfolio",
"item": [
{
"name": "/api/portfolio",
"request": {
"method": "GET",
"header": [],
"url": {
"raw": "{{BASE_URL}}/api/portfolio",
"host": [
"{{BASE_URL}}"
],
"path": [
"api",
"portfolio"
]
}
},
"response": []
}
]
}
],
"auth": {
@ -341,7 +364,7 @@
"bearer": [
{
"key": "token",
"value": "access_token",
"value": "eyJ0eXAiOiJKV1QiLCJhbGciOiJIUzI1NiJ9.eyJ1c2VybmFtZSI6IlVzZXJuYW1lIiwiZXhwIjoxNjQ3Mjk5MzIxfQ.5UoHi9Zu6p9szSVKK2_1Ln2uru4RVTGQl0MyHDB4sqg",
"type": "string"
}
]

View File

@ -0,0 +1,31 @@
import os
from flask import Blueprint, 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 models import Transaction
portfolio_blueprint = Blueprint('portfolio', __name__, url_prefix='/api')
__location__ = os.path.realpath(os.path.join(os.getcwd(), os.path.dirname(__file__)))
@portfolio_blueprint.route('/portfolio', methods=['GET'])
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()
return_portfolio = {}
transactions = db.session.query(Transaction).filter_by(user_id=get_user_id_from_username(username)).all()
if transactions is not None:
for row in transactions:
if row.symbol in return_portfolio:
return_portfolio[row.symbol]['count'] += row.count
return_portfolio[row.symbol]['last_transaction'] += row.time
else:
return_portfolio[row.symbol] = {"count": row.count, "last_transaction": row.time}
return jsonify({"status": 200, "text": "Successfully loaded symbols", "data": return_portfolio})

View File

@ -7,6 +7,7 @@ from api_blueprint_keyword import keyword_blueprint
from api_blueprint_shares import shares_blueprint
from api_blueprint_user import users_blueprint
from api_blueprint_transactions import transaction_blueprint
from webservice.api_blueprint_portfolio import portfolio_blueprint
def create_app():
@ -27,6 +28,7 @@ def create_app():
application.register_blueprint(keyword_blueprint)
application.register_blueprint(shares_blueprint)
application.register_blueprint(transaction_blueprint)
application.register_blueprint(portfolio_blueprint)
application.register_blueprint(users_blueprint)
# interface blueprint