def user_increase():
"""
新增用户
:return:
"""
input_json = request.json
if input_json:
user = User.get_or_none(User.user_name == input_json["user_name"])
if user:
logging.info("user already exist %s %s", user.user_name, user.id)
return {"msg":"user create fail. user already exist"}
else:
password = bytes(input_json["password"], "utf-8")
input_json["password"] = bcrypt.hashpw(password, bcrypt.gensalt())
User.create(**input_json)
return {"msg":"user create success"}
return {"msg":"invalid paramter"}
def createSuperUser():
from apps.models import User, Role
from apps.forms import checkEmail, checkUsername, checkPassword
# 从数据库绑定admin id
role_admin = Role.select(Role.id).where(Role.name == "admin").get().id
if not role_admin:
return ("必须先在Role表中创建管理员角色")
email = click.prompt("请输入邮箱")
checkEmail(email)
name = click.prompt("请输入用户名")
checkUsername(name)
password = click.prompt("请输入密码", hide_input=True, confirmation_prompt=True)
checkPassword(password)
password = User.generalPassword(password)
User.create(email=email,
name=name,
password=password,
sex=User.sex.choices.unknown,
role_id=role_admin,
signature="I am super admin")
print("成功创建超级管理员账户；%s" % name)
