spy.py 文件源码

def add_members_to_group(self, members_username, group_name):
        found_group = None

        if type(members_username) != list:
            members_username = [members_username]

        for group in self.groups:
            if group['name'] == group_name:
                found_group = group
                break

        if found_group:
            new_members = [
                member for member in members_username
                if member not in found_group['users']
            ]
            for member in new_members:
                self.bulk.find({'$and': [
                    {'username': self.username},
                    {'groups.name': group_name}
                ]}).update({'$push': {'groups.$.users': member}})

            try:
                self.bulk.execute()
            except InvalidOperation as e:
                print(e)
        else:
            print('Group {} does not exist!'.format(group_name))