def register(cls, user_name, real_name, password, email, telephone, role_id, job): if UserManager.verify_existence(user_name): return jsonify({"code": 400, "message": "用户已存在,请更换用户名"}) user = User() user.user_name = user_name user.real_name = real_name user.email = email user.telephone = telephone user.role_id = role_id user.job = job user.create_time = TimeManager.now() user.pwd = password db.session.add(user) user.save() return jsonify({"code": 200, "message": "注册成功"})
def init_db(app): # 生成数据库表 db.create_all() app.logger.debug('创建表') if Role.query.first(): return role_admin = Role(role_name="管理员", role_desc="系统管理员") role_audit = Role(role_name="审计员", role_desc="审计员") role_ordinary = Role(role_name="普通用户", role_desc="普通管理员") db.session.add_all([role_admin, role_audit, role_ordinary]) db.session.commit() app.logger.debug('初始化角色成功') admin = User(user_name="admin", role_id=1, create_time=TimeManager.now(), status=1, last_passwd_change_time=TimeManager.now()) admin.pwd = 'admin' db.session.add(admin) db.session.commit() app.logger.debug('初始化用户成功')