108 lines
3.4 KiB
Python
108 lines
3.4 KiB
Python
__author__ = "Steffen Vogel"
|
|
__copyright__ = "Copyright 2015-2017, Steffen Vogel"
|
|
__license__ = "GPLv3"
|
|
__maintainer__ = "Steffen Vogel"
|
|
__email__ = "post@steffenvogel.de"
|
|
|
|
"""
|
|
This file is part of transWhat
|
|
|
|
transWhat is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
any later version.
|
|
|
|
transwhat is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with transWhat. If not, see <http://www.gnu.org/licenses/>.
|
|
"""
|
|
|
|
from Spectrum2 import protocol_pb2
|
|
|
|
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 = protocol_pb2.PARTICIPANT_FLAG_MODERATOR
|
|
else:
|
|
flags = protocol_pb2.PARTICIPANT_FLAG_NONE
|
|
if number == yourNumber:
|
|
flags = flags | protocol_pb2.PARTICIPANT_FLAG_ME
|
|
|
|
try:
|
|
self._updateParticipant(number, flags, protocol_pb2.STATUS_ONLINE,
|
|
self.backend.sessions[self.user].buddies[number].image_hash)
|
|
except KeyError:
|
|
self._updateParticipant(number, flags, protocol_pb2.STATUS_ONLINE)
|
|
|
|
def removeParticipants(self, participants):
|
|
for jid in participants:
|
|
number = jid.split('@')[0]
|
|
nick = self.participants[number]
|
|
flags = protocol_pb2.PARTICIPANT_FLAG_NONE
|
|
self._updateParticipant(number, flags, protocol_pb2.STATUS_NONE)
|
|
del self.participants[number]
|
|
|
|
def leaveRoom(self):
|
|
for number in self.participants:
|
|
nick = self.participants[number]
|
|
flags = protocol_pb2.PARTICIPANT_FLAG_ROOM_NOT_FOUND
|
|
self._updateParticipant(number, flags, protocol_pb2.STATUS_NONE)
|
|
|
|
def changeNick(self, number, new_nick):
|
|
if self.participants[number] == new_nick:
|
|
return
|
|
if number == self.owner:
|
|
flags = protocol_pb2.PARTICIPANT_FLAG_MODERATOR
|
|
else:
|
|
flags = protocol_pb2.PARTICIPANT_FLAG_NONE
|
|
self._updateParticipant(number, flags, 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)
|