def __init__(self):
"""Initialize the server manager object.
The parameters are passed on to the init function of server manager
"""
self.user = User("")
self.usermanager = UserManager()
self.absolutePath = os.path.dirname(
os.path.abspath(inspect.getfile(inspect.currentframe())))
self.fileManager = FileManager(self.absolutePath)
super().__init__()
def login():
if request.method == "POST":
email = request.form['email']
passwd = request.form['password']
# Create login logic and proceed with the authentication
user_class = User()
user_class.id = email
flask_login.login_user(user_class)
return redirect('/protected')
def post(self):
username = self.request.get("username")
password = self.request.get("password")
verify = self.request.get("verify")
email = self.request.get("email")
validCheck = verifySignup(username, password, verify, email)
if validCheck.verify():
password = h.makePasswordHash(password)
if email:
user = User(username=username, password=password, email=email)
else:
user = User(username=username, password=password)
user.put()
self.response.headers.add_header(
'Set-Cookie',
'username=%s; Path=/' % str(h.makeSecureCookie(username)))
if user.is_saved():
self.redirect('/blog/welcome')
else:
self.renderFront(username, "", "", email, validCheck.errors)
def register():
try:
userId = str(uuid.uuid4())
firstName = request.json['firstName']
lastName = request.json['lastName']
email = request.json['email']
password = request.json['password']
hashed_password = generate_password_hash(password, method='sha256')
user = User(userId, firstName, lastName, email, hashed_password)
db.session.add(user)
db.session.commit()
return get_user_schema.jsonify(user)
except:
return {"msg": "Something went wrong"}, 500
def handle_updates(self):
update = bottle_request.json
print("\n\nNEW UPDATE\n{}\n\n".format(update)) #just for log
if (update.get("callback_query")): #handle button click
requests.get(
self.api_base_url + "answerCallbackQuery",
{'callback_query_id': update.get("callback_query").get("id")})
user_id = update.get("callback_query").get("from").get("id")
if (update.get("callback_query").get("data") == "google-pressed"):
self.show_url(
update.get("callback_query").get("message").get(
"chat").get("id"),
update.get("callback_query").get("message").get(
"message_id"))
else:
self.send_message_to_specific_person(
user_id,
update.get("callback_query").get("data"))
return
user_id = update.get("message").get("from").get("id")
user = self.users_list.get(user_id)
if (user):
MongoDB.newInteractionFromUser(user_id)
if (not user.greeted):
self.start_the_chat(user)
return
elif (not user.received_inicial_message):
self.send_initial_message(user)
else:
self.send_message_to_specific_person(
user_id,
"A partir daqui eu ainda nao sei o que fazer hehehe desculpa"
)
else:
# setup users
MongoDB.insertNewUser(update.get("message").get("from"))
first_name = update.get("message").get("from").get("first_name")
last_name = update.get("message").get("from").get("last_name")
username = update.get("message").get("from").get("username")
user = User(user_id, first_name, last_name, username)
self.users_list[user_id] = user
# greet user
self.start_the_chat(user)
return {"status": "up"}
def new_user():
login = request.json.get('Login')
password = request.json.get('password')
if login is None or password is None:
abort(400) # missing arguments
if User.query.filter_by(Login=login).first() is not None:
abort(400) # existing user
user = User(Login=login)
user.hash_password(password)
user.FirstName = request.json.get('FirstName')
user.LastName = request.json.get('LastName')
user.CreationDate = datetime.utcnow()
db.session.add(user)
db.session.commit()
user_dict = user.dict_format()
formatted_result = dict(User=user_dict)
return jsonify(result=formatted_result)
def getUserCredentials(self, username):
"""Get user data for specified username
Parameters:
------------------------------------------
username : string
username to fetch data
Return : User
returns the User Data Model for the specified username
"""
user = User(username)
with open(self.user_db) as u_file:
for cnt, line in enumerate(u_file):
if cnt > 0:
attr = line.split(',')
if attr[0] == username:
user.privilege = attr[2].strip("\t\r\n")
user.password = attr[1]
user.rootDirectory = "/User/" + username
user.currentDirectory = user.rootDirectory
break
return user
from flask import Flask, request, Response, jsonify
import jwt
import datetime, json
from UserModel import User
from ConversationModel import *
from functools import wraps
from HotorBot import *
from settings import *
user = User()
conv = Conversation()
app.config['SECRET_KEY'] = 'marcos'
def token_required(func):
@wraps(func)
def wrapper(*args, **kwargs):
token = request.args.get('token')
try:
jwt.decode(token, app.config['SECRET_KEY'], algorithms=['HS256'])
except:
return jsonify({'Error': 'Provide a valid JWT.'}), 401
return func(*args, **kwargs)
return wrapper
def getJWT():
exp_date = datetime.datetime.utcnow() + datetime.timedelta(minutes=30)
token = jwt.encode({'exp': exp_date},
app.config['SECRET_KEY'],
def load_user(user_id):
user = User()
user.id = user_id
return user