74 lines
2 KiB
Python
74 lines
2 KiB
Python
|
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
|