transpub/transWhat/group.py

import Spectrum2

class Group():

	def __init__(self, id, owner, subject, subjectOwner, backend, user):
		self.id = id
		self.subject = subject
		self.subjectOwner = subjectOwner
		self.owner = owner
		self.joined = False
		self.backend = backend
		self.user = user

		self.nick = "me"
		# Participants is a number -> nickname dict
		self.participants = {}

	def addParticipants(self, participants, buddies, yourNumber):
		"""
		Adds participants to the group.

		Args:
			- participants: (Iterable) phone numbers of participants
			- buddies: (dict) Used to get the nicknames of the participants
			- yourNumber: The number you are using
		"""
		for jid in participants:
			number = jid.split('@')[0]
			try:
				nick = buddies[number].nick
			except KeyError:
				nick = number
			if number == yourNumber:
				nick = self.nick
			if nick == "":
				nick = number
			self.participants[number] = nick

	def sendParticipantsToSpectrum(self, yourNumber):
		for number, nick in self.participants.iteritems():
			if number == self.owner:
				flags = Spectrum2.protocol_pb2.PARTICIPANT_FLAG_MODERATOR
			else:
				flags = Spectrum2.protocol_pb2.PARTICIPANT_FLAG_NONE
			if number == yourNumber:
				flags = flags | Spectrum2.protocol_pb2.PARTICIPANT_FLAG_ME
			
			try:
				self._updateParticipant(number, flags, Spectrum2.protocol_pb2.STATUS_ONLINE, 
					self.backend.sessions[self.user].buddies[number].image_hash)
			except KeyError:
				self._updateParticipant(number, flags, Spectrum2.protocol_pb2.STATUS_ONLINE)

	def removeParticipants(self, participants):
		for jid in participants:
			number = jid.split('@')[0]
			nick = self.participants[number]
			flags = Spectrum2.protocol_pb2.PARTICIPANT_FLAG_NONE
			self._updateParticipant(number, flags, Spectrum2.protocol_pb2.STATUS_NONE)
			del self.participants[number]

	def leaveRoom(self):
		for number in self.participants:
			nick = self.participants[number]
			flags = Spectrum2.protocol_pb2.PARTICIPANT_FLAG_ROOM_NOT_FOUND
			self._updateParticipant(number, flags, Spectrum2.protocol_pb2.STATUS_NONE)

	def changeNick(self, number, new_nick):
		if self.participants[number] == new_nick:
			return
		if number == self.owner:
			flags = Spectrum2.protocol_pb2.PARTICIPANT_FLAG_MODERATOR
		else:
			flags = Spectrum2.protocol_pb2.PARTICIPANT_FLAG_NONE
		self._updateParticipant(number, flags, Spectrum2.protocol_pb2.STATUS_ONLINE, new_nick)
		self.participants[number] = new_nick

	def _updateParticipant(self, number, flags, status, imageHash = "", newNick = ""):
		nick = self.participants[number]
		# Notice the status message is the buddy's number
		if self.joined:
			self.backend.handleParticipantChanged(
					self.user, nick, self.id, flags,
					status, number, newname = newNick, iconHash = imageHash)