89 lines
2.5 KiB
Python
89 lines
2.5 KiB
Python
|
import datetime
|
||
|
|
import os
|
||
|
|
|
||
|
|
from apiflask import APIBlueprint, abort
|
||
|
|
|
||
|
|
from app.models import SharePrice
|
||
|
|
from app.db import database as db
|
||
|
|
from app.helper_functions import make_response
|
||
|
|
from app.auth import auth
|
||
|
|
from app.schema import SymbolPriceSchema
|
||
|
|
|
||
|
|
share_price_blueprint = APIBlueprint('share_price', __name__, url_prefix='/api')
|
||
|
|
__location__ = os.path.realpath(os.path.join(os.getcwd(), os.path.dirname(__file__)))
|
||
|
|
|
||
|
|
|
||
|
|
@share_price_blueprint.route('/symbols', methods=['GET'])
|
||
|
|
@share_price_blueprint.output({}, 200)
|
||
|
|
@share_price_blueprint.auth_required(auth)
|
||
|
|
@share_price_blueprint.doc(summary="Returns all transaction symbols", description="Returns all transaction symbols for all users")
|
||
|
|
def get_transaction_symbols():
|
||
|
|
symbols = db.session.execute("SELECT symbol FROM `transactions` GROUP BY symbol;").all()
|
||
|
|
|
||
|
|
return_symbols = []
|
||
|
|
for s in symbols:
|
||
|
|
return_symbols.append(s[0])
|
||
|
|
|
||
|
|
return make_response(return_symbols, 200, "Successfully loaded symbols")
|
||
|
|
|
||
|
|
|
||
|
|
@share_price_blueprint.route('/symbol', methods=['POST'])
|
||
|
|
@share_price_blueprint.output({}, 200)
|
||
|
|
@share_price_blueprint.input(schema=SymbolPriceSchema)
|
||
|
|
@share_price_blueprint.auth_required(auth)
|
||
|
|
@share_price_blueprint.doc(summary="Returns all transaction symbols", description="Returns all transaction symbols for all users")
|
||
|
|
def add_symbol_price(data):
|
||
|
|
if not check_if_symbol_data_exists(data):
|
||
|
|
abort(400, message="Symbol missing")
|
||
|
|
|
||
|
|
if not check_if_price_data_exists(data):
|
||
|
|
abort(400, message="Price missing")
|
||
|
|
|
||
|
|
if not check_if_time_data_exists(data):
|
||
|
|
abort(400, message="Time missing")
|
||
|
|
|
||
|
|
symbol = data['symbol']
|
||
|
|
price = data['price']
|
||
|
|
time = data['time']
|
||
|
|
|
||
|
|
share_price = SharePrice(
|
||
|
|
symbol=symbol,
|
||
|
|
price=price,
|
||
|
|
date=datetime.datetime.strptime(time, '%Y-%m-%dT%H:%M:%S.%fZ'),
|
||
|
|
)
|
||
|
|
|
||
|
|
db.session.add(share_price)
|
||
|
|
db.session.commit()
|
||
|
|
|
||
|
|
return make_response(share_price.as_dict(), 200, "Successfully added price")
|
||
|
|
|
||
|
|
|
||
|
|
def check_if_symbol_data_exists(data):
|
||
|
|
if 'symbol' not in data:
|
||
|
|
return False
|
||
|
|
|
||
|
|
if data['symbol'] == "" or data['symbol'] is None:
|
||
|
|
return False
|
||
|
|
|
||
|
|
return True
|
||
|
|
|
||
|
|
|
||
|
|
def check_if_price_data_exists(data):
|
||
|
|
if 'price' not in data:
|
||
|
|
return False
|
||
|
|
|
||
|
|
if data['price'] == "" or data['price'] is None:
|
||
|
|
return False
|
||
|
|
|
||
|
|
return True
|
||
|
|
|
||
|
|
|
||
|
|
def check_if_time_data_exists(data):
|
||
|
|
if 'time' not in data:
|
||
|
|
return False
|
||
|
|
|
||
|
|
if data['time'] == "" or data['time'] is None:
|
||
|
|
return False
|
||
|
|
|
||
|
|
return True
|