Added comments

api/app/blueprints/keyword.py

@@ -31,9 +31,10 @@ def add_keyword(data):
 
     key = data['keyword']
 
+    # Check if keyword already exists
     check_keyword = db.session.query(Keyword).filter_by(keyword=key, email=email).first()
     if check_keyword is None:
-        # Keyword doesn't exist yet for this user
+        # Keyword doesn't exist yet for this user -> add it
         new_keyword = Keyword(
             email=email,
             keyword=key
@@ -54,17 +55,17 @@ def add_keyword(data):
 def remove_keyword(data):
     email = get_email_or_abort_401()
 
+    # Check if request data is valid
     if not check_if_keyword_data_exists(data):
         abort(400, message="Keyword missing")
 
-    key = data['keyword']
-
-    check_keyword = db.session.query(Keyword).filter_by(keyword=key, email=email).first()
-
+    # Check if keyword exists
+    check_keyword = db.session.query(Keyword).filter_by(keyword=data['keyword'], email=email).first()
     if check_keyword is None:
         return abort(500, "Keyword doesn't exist for this user")
     else:
-        db.session.query(Keyword).filter_by(keyword=key, email=email).delete()
+        # Keyword exists -> delete it
+        db.session.query(Keyword).filter_by(keyword=data['keyword'], email=email).delete()
         db.session.commit()
 
         return make_response({}, 200, "Successfully removed keyword")
@@ -80,6 +81,8 @@ def get_keywords():
     return_keywords = []
     keywords = db.session.query(Keyword).filter_by(email=email).all()
 
+    # If no keywords exist for this user -> return empty list
+    # Otherwise iterate over all keywords, convert them to json and add them to the return list
     if keywords is not None:
         for row in keywords:
             return_keywords.append(row.as_dict())

api/app/blueprints/portfolio.py

@@ -25,8 +25,12 @@ def get_portfolio():
     email = get_email_or_abort_401()
 
     return_portfolio = []
+
+    # Get all transactions of current user
     transactions = db.session.execute("SELECT symbol, SUM(count), SUM(price), MAX(time) FROM `transactions` WHERE email = '" + email + "' GROUP BY symbol;").all()
 
+    # If there are no transactions, return empty portfolio
+    # Otherwise calculate portfolio
     if transactions is not None:
         for row in transactions:
             data = {
@@ -37,6 +41,7 @@ def get_portfolio():
                 'current_price': 0
             }
 
+            # Add current share value to portfolio
             query_share_price = db.session.query(SharePrice).filter_by(symbol=row[0]).order_by(SharePrice.date.desc()).first()
             if query_share_price is not None:
                 data['current_price'] = query_share_price.as_dict()['price']

api/app/blueprints/share_price.py

@@ -24,6 +24,7 @@ __location__ = os.path.realpath(os.path.join(os.getcwd(), os.path.dirname(__file
 @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():
+    # Get all transaction symbols
     symbols = db.session.execute("SELECT symbol FROM `transactions` GROUP BY symbol;").all()
 
     return_symbols = []
@@ -39,6 +40,7 @@ def get_transaction_symbols():
 @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):
+    # Check if required data is available
     if not check_if_symbol_data_exists(data):
         abort(400, message="Symbol missing")
 
@@ -48,14 +50,11 @@ def add_symbol_price(data):
     if not check_if_time_data_exists(data):
         abort(400, message="Time missing")
 
-    symbol = data['symbol']
-    price = data['price']
-    time = data['time']
-
+    # Add share price
     share_price = SharePrice(
-        symbol=symbol,
-        price=price,
-        date=datetime.datetime.strptime(time, '%Y-%m-%dT%H:%M:%S.%fZ'),
+        symbol=data['symbol'],
+        price=data['price'],
+        date=datetime.datetime.strptime(data['time'], '%Y-%m-%dT%H:%M:%S.%fZ'),
     )
 
     db.session.add(share_price)

api/app/blueprints/shares.py

@@ -26,17 +26,17 @@ __location__ = os.path.realpath(os.path.join(os.getcwd(), os.path.dirname(__file
 def add_symbol(data):
     email = get_email_or_abort_401()
 
+    # Check if required data is available
     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()
+    # Check if share already exists
+    check_share = db.session.query(Share).filter_by(symbol=data['symbol'], email=email).first()
     if check_share is None:
-        # Keyword doesn't exist yet for this user
+        # Keyword doesn't exist yet for this user -> add it
         new_symbol = Share(
             email=email,
-            symbol=symbol
+            symbol=data['symbol']
         )
         db.session.add(new_symbol)
         db.session.commit()
@@ -54,17 +54,17 @@ def add_symbol(data):
 def remove_symbol(data):
     email = get_email_or_abort_401()
 
+    # Check if required data is available
     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()
-
+    # Check if share exists
+    check_share = db.session.query(Share).filter_by(symbol=data['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()
+        # Delete share
+        db.session.query(Share).filter_by(symbol=data['symbol'], email=email).delete()
         db.session.commit()
 
         return make_response({}, 200, "Successfully removed symbol")
@@ -80,6 +80,8 @@ def get_symbol():
     return_symbols = []
     symbols = db.session.query(Share).filter_by(email=email).all()
 
+    # If no shares exist for this user -> return empty list
+    # Otherwise iterate over all shares, convert them to json and add them to the return list
     if symbols is not None:
         for row in symbols:
             return_symbols.append(row.as_dict())

api/app/blueprints/telegram.py

@@ -24,11 +24,13 @@ __location__ = os.path.realpath(os.path.join(os.getcwd(), os.path.dirname(__file
 def add_keyword(data):
     email = get_email_or_abort_401()
 
+    # Check if request data is valid
     if not check_if_telegram_user_id_data_exists(data):
         abort(400, message="User ID missing")
 
     query_user = get_user(email)
 
+    # Change user id
     query_user.telegram_user_id = data['telegram_user_id']
     db.session.commit()
 

api/app/blueprints/transactions.py

@@ -27,6 +27,7 @@ __location__ = os.path.realpath(os.path.join(os.getcwd(), os.path.dirname(__file
 def add_transaction(data):
     email = get_email_or_abort_401()
 
+    # Check if required data is available
     if not check_if_symbol_data_exists(data):
         abort(400, "Symbol missing")
 
@@ -39,6 +40,7 @@ def add_transaction(data):
     if not check_if_price_data_exists(data):
         abort(400, "Price missing")
 
+    # Add transaction
     new_transaction = Transaction(
         email=email,
         symbol=data['symbol'],
@@ -60,8 +62,11 @@ def get_transaction():
     email = get_email_or_abort_401()
 
     return_transactions = []
+
+    # Get all transactions
     transactions = db.session.query(Transaction).filter_by(email=email).all()
 
+    # Iterate over transactions and add them to return_transactions
     if transactions is not None:
         for row in transactions:
             return_transactions.append(row.as_dict())

api/app/blueprints/user.py

@@ -28,6 +28,8 @@ def users():
     abort_if_no_admin()
 
     res = []
+
+    # Query all users and convert them to dicts
     for i in User.query.all():
         res.append(i.as_dict())
 
@@ -41,6 +43,7 @@ def users():
 def user():
     email = get_email_or_abort_401()
 
+    # Query current user
     query_user = get_user(email)
 
     return make_response(query_user.as_dict(), 200, "Successfully received current user data")
@@ -51,23 +54,26 @@ def user():
 @users_blueprint.input(schema=LoginDataSchema)
 @users_blueprint.doc(summary="Login", description="Returns jwt token if username and password match, otherwise returns error")
 def login(data):
+    # Check if required data is available
     if not check_if_password_data_exists(data):
         abort(400, "Password missing")
 
     if not check_if_email_data_exists(data):
         abort(400, "Email missing")
 
-    email = data['email']
-    password = data['password']
+    # Query current user
+    query_user = get_user(data['email'])
 
-    query_user = get_user(email)
-
-    if not check_password(query_user.password, password.encode("utf-8")):  # Password incorrect
+    # Check if password matches
+    if not check_password(query_user.password, data['password'].encode("utf-8")):  # Password incorrect
         abort(500, message="Unable to login")
 
+    # Check if user is bot
     if query_user.email == current_app.config['BOT_EMAIL']:
+        # Set bot token valid for 1 year
         token = jwt.encode({'email': query_user.email, 'exp': datetime.datetime.utcnow() + datetime.timedelta(days=365)}, current_app.config['SECRET_KEY'], "HS256")
     else:
+        # Set token valid for 1 day
         token = jwt.encode({'email': query_user.email, 'exp': datetime.datetime.utcnow() + datetime.timedelta(days=1)}, current_app.config['SECRET_KEY'], "HS256")
 
     return make_response({"token": token}, 200, "Successfully logged in")
@@ -78,6 +84,7 @@ def login(data):
 @users_blueprint.input(schema=RegisterDataSchema)
 @users_blueprint.doc(summary="Register", description="Registers user")
 def register(data):
+    # Check if required data is available
     if not check_if_email_data_exists(data):
         abort(400, "Email missing")
 
@@ -87,19 +94,16 @@ def register(data):
     if not check_if_password_data_exists(data):
         abort(400, "Password missing")
 
-    email = data['email']
-    username = data['username']
-    password = data['password']
-
-    query_user = db.session.query(User).filter_by(email=email).first()
-
-    if query_user is not None:  # Username already exist
+    # Check if user already exists
+    query_user = db.session.query(User).filter_by(email=data['email']).first()
+    if query_user is not None:
         abort(500, message="Email already exist")
 
+    # Add user to database
     new_user = User(
-        email=email,
-        username=username,
-        password=hash_password(password),
+        email=data['email'],
+        username=data['username'],
+        password=hash_password(data['password']),
         admin=False,
         cron="0 8 * * *"
     )
@@ -117,11 +121,14 @@ def register(data):
 def update_user(data):
     email = get_email_or_abort_401()
 
+    # Query current user
     query_user = get_user(email)
 
+    # Check if password data is available -> if, change password
     if check_if_password_data_exists(data):
         query_user.password = hash_password(data['password'])
 
+    # Check if username data is available -> if, change username
     if check_if_username_data_exists(data):
         query_user.username = data['username']
 
@@ -138,18 +145,18 @@ def update_user(data):
 def set_admin(data):
     abort_if_no_admin()  # Only admin users can do this
 
+    # Check if required data is available
     if not check_if_email_data_exists(data):
         abort(400, "Email missing")
 
     if not check_if_admin_data_exists(data):
         abort(400, "Admin data missing")
 
-    email = data['email']
-    admin = data['admin']
+    # Get user by email
+    query_user = get_user(data['email'])
 
-    query_user = get_user(email)
-
-    query_user.admin = admin
+    # Update user admin state
+    query_user.admin = data['admin']
     db.session.commit()
 
     return make_response({}, 200, "Successfully updated users admin rights")
@@ -163,9 +170,11 @@ def set_admin(data):
 def set_cron(data):
     email = get_email_or_abort_401()
 
+    # Check if required data is available
     if not check_if_cron_data_exists(data):
         abort(400, "Cron data missing")
 
+    # Update user cron
     get_user(email).cron = data['cron']
     db.session.commit()
 
@@ -178,18 +187,22 @@ def set_cron(data):
 @users_blueprint.auth_required(auth)
 @users_blueprint.doc(summary="Delete user", description="Deletes user by username")
 def delete_user(data):
+    # Check if required data is available
     if not check_if_email_data_exists(data):
         abort(400, "Email missing")
 
-    email = data['email']
-
-    if email == get_email_or_abort_401():  # Username is same as current user
-        db.session.query(User).filter_by(email=email).delete()
+    # Check if email to delete is current user
+    # -> if, delete user
+    # -> if not, check if user is admin
+    # -> if, delete user
+    # -> else, abort
+    if data['email'] == get_email_or_abort_401():  # Username is same as current user
+        db.session.query(User).filter_by(email=data['email']).delete()
         db.session.commit()
-    else:  # Delete different user than my user -> only admin users
+    else:
         abort_if_no_admin()
 
-        db.session.query(User).filter_by(email=email).delete()
+        db.session.query(User).filter_by(email=data['email']).delete()
         db.session.commit()
 
     return make_response({}, 200, "Successfully removed user")