Wednesday, July 30, 2008

OO PYTHON ONE: inheritance

class Fruit(object):
def __init__(self,name,color):
self.name = name
self.color = color

def getName(self):
return self.name;

def getColor(self):
return self.color;

def setColor(self, color):
self.color = color

def setName(self, name):
self.name = name

class Melon(Fruit):

def __init__(self, name, color):
super(Melon, self).__init__(name, color)
self.shape = 'round'

def setColor(self):
super(Melon, self).setColor('orange')


def getShape(self):
return self.shape;


fruits = []

names = ['apple', 'pear', 'peach', 'pineapple','cantalope','watermelon']
colors = ['white','red', 'pink','green','yellow','blue']

for name in names:
fruits.append(Fruit(name,colors[random.randint(0,5)]))

for fruit in fruits:
print "Fruit " + fruit.getName() + "'s color is " + fruit.getColor()

for fruit in fruits:
fruit.color = 'transparent';
print "Fruit " + fruit.getName() + "'s color has been changed to " + fruit.getColor()


melon1 = Melon('wintermelon','green')
melon1.setColor()

print melon1.getName()
print melon1.getColor()