生成对象或对象列表的对象工厂

时间:2010-09-07 03:25:19

标签: python object factory

我有以下代码:

def f(cls, value):
  # cls is a class
  # value is a string value
  if cls == str:
    pass # value is already the right type
  elif cls == int:
    value = int(value)
  elif cls == C1:
    value = C1(value)
  elif cls == C2:
    value = C2(value)
  elif cls == C3
    # in this case, we convert the string into a list of objects of class C3
    value = C3.parse(value)
  else
    raise UnknownClass(repr(cls))
  return value

显然,我正试图用以下内容替换它:

def f(cls, value)
   return cls(value)

不幸的是,在某些情况下(如果cls == C3),输入的解析会导致该类的对象列表,而不仅仅是一个对象。处理这个问题的方法是什么?我可以访问所有课程。

2 个答案:

答案 0 :(得分:2)

如果大多数案例最好通过调用cls来处理,而其中一些最好处理,最简单的方法就是挑出后者:

themap = {C3: C3.parse}
for C in (str, C1, C2):
    themap[C] = C

def f(cls, value):
    wot = themap.get(cls)
    if wot is None:
        raise UnknownClass(repr(cls))
    return wot(value)

请注意,在字符串上调用str是一个相当快的noop,因此通常值得避免“挑出”特定情况以支持代码简单。

答案 1 :(得分:0)

这取决于你对列表的处理方式。这是最简单的方法:

obj = cls(value)

如果type(obj)== list:

handle_list(OBJ)

返回obj