55 lines
1.9 KiB
Python
55 lines
1.9 KiB
Python
import os
|
|
import datetime
|
|
|
|
from flask import Blueprint, jsonify, request
|
|
|
|
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
|
|
|
|
transaction_blueprint = Blueprint('transaction', __name__, url_prefix='/api')
|
|
__location__ = os.path.realpath(os.path.join(os.getcwd(), os.path.dirname(__file__)))
|
|
|
|
|
|
@transaction_blueprint.route('/transaction', methods=['POST'])
|
|
def add_transaction():
|
|
# 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()
|
|
|
|
request_data = request.get_json()
|
|
symbol = request_data['symbol']
|
|
time = datetime.datetime.strptime(request_data['time'], '%Y-%m-%dT%H:%M:%S.%fZ')
|
|
count = request_data['count']
|
|
price = request_data['price']
|
|
|
|
new_transcation = Transaction(
|
|
user_id=get_user_id_from_username(username),
|
|
symbol=symbol,
|
|
time=time,
|
|
count=count,
|
|
price=price
|
|
)
|
|
db.session.add(new_transcation)
|
|
db.session.commit()
|
|
|
|
return jsonify({"status": 200, "text": "Successfully added transaction", "data": new_transcation.as_dict()})
|
|
|
|
|
|
@transaction_blueprint.route('/transactions', methods=['GET'])
|
|
def get_transaction():
|
|
# 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_transactions = []
|
|
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:
|
|
return_transactions.append(row.as_dict())
|
|
|
|
return jsonify({"status": 200, "text": "Successfully loaded transactions", "data": return_transactions})
|