def load_media(token, filename):
print "here"
result = database_helper.check_token(token)
if result:
return send_from_directory("loadmedia", filename)
else:
return jsonify({"success": False, "message": 'not allowed'})
def get_user_messages_by_email(email):
token = request.headers.get('token')
if token and database_helper.check_token(token):
result = database_helper.get_user_messages_by_email(email)
if result:
return json.dumps({"success": "true", "message": "User messages retrieved.", "data": result}), 200
else:
return json.dumps({"success": "false", "message": "Something went wrong!"}), 500
else:
return json.dumps({"success": "false", "message": "Something went wrong!"}), 400
def get_user_data_by_email(email):
token = request.headers.get("Authorization")
if not database_helper.check_token(token):
return jsonify(success=False, message="Token does not exist.")
if database_helper.get_user_data(email):
data = database_helper.get_user_data(email)
return jsonify(success="True", message="User data sent.", data=data)
else:
return jsonify(success=False, message="User does not exist.")
def sign_out():
data = request.get_json()
token = request.headers.get('token')
if token and database_helper.check_token(token):
result = database_helper.sign_out(token)
if result == True:
return json.dumps({"success": "true","message":"Successfully signed out."}), 200
else:
return json.dumps({"success": "false", "message": "Something went wrong!"}), 500
else:
return json.dumps({"success": "false", "message": "Something went wrong!"}), 400
def post_message():
data = request.get_json()
token = request.headers.get('token')
if token and 'content' in data and 'email' not in data and database_helper.check_token(token):
email = database_helper.get_email_from_token(token)
writer = email
result = database_helper.post_message(email, writer, data['content'])
if result == True:
return json.dumps({"success": "true", "message": "Message posted"}), 200
else:
return json.dumps({"success": "false", "message": "Something went wrong!"}), 500
elif token and 'email' in data and 'content' in data and database_helper.check_token(token):
writer = database_helper.get_email_from_token(token)
result = database_helper.post_message(data['email'], writer, data['content'])
if result == True:
return json.dumps({"success": "true", "message": "Message posted"}), 200
else:
return json.dumps({"success": "false", "message": "Something went wrong!"}), 500
else:
return json.dumps({"success": "false", "message": "Missing field"}), 400
def api():
if request.environ.get('wsgi.websocket'):
ws = request.environ['wsgi.websocket']
while True:
msg = ws.receive()
if database_helper.check_token(msg):
email = database_helper.get_email_by_token(msg)
webSockets[email] = ws
else:
ws.close()
return "OK"
def get_user_messages_by_token():
token = request.headers.get("Authorization")
if database_helper.check_token(token):
email = database_helper.get_email_by_token(token)
if database_helper.get_user_messages(email):
messages = database_helper.get_user_messages(email)
return jsonify(success=True, message="Messages recieved!", data=messages)
else:
return jsonify(success=False, message="User has no messages yet.")
else:
return jsonify(success=False, message="Not logged in.")
def get_user_messages_by_email(email):
token = request.headers.get("Authorization")
if not database_helper.check_token(token):
return jsonify(success=False, message="Token does not exist.")
if database_helper.get_user_data(email):
if database_helper.get_user_messages(email):
messages = database_helper.get_user_messages(email)
return jsonify(success=True, message="Messages recieved!", data=messages)
else:
return jsonify(success=False, message="User has no messages yet.")
else:
return jsonify(success=False, message="User does not exist.")
def api():
if request.environ.get('wsgi.websocket'):
ws = request.environ['wsgi.websocket']
while True:
msg = ws.receive()
message = json.loads(msg)
if message["id"] == "token":
if database_helper.check_token(message["token"]):
email = database_helper.get_email_by_token(message["token"])
webSockets[email] = ws
else:
ws.close()
elif message["id"] == "chartUpdate":
if database_helper.check_token(message["token"]):
email = database_helper.get_email_by_token(message["token"])
chartsUpdate(email)
else:
ws.close()
else:
ws.close()
return "OK"
def Change_password():
data = request.get_json()
token=request.headers.get('token')
if token and 'oldPassword' in data and 'newPassword' in data and database_helper.check_token(token):
email = database_helper.get_email_from_token(token)
old_password = database_helper.get_old_password(email)
if data['oldPassword'] == old_password and 5 <= len(data['newPassword']) <= 30 and data['newPassword']==data['repeatPassword']:
result = database_helper.Change_password(token, data['newPassword'])
if result == True:
return json.dumps({"success": "true", "message": "Password changed!"}), 200
else:
return json.dumps({"success": "false", "message": "Something wxent wrong!"}), 500
else:
return json.dumps({"success": "false", "message": "Check again your password!"}), 400
else:
return json.dumps({"success": "false", "message": "Missing field!"}), 400
def change_password():
token = request.headers.get("Authorization")
data = request.get_json()
keys = ["old_password", "new_password"]
if not check_input_data(data, keys):
return jsonify(success=False, message="Wrong input data.")
if not database_helper.check_token(token):
return jsonify(success=False, message="You must be logged in to change password!")
email = database_helper.get_email_by_token(token)
if database_helper.check_valid_user(email, data['old_password']):
database_helper.update_password(data['new_password'], email)
return jsonify(success=True, message="Password changed!")
else:
return jsonify(success=False, message="Wrong old password!")
def post_message():
token = request.headers.get("Authorization")
data = request.get_json()
keys = ["message", "recipient"]
if not check_input_data(data, keys):
return jsonify(success=False, message="Wrong input data.")
if not database_helper.check_token(token):
return jsonify(success=False, message="Not signed in!")
if not database_helper.get_user_data(data['recipient']):
return jsonify(success=False, message="User does not exist.")
sender = database_helper.get_email_by_token(token)
try:
database_helper.add_message(sender, data['message'], data['recipient'])
return jsonify(success=True, message="Message posted!")
except:
return jsonify(success=False, message="Could not send message!")
