Save symbol price in database

api/app/blueprints/share_price.py

@@ -0,0 +1,88 @@
+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

api/app/blueprints/transactions.py

@@ -1,13 +1,13 @@
-import os
 import datetime
+import os
 
 from apiflask import abort, APIBlueprint
 
+from app.auth import auth
 from app.db import database as db
 from app.helper_functions import make_response, get_email_or_abort_401
 from app.models import Transaction
 from app.schema import TransactionSchema, TransactionResponseSchema
-from app.auth import auth
 
 transaction_blueprint = APIBlueprint('transaction', __name__, url_prefix='/api')
 __location__ = os.path.realpath(os.path.join(os.getcwd(), os.path.dirname(__file__)))