DHBW/TelegramAktienBot
1
0
Fork 0
Code Issues 6 Pull Requests 26 Actions Packages Projects Releases Wiki Activity

Check if token is empty or invalid, update postman

Browse Source
This commit is contained in:
Administrator
2022-03-14 22:57:03 +01:00
parent 4cda34e2c8
commit d03cd1d2f8
5 changed files with 52 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
webservice/api_blueprint_keyword.py
View File

@@ -3,7 +3,7 @@ import os
from flask import Blueprint, jsonify, request
from db import db
from helper_functions import get_username_from_token_data, extract_token_data, get_token, get_user_id_from_username
from helper_functions import get_username_from_token_data, extract_token_data, get_token, get_user_id_from_username, return_401
from models import Keyword
keyword_blueprint = Blueprint('keyword', __name__, url_prefix='/api')
@@ -12,11 +12,13 @@ __location__ = os.path.realpath(os.path.join(os.getcwd(), os.path.dirname(__file
@keyword_blueprint.route('/keyword', methods=['POST'])
def add_keyword():
request_data = request.get_json()
key = request_data['keyword']
# get username from jwt token
username = get_username_from_token_data(extract_token_data(get_token()))
if username is None: # If token not provided or invalid -> return 401 code
return return_401()
request_data = request.get_json()
key = request_data['keyword']
check_keyword = db.session.query(Keyword).filter_by(keyword=key, user_id=get_user_id_from_username(username)).first()
if check_keyword is None:
@@ -35,11 +37,13 @@ def add_keyword():
@keyword_blueprint.route('/keyword', methods=['DELETE'])
def remove_keyword():
request_data = request.get_json()
key = request_data['keyword']
# get username from jwt token
username = get_username_from_token_data(extract_token_data(get_token()))
if username is None: # If token not provided or invalid -> return 401 code
return return_401()
request_data = request.get_json()
key = request_data['keyword']
db.session.query(Keyword).filter_by(keyword=key, user_id=get_user_id_from_username(username)).delete()
db.session.commit()
@@ -51,6 +55,8 @@ def remove_keyword():
def get_keywords():
# get username from jwt token
username = get_username_from_token_data(extract_token_data(get_token()))
if username is None: # If token not provided or invalid -> return 401 code
return return_401()
return_keywords = []
keywords = db.session.query(Keyword).filter_by(user_id=get_user_id_from_username(username)).all()