Added portfolio endpoint
This commit is contained in:
parent
d03cd1d2f8
commit
859935cc01
@ -334,6 +334,29 @@
|
|||||||
"response": []
|
"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": {
|
"auth": {
|
||||||
@ -341,7 +364,7 @@
|
|||||||
"bearer": [
|
"bearer": [
|
||||||
{
|
{
|
||||||
"key": "token",
|
"key": "token",
|
||||||
"value": "access_token",
|
"value": "eyJ0eXAiOiJKV1QiLCJhbGciOiJIUzI1NiJ9.eyJ1c2VybmFtZSI6IlVzZXJuYW1lIiwiZXhwIjoxNjQ3Mjk5MzIxfQ.5UoHi9Zu6p9szSVKK2_1Ln2uru4RVTGQl0MyHDB4sqg",
|
||||||
"type": "string"
|
"type": "string"
|
||||||
}
|
}
|
||||||
]
|
]
|
||||||
|
31
webservice/api_blueprint_portfolio.py
Normal file
31
webservice/api_blueprint_portfolio.py
Normal 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})
|
@ -7,6 +7,7 @@ from api_blueprint_keyword import keyword_blueprint
|
|||||||
from api_blueprint_shares import shares_blueprint
|
from api_blueprint_shares import shares_blueprint
|
||||||
from api_blueprint_user import users_blueprint
|
from api_blueprint_user import users_blueprint
|
||||||
from api_blueprint_transactions import transaction_blueprint
|
from api_blueprint_transactions import transaction_blueprint
|
||||||
|
from webservice.api_blueprint_portfolio import portfolio_blueprint
|
||||||
|
|
||||||
|
|
||||||
def create_app():
|
def create_app():
|
||||||
@ -27,6 +28,7 @@ def create_app():
|
|||||||
application.register_blueprint(keyword_blueprint)
|
application.register_blueprint(keyword_blueprint)
|
||||||
application.register_blueprint(shares_blueprint)
|
application.register_blueprint(shares_blueprint)
|
||||||
application.register_blueprint(transaction_blueprint)
|
application.register_blueprint(transaction_blueprint)
|
||||||
|
application.register_blueprint(portfolio_blueprint)
|
||||||
application.register_blueprint(users_blueprint)
|
application.register_blueprint(users_blueprint)
|
||||||
|
|
||||||
# interface blueprint
|
# interface blueprint
|
||||||
|
Loading…
Reference in New Issue
Block a user