Poodletooth-iLand/toontown/estate/FlowerBasket.py

import GardenGlobals
from direct.directnotify import DirectNotifyGlobal
import FlowerBase

class FlowerBasket:
    notify = DirectNotifyGlobal.directNotify.newCategory('FlowerBasket')

    def __init__(self):
        self.flowerList = []

    def __len__(self):
        return len(self.flowerList)

    def getFlower(self):
        return self.flowerList

    def makeFromNetLists(self, speciesList, varietyList):
        self.flowerList = []
        for species, variety in zip(speciesList, varietyList):
            self.flowerList.append(FlowerBase.FlowerBase(species, variety))

    def getNetLists(self):
        speciesList = []
        varietyList = []
        for flower in self.flowerList:
            speciesList.append(flower.getSpecies())
            varietyList.append(flower.getVariety())

        return [speciesList, varietyList]

    def hasFlower(self, species, variety):
        for flower in self.flowerList:
            if flower.getSpecies() == species and flower.getVariety() == variety:
                return 1

        return 0

    def addFlower(self, species, variety):
        self.flowerList.append(FlowerBase.FlowerBase(species, variety))
        return 1

    def removeFishAtIndex(self, index):
        if index >= len(self.flowerList):
            return 0
        else:
            del self.flowerList[i]
            return 1

    def generateRandomBasket(self):
        import random
        numFish = random.randint(1, 20)
        self.flowerList = []
        for i in xrange(numFish):
            species, variety = GardenGlobals.getRandomFlower()
            self.addFlower(species, variety)

    def getTotalValue(self):
        value = 0
        for flower in self.flowerList:
            value += flower.getValue()

        return value

    def __str__(self):
        numFlower = len(self.flowerList)
        value = 0
        txt = 'Flower Basket (%s flower):' % numFlower
        for flower in self.flowerList:
            txt += '\n' + str(flower)

        value = self.getTotalValue()
        txt += '\nTotal value: %s' % value
        return txt