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)