def init(force=False): db.create_all() if force: db.drop_all() db.create_all() elif (manager.user_model.get(one=False) or manager.user_group_model.get(one=False) or manager.group_model.get(one=False)): exit("表中存在数据,初始化失败") with db.auto_commit(): # 创建一个超级管理员分组 root_group = manager.group_model() root_group.name = "Root" root_group.info = "超级用户组" root_group.level = GroupLevelEnum.ROOT.value db.session.add(root_group) # 创建一个超级管理员 root = manager.user_model() root.username = "******" db.session.add(root) db.session.flush() root.password = "******" # root用户 and 超级管理员分组 对应关系写入user_group表中 user_group = manager.user_group_model() user_group.user_id = root.id user_group.group_id = root_group.id db.session.add(user_group) # 添加 默认游客组 guest_group = manager.group_model() guest_group.name = "Guest" guest_group.info = "游客组" guest_group.level = GroupLevelEnum.GUEST.value db.session.add(guest_group) # 初始化权限 manager.sync_permissions()
def register(json: UserRegisterSchema): """ 注册新用户 """ if manager.user_model.count_by_username(g.username) > 0: raise Duplicated("用户名重复,请重新输入") # type: ignore if g.email and g.email.strip() != "": if manager.user_model.count_by_email(g.email) > 0: raise Duplicated("注册邮箱重复,请重新输入") # type: ignore # create a user with db.auto_commit(): user = manager.user_model() user.username = g.username if g.email and g.email.strip() != "": user.email = g.email db.session.add(user) db.session.flush() user.password = g.password group_ids = g.group_ids # 如果没传分组数据,则将其设定为 guest 分组 if len(group_ids) == 0: from lin import GroupLevelEnum group_ids = [GroupLevelEnum.GUEST.value] for group_id in group_ids: user_group = manager.user_group_model() user_group.user_id = user.id user_group.group_id = group_id db.session.add(user_group) return Success("用户创建成功") # type: ignore
def _register_user(form: RegisterForm): with db.auto_commit(): user = manager.user_model() user.username = form.username.data if form.email.data and form.email.data.strip() != "": user.email = form.email.data db.session.add(user) db.session.flush() user.password = form.password.data group_ids = form.group_ids.data # 如果没传分组数据,则将其设定为 id 2 的 guest 分组 if len(group_ids) == 0: group_ids = [2] for group_id in group_ids: user_group = manager.user_group_model() user_group.user_id = user.id user_group.group_id = group_id db.session.add(user_group)