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))
