def _rebuild_user_from_message(self, user_id, group_id, socket): user = self._get_user_by_id(user_id) group = self._get_group_by_id(group_id) user_recreated = False if not user: user_recreated = True user = User(user_id, socket) self.users.append(user) app.logger.info( u'Recreated user with id {} from message'.format(user_id)) if not group: group = Group(group_id) group.add_user(user) app.logger.info( u'Recreated group with id {} with user with id {} from message' .format(group_id, user_id)) elif user_recreated: group.add_user(user) app.logger.info( u'Added user with id {} back to group with id {} from message'. format(user_id, group_id)) return user, group
def add_user_to_group(self, user_id, group_id, socket): user = self._get_user_by_id(user_id) group = self._get_group_by_id(group_id) if not user: socket.send( serialize({ 'type': 'error', 'message': 'Invalid user ID' })) return if not group: group = Group(group_id) self.groups.append(group) app.logger.info(u'Created new group with id {}'.format( user_id, group_id)) added = group.add_user(user) if user.id in self.user_group_map and added: self.user_group_map[user.id].append(group) elif added: self.user_group_map[user.id] = [group] if added: app.logger.info(u'Added user {} to group {}'.format( user_id, group_id)) else: app.logger.info( u'Tried to add user {} to group {} but theyre already in there' .format(user_id, group_id))