def db_update_user(**kwargs): """ update a user info in database 数据库更新用户信息 """ groups_post = kwargs.pop('groups') admin_groups_post = kwargs.pop('admin_groups') user_id = kwargs.pop('user_id') user = User.objects.filter(id=user_id) if user: user_get = user[0] password = kwargs.pop('password') user.update(**kwargs) if password.strip(): user_get.set_password(password) user_get.save() else: return None group_select = [] if groups_post: for group_id in groups_post: group = UserGroup.objects.filter(id=group_id) group_select.extend(group) user_get.group = group_select if admin_groups_post != '': user_get.admingroup_set.all().delete() for group_id in admin_groups_post: group = get_object(UserGroup, id=group_id) AdminGroup(user=user, group=group).save()
def db_update_user(**kwargs): """ update a user info in database 数据库更新用户信息 """ # groups_post = kwargs.pop('groups') admin_groups_post = kwargs.pop('admin_groups') user_id = kwargs.pop('user_id') user = User.objects.filter(id=user_id) user_get = User.objects.get(id=user_id) if user: pwd = kwargs.pop('password') user.update(**kwargs) if pwd != '': user_get.set_password(pwd) user_get.save() else: return None group_select = [] # if groups_post: # for group_id in groups_post: # group = CMDB_Group.objects.filter(id=group_id) # group_select.extend(group) # user_get.group_asset = group_select if admin_groups_post != '': user_get.admingroup_set.all().delete() for group_id in admin_groups_post: group = get_object(UserGroup, id=group_id) AdminGroup(user=user, group=group).save()
def db_add_user(**kwargs): """ add a user in database 数据库中添加用户 """ groups_post = kwargs.pop('groups') admin_groups = kwargs.pop('admin_groups') role = kwargs.get('role', 'CU') user = User(**kwargs) user.set_password(kwargs.get('password')) user.save() if groups_post: group_select = [] for group_id in groups_post: group = UserGroup.objects.filter(id=group_id) group_select.extend(group) user.group = group_select if admin_groups and role == 'GA': # 如果是组管理员就要添加组管理员和组到管理组中 for group_id in admin_groups: group = get_object(UserGroup, id=group_id) if group: AdminGroup(user=user, group=group).save() return user