TelegramAktienBot/api/app/blueprints/shares.py

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