63 lines
2.5 KiB
Python
63 lines
2.5 KiB
Python
|
from direct.directnotify import DirectNotifyGlobal
|
||
|
from direct.distributed.ClockDelta import globalClockDelta
|
||
|
|
||
|
from toontown.estate import BankGlobals
|
||
|
from toontown.estate.DistributedFurnitureItemAI import DistributedFurnitureItemAI
|
||
|
|
||
|
|
||
|
class DistributedBankAI(DistributedFurnitureItemAI):
|
||
|
notify = DirectNotifyGlobal.directNotify.newCategory('DistributedBankAI')
|
||
|
|
||
|
def __init__(self, air, house, furnitureMgr, catalogItem):
|
||
|
DistributedFurnitureItemAI.__init__(self, air, house, furnitureMgr, catalogItem)
|
||
|
self.ownerId = self.house.avatarId
|
||
|
self.avId = 0
|
||
|
|
||
|
def avatarEnter(self):
|
||
|
avId = self.air.getAvatarIdFromSender()
|
||
|
if avId != self.ownerId:
|
||
|
self.d_setMovie(BankGlobals.BANK_MOVIE_NOT_OWNER, avId)
|
||
|
return
|
||
|
|
||
|
# Display the GUI:
|
||
|
self.d_setMovie(BankGlobals.BANK_MOVIE_GUI, avId)
|
||
|
|
||
|
def d_setMovie(self, mode, avId):
|
||
|
self.sendUpdate('setMovie', [mode, avId, globalClockDelta.getRealNetworkTime()])
|
||
|
|
||
|
def transferMoney(self, amount):
|
||
|
avId = self.air.getAvatarIdFromSender()
|
||
|
if avId != self.ownerId:
|
||
|
self.air.writeServerEvent('suspicious', avId, 'av tried to transfer money with bank they don\'t own')
|
||
|
return
|
||
|
|
||
|
av = self.air.doId2do.get(avId)
|
||
|
if not av:
|
||
|
self.air.writeServerEvent('suspicious', avId, 'av tried to transfer money with bank not on their shard')
|
||
|
return
|
||
|
|
||
|
if av.getLocation() != self.getLocation():
|
||
|
self.air.writeServerEvent('suspicious', avId, 'av tried to transfer money with bank not in their zone')
|
||
|
return
|
||
|
|
||
|
bankMoney = av.getBankMoney()
|
||
|
money = av.getMoney()
|
||
|
if amount < 0 and amount < av.getMaxMoney():
|
||
|
if bankMoney + amount < 0:
|
||
|
self.air.writeServerEvent('suspicious', avId, 'av tried to withdraw more money than they have')
|
||
|
return
|
||
|
|
||
|
av.b_setMoney(money - amount)
|
||
|
av.b_setBankMoney(bankMoney + amount)
|
||
|
self.d_setMovie(BankGlobals.BANK_MOVIE_WITHDRAW, avId)
|
||
|
elif amount > 0 and money < av.getMaxBankMoney():
|
||
|
if money < amount:
|
||
|
self.air.writeServerEvent('suspicious', avId, 'av tried to deposit more money than they have')
|
||
|
return
|
||
|
|
||
|
av.b_setMoney(money - amount)
|
||
|
av.b_setBankMoney(bankMoney + amount)
|
||
|
self.d_setMovie(BankGlobals.BANK_MOVIE_DEPOSIT, avId)
|
||
|
elif amount == 0:
|
||
|
self.d_setMovie(BankGlobals.BANK_MOVIE_NO_OP, avId)
|