diff --git a/otp/friends/FriendManagerAI.py b/otp/friends/FriendManagerAI.py index 616cfbc..6db8ef8 100644 --- a/otp/friends/FriendManagerAI.py +++ b/otp/friends/FriendManagerAI.py @@ -1,5 +1,9 @@ from direct.directnotify import DirectNotifyGlobal -from direct.distributed.DistributedObjectAI import DistributedObjectAI +from direct.distributed.DistributedObjectGlobalAI import DistributedObjectGlobalAI -class FriendManagerAI(DistributedObjectAI): + +class FriendManagerAI(DistributedObjectGlobalAI): notify = DirectNotifyGlobal.directNotify.newCategory('FriendManagerAI') + + def friendQuery(self, inviteeId): + self.notify.info('Hello world: %s' % inviteeId) diff --git a/toontown/ai/ToontownAIRepository.py b/toontown/ai/ToontownAIRepository.py index 122949b..ae74e88 100644 --- a/toontown/ai/ToontownAIRepository.py +++ b/toontown/ai/ToontownAIRepository.py @@ -86,6 +86,7 @@ class ToontownAIRepository(ToontownInternalRepository): self.trophyMgr = None self.safeZoneManager = None self.magicWordManager = None + self.friendManager = None self.zoneTable = {} self.dnaStoreMap = {} self.dnaDataMap = {} @@ -208,6 +209,9 @@ class ToontownAIRepository(ToontownInternalRepository): self.magicWordManager = ToontownMagicWordManagerAI(self) self.magicWordManager.generateWithRequired(OTP_ZONE_ID_MANAGEMENT) + # Generate our friend manager... + self.friendManager = self.generateGlobalObject(OTP_DO_ID_FRIEND_MANAGER, 'FriendManager') + def generateHood(self, hoodConstructor, zoneId): # Bossbot HQ doesn't use DNA, so we skip over that. if zoneId != ToontownGlobals.BossbotHQ: