Tests

- Improved directory structure
- Added functional and unit tests

api/app/blueprints/shares.py

@@ -0,0 +1,91 @@
+import os
+
+from apiflask import APIBlueprint, abort
+
+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 Share
+from app.schema import SymbolSchema, SymbolResponseSchema, DeleteSuccessfulSchema
+
+shares_blueprint = APIBlueprint('share', __name__, url_prefix='/api')
+__location__ = os.path.realpath(os.path.join(os.getcwd(), os.path.dirname(__file__)))
+
+
+@shares_blueprint.route('/share', methods=['POST'])
+@shares_blueprint.output(SymbolResponseSchema(many=True), 200)
+@shares_blueprint.input(schema=SymbolSchema)
+@shares_blueprint.auth_required(auth)
+@shares_blueprint.doc(summary="Add new symbol", description="Adds new symbol for current user")
+def add_symbol(data):
+    email = get_email_or_abort_401()
+
+    if not check_if_symbol_data_exists(data):
+        abort(400, message="Symbol missing")
+
+    symbol = data['symbol']
+
+    check_share = db.session.query(Share).filter_by(symbol=symbol, email=email).first()
+    if check_share is None:
+        # Keyword doesn't exist yet for this user
+        new_symbol = Share(
+            email=email,
+            symbol=symbol
+        )
+        db.session.add(new_symbol)
+        db.session.commit()
+
+        return make_response(new_symbol.as_dict(), 200, "Successfully added symbol")
+    else:
+        abort(500, "Symbol already exist for this user")
+
+
+@shares_blueprint.route('/share', methods=['DELETE'])
+@shares_blueprint.output(DeleteSuccessfulSchema, 200)
+@shares_blueprint.input(schema=SymbolSchema)
+@shares_blueprint.auth_required(auth)
+@shares_blueprint.doc(summary="Removes existing symbol", description="Removes existing symbol for current user")
+def remove_symbol(data):
+    email = get_email_or_abort_401()
+
+    if not check_if_symbol_data_exists(data):
+        abort(400, message="Symbol missing")
+
+    symbol = data['symbol']
+
+    check_share = db.session.query(Share).filter_by(symbol=symbol, email=email).first()
+
+    if check_share is None:
+        abort(500, "Symbol doesn't exist for this user")
+    else:
+        db.session.query(Share).filter_by(symbol=symbol, email=email).delete()
+        db.session.commit()
+
+        return make_response({}, 200, "Successfully removed symbol")
+
+
+@shares_blueprint.route('/shares', methods=['GET'])
+@shares_blueprint.output(SymbolResponseSchema(many=True), 200)
+@shares_blueprint.auth_required(auth)
+@shares_blueprint.doc(summary="Returns all symbols", description="Returns all symbols for current user")
+def get_symbol():
+    email = get_email_or_abort_401()
+
+    return_symbols = []
+    symbols = db.session.query(Share).filter_by(email=email).all()
+
+    if symbols is not None:
+        for row in symbols:
+            return_symbols.append(row.as_dict())
+
+    return make_response(return_symbols, 200, "Successfully loaded symbols")
+
+
+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