135 lines
4.4 KiB
Python
135 lines
4.4 KiB
Python
import CatalogItem
|
|
import time
|
|
from toontown.toonbase import ToontownGlobals
|
|
from toontown.toonbase import TTLocalizer
|
|
from otp.otpbase import OTPLocalizer
|
|
from direct.interval.IntervalGlobal import *
|
|
from toontown.toontowngui import TTDialog
|
|
|
|
class CatalogRentalItem(CatalogItem.CatalogItem):
|
|
|
|
def makeNewItem(self, typeIndex, duration, cost):
|
|
self.typeIndex = typeIndex
|
|
self.duration = duration
|
|
self.cost = cost
|
|
CatalogItem.CatalogItem.makeNewItem(self)
|
|
|
|
def getRentalType(self):
|
|
return self.typeIndex
|
|
|
|
def getDuration(self):
|
|
return self.duration
|
|
|
|
def getPurchaseLimit(self):
|
|
return 0
|
|
|
|
def reachedPurchaseLimit(self, avatar):
|
|
if self in avatar.onOrder or self in avatar.mailboxContents or self in avatar.onGiftOrder or self in avatar.awardMailboxContents or self in avatar.onAwardOrder:
|
|
return 1
|
|
return 0
|
|
|
|
def saveHistory(self):
|
|
return 1
|
|
|
|
def getTypeName(self):
|
|
return TTLocalizer.RentalTypeName
|
|
|
|
def getName(self):
|
|
hours = int(self.duration / 60)
|
|
if self.typeIndex == ToontownGlobals.RentalCannon:
|
|
return '%s %s %s %s' % (hours,
|
|
TTLocalizer.RentalHours,
|
|
TTLocalizer.RentalOf,
|
|
TTLocalizer.RentalCannon)
|
|
elif self.typeIndex == ToontownGlobals.RentalGameTable:
|
|
return '%s %s %s' % (hours, TTLocalizer.RentalHours, TTLocalizer.RentalGameTable)
|
|
else:
|
|
return TTLocalizer.RentalTypeName
|
|
|
|
def recordPurchase(self, avatar, optional):
|
|
self.notify.debug('rental -- record purchase')
|
|
if avatar:
|
|
self.notify.debug('rental -- has avater')
|
|
estate = simbase.air.estateManager.toon2estate.get(avatar)
|
|
if estate:
|
|
self.notify.debug('rental -- has estate')
|
|
estate.rentItem(self.typeIndex, self.duration)
|
|
else:
|
|
self.notify.debug('rental -- something not there')
|
|
return ToontownGlobals.P_ItemAvailable
|
|
|
|
def getPicture(self, avatar):
|
|
scale = 1
|
|
heading = 0
|
|
pitch = 30
|
|
roll = 0
|
|
spin = 1
|
|
down = -1
|
|
if self.typeIndex == ToontownGlobals.RentalCannon:
|
|
model = loader.loadModel('phase_4/models/minigames/toon_cannon')
|
|
scale = 0.5
|
|
heading = 45
|
|
elif self.typeIndex == ToontownGlobals.RentalGameTable:
|
|
model = loader.loadModel('phase_6/models/golf/game_table')
|
|
self.hasPicture = True
|
|
return self.makeFrameModel(model, spin)
|
|
|
|
def output(self, store = -1):
|
|
return 'CatalogRentalItem(%s%s)' % (self.typeIndex, self.formatOptionalData(store))
|
|
|
|
def compareTo(self, other):
|
|
return self.typeIndex - other.typeIndex
|
|
|
|
def getHashContents(self):
|
|
return self.typeIndex
|
|
|
|
def getBasePrice(self):
|
|
if self.typeIndex == ToontownGlobals.RentalCannon:
|
|
return self.cost
|
|
elif self.typeIndex == ToontownGlobals.RentalGameTable:
|
|
return self.cost
|
|
else:
|
|
return 50
|
|
|
|
def decodeDatagram(self, di, versionNumber, store):
|
|
CatalogItem.CatalogItem.decodeDatagram(self, di, versionNumber, store)
|
|
if versionNumber >= 7:
|
|
self.cost = di.getUint16()
|
|
else:
|
|
self.cost = 1000
|
|
self.duration = di.getUint16()
|
|
self.typeIndex = di.getUint16()
|
|
|
|
def encodeDatagram(self, dg, store):
|
|
CatalogItem.CatalogItem.encodeDatagram(self, dg, store)
|
|
dg.addUint16(self.cost)
|
|
dg.addUint16(self.duration)
|
|
dg.addUint16(self.typeIndex)
|
|
|
|
def getDeliveryTime(self):
|
|
return 1
|
|
|
|
def isRental(self):
|
|
return 1
|
|
|
|
def acceptItem(self, mailbox, index, callback):
|
|
self.confirmRent = TTDialog.TTGlobalDialog(doneEvent='confirmRent', message=TTLocalizer.MessageConfirmRent, command=Functor(self.handleRentConfirm, mailbox, index, callback), style=TTDialog.TwoChoice)
|
|
self.confirmRent.show()
|
|
|
|
def handleRentConfirm(self, mailbox, index, callback, choice):
|
|
if choice > 0:
|
|
mailbox.acceptItem(self, index, callback)
|
|
else:
|
|
callback(ToontownGlobals.P_UserCancelled, self, index)
|
|
if self.confirmRent:
|
|
self.confirmRent.cleanup()
|
|
self.confirmRent = None
|
|
return
|
|
|
|
|
|
def getAllRentalItems():
|
|
list = []
|
|
for rentalType in (ToontownGlobals.RentalCannon,):
|
|
list.append(CatalogRentalItem(rentalType, 2880, 1000))
|
|
|
|
return list
|