def signup(uuid, name):
"""Attempts to create user and then return the UUID of the newly signed up user
Args:
uuid (str): user id generated by facebook javascript SDK
Returns:
404 if user already exists, 201 if sign up is successful
"""
db = mongo.db.users
db_user_entry = db.find_one({'UUID': uuid})
# User already exists
if not (db_user_entry is None):
print('USER EXISTS')
return jsonify({"code": 404}), 404
new_user = User(name, uuid)
active_users[uuid] = new_user
serialized_user_obj = pickle.dumps(new_user)
db_user_entry = {"UUID": uuid, "serialized_user_obj": serialized_user_obj}
db.insert_one(db_user_entry)
return jsonify({"code": 201}), 201
def login():
form = forms.LoginForm()
if form.validate_on_submit():
admintable = all_table['admin']
user = db.session.query(admintable).filter_by(id=form.id.data).first()
if user is not None:
user = User(user._mapping)
if user.password == md5(
form.password.data.encode('ascii')).digest():
login_user(user)
return redirect(url_for('index'))
flash('用户名或密码不正确')
return render_template('login.html', form=form)
def create_test_dummy():
uuid = 666
name = "Test Dummy"
db = mongo.db.users
db_user_entry = db.find_one({'UUID': uuid})
# User already exists
if not (db_user_entry is None):
print('USER EXISTS')
return jsonify({"code": 404}), 404
current_user = User(name, uuid)
current_user.upload_data("tests/html/dummy_messages2.csv")
current_user.process_data()
active_users[uuid] = current_user
serialized_user_obj = pickle.dumps(current_user)
db_user_entry = {"UUID": uuid, "serialized_user_obj": serialized_user_obj}
db.insert_one(db_user_entry)
return jsonify({"code": 201})