Poodletooth-iLand/toontown/catalog/CatalogRentalItem.py
2015-03-03 17:10:12 -05:00

132 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._lookupEstate(avatar.doId)
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 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