фабрика объектов, которая генерирует объект или список объектов

Question

У меня есть следующий код:

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) при разборе входных данных получается список объектов этого класса, а не только один объект. Какой изящный способ справиться с этим? У меня есть доступ ко всем классам.

Answer 1

Если большинство дел лучше всего обрабатываются путем вызова 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, поэтому, как правило, стоит избегать «выделения» этого конкретного случая в пользу простоты кода.

Answer 2

Это зависит от того, что вы будете делать со списком. Это самый простой способ:

obj = cls (значение)

if type (obj) == список:

handle_list (объект)

возврат объекта