shapefactory1.py

来自「this is the most basic to learn python」· Python 代码 · 共 52 行

#: c05:shapefact1:ShapeFactory1.py
# A simple static factory method.
from __future__ import generators
import random

class Shape(object):
  # Create based on class name:
  def factory(type):
    #return eval(type + "()")
    if type == "Circle": return Circle()
    if type == "Square": return Square()
    assert 1, "Bad shape creation: " + type
  factory = staticmethod(factory)

class Circle(Shape):
  def draw(self): print "Circle.draw" 
  def erase(self): print "Circle.erase" 

class Square(Shape):
  def draw(self): print "Square.draw" 
  def erase(self): print "Square.erase" 

# Generate shape name strings:
def shapeNameGen(n):
  types = Shape.__subclasses__()
  for i in range(n):
    yield random.choice(types).__name__

shapes = \
  [ Shape.factory(i) for i in shapeNameGen(7)]

for shape in shapes:
  shape.draw()
  shape.erase()
#<hr>
output = '''
Square.draw
Square.erase
Circle.draw
Circle.erase
Square.draw
Square.erase
Circle.draw
Circle.erase
Circle.draw
Circle.erase
Circle.draw
Circle.erase
Square.draw
Square.erase
'''