103 lines
3.7 KiB
Python
103 lines
3.7 KiB
Python
from direct.directnotify import DirectNotifyGlobal
|
|
from direct.distributed.DistributedObjectGlobalUD import DistributedObjectGlobalUD
|
|
|
|
|
|
class FriendsOperation:
|
|
|
|
def __init__(self, friendsManager, sender):
|
|
self.friendsManager = friendsManager
|
|
self.sender = sender
|
|
|
|
def _handleDone(self):
|
|
# TODO
|
|
pass
|
|
|
|
def _handleError(self, error):
|
|
# TODO
|
|
pass
|
|
|
|
|
|
class GetFriendsListOperation(FriendsOperation):
|
|
|
|
def __init__(self, friendsManager, sender):
|
|
FriendsOperation.__init__(self, friendsManager, sender)
|
|
self.friendsList = None
|
|
self.tempFriendsList = None
|
|
self.onlineFriends = None
|
|
self.currentFriendIdx = None
|
|
|
|
def start(self):
|
|
self.friendsList = []
|
|
self.tempFriendsList = []
|
|
self.onlineFriends = []
|
|
self.currentFriendIdx = 0
|
|
self.friendsManager.air.dbInterface.queryObject(self.friendsManager.air.dbId, self.sender,
|
|
self.__handleSenderRetrieved)
|
|
|
|
def __handleSenderRetrieved(self, dclass, fields):
|
|
if dclass != self.friendsManager.air.dclassesByName['DistributedToonUD']:
|
|
self._handleError('Retrieved sender is not a DistributedToonUD!')
|
|
return
|
|
|
|
self.tempFriendsList = fields['setFriendsList'][0]
|
|
if len(self.tempFriendsList) <= 0:
|
|
self.__sendFriendsList()
|
|
self._handleDone()
|
|
return
|
|
|
|
self.friendsManager.air.dbInterface.queryObject(self.friendsManager.air.dbId, self.tempFriendsList[0][0],
|
|
self.__handleFriendRetrieved)
|
|
|
|
def __handleFriendRetrieved(self, dclass, fields):
|
|
if dclass != self.friendsManager.air.dclassesByName['DistributedToonUD']:
|
|
self._handleError('Retrieved friend is not a DistributedToonUD!')
|
|
return
|
|
|
|
friendId = self.tempFriendsList[self.currentFriendIdx][0]
|
|
self.friendsList.append([friendId, fields['setName'][0], fields['setDNAString'][0], fields['setPetId'][0]])
|
|
if len(self.friendsList) >= len(self.tempFriendsList):
|
|
self.__checkFriendsOnline()
|
|
return
|
|
|
|
self.currentFriendIdx += 1
|
|
self.friendsManager.air.dbInterface.queryObject(self.friendsManager.air.dbId,
|
|
self.tempFriendsList[self.currentFriendIdx][0],
|
|
self.__handleFriendRetrieved)
|
|
|
|
def __checkFriendsOnline(self):
|
|
self.currentFriendIdx = 0
|
|
for friendDetails in self.friendsList:
|
|
self.friendsManager.air.getActivated(friendDetails[0], self.__gotActivatedResp)
|
|
|
|
def __gotActivatedResp(self, avId, activated):
|
|
self.currentFriendIdx += 1
|
|
if activated:
|
|
self.onlineFriends.append(avId)
|
|
|
|
if self.currentFriendIdx >= len(self.friendsList):
|
|
self.__sendFriendsList()
|
|
self._handleDone()
|
|
|
|
def __sendFriendsList(self):
|
|
self.friendsManager.sendUpdateToAvatarId(self.sender, 'getFriendsListResponse', [self.friendsList])
|
|
|
|
|
|
class ToontownFriendsManagerUD(DistributedObjectGlobalUD):
|
|
notify = DirectNotifyGlobal.directNotify.newCategory('ToontownFriendsManagerUD')
|
|
|
|
def __init__(self, air):
|
|
DistributedObjectGlobalUD.__init__(self, air)
|
|
self.operations = []
|
|
|
|
def runOperation(self, operationType, *args):
|
|
sender = self.air.getAvatarIdFromSender()
|
|
if not sender:
|
|
return
|
|
|
|
newOperation = operationType(self, sender)
|
|
self.operations.append(newOperation)
|
|
newOperation.start(*args)
|
|
|
|
def getFriendsListRequest(self):
|
|
self.runOperation(GetFriendsListOperation)
|