def register():
try:
request_data = request.get_json()
if query_user(request_data["email"]):
return jsonify({'status': 'failed', 'data': '该邮箱已存在'})
password_hash = generate_password_hash(request_data["password"])
user_data_store.create_user(email=request_data['email'],
password=password_hash,
createAt=datetime.utcnow())
user = user_data_store.find_user(email=request_data['email'])
for role in request_data['roles']:
user_data_store.add_role_to_user(user, role)
current_app.logger.info("register user successfully. email: %s" %
str(request_data['email']))
return jsonify({'status': 'ok', 'data': '注册成功'})
except BaseException as e:
current_app.logger.error("register user failed. - %s" % str(e))
return jsonify({'status': 'failed', 'data': 'register failed %s' % e})
def register():
if request.method == 'POST':
f_name = request.form['first_name']
l_name = request.form['last_name']
email = request.form['email']
password = request.form['password']
a_password = request.form['repeat_password']
if password == a_password:
user_data_store.create_user(
first_name=f_name,
last_name=l_name,
email=email,
password=password,
)
db.session.commit()
return redirect(url_for("security.login"))
return render_template('security/register.html', title='Регистрация')
def add_admin_user():
try:
(existed, data) = admin_user_existed()
if existed:
return jsonify({'status': 'ok', data: '"管理员用户已经存在, 邮箱为 %s' % data})
create_roles()
request_data = request.get_json()
email = request_data['email']
password = request_data['password']
roles = ['admin', 'user', 'project']
password_hash = generate_password_hash(password)
user_data_store.create_user(email=email,
password=password_hash,
createAt=datetime.utcnow())
user = user_data_store.find_user(email=email)
for role in roles:
user_data_store.add_role_to_user(user, role)
return jsonify({'status': 'ok', 'data': '添加管理员用户成功'})
except BaseException as e:
current_app.logger.error("add_admin_user failed. - %s" % str(e))
return jsonify({'status': 'failed', 'data': "出错了, Error: %s" % e})
from app import db, user_data_store
from user.models import User, Role
user_data_store.create_user(
first_name="Сергей",
last_name="Кайнов",
email="*****@*****.**",
password="******",
)
user_data_store.create_role(
naem="admin",
description="admin",
)
db.session.commit()
user = User.query.first()
role = Role.query.first()
user_data_store.add_role_to_user(user, role)
db.session.commit()