返回两种类型的字符串之一

Question

我有一个 Python 函数，它可以返回两种类型的返回值之一，animal 或 food。两者都是字符串类型。现在，我通过总是返回两个值来区分两者，断言其中只有一个是True：

def help(x):
    if some_condition(x):
        return 'monkey', None
    elif # ...
        animal = some_fantasy_animal(x)
        return animal, None
    elif #..
        food = some_food_no_one_has_ever_heard_of(x)
        return None, food
    elif # ...
        return None, 'steak'
    # ...



def main():
    animal, food = help('ddd')
    if animal:
        assert(not food)
        # do something useful
        pass
    else:
        assert(food)
        # do something useful
        pass
    return

必须有更好的方法来处理这个问题。有哪些选择？

Answer 1

你的场景有点奇怪

您可能想要创建一个 Animal 类和一个 Food 类，并让它们派生自某个公共基类。然后您可以使用isinstance 或更好，忘记if 条件并调用基类中定义的方法，您可以为每种类型覆盖该方法。

如果 Food 和 Animal 之间没有共同点，那么您可能不应该有一个可以返回任一类型的函数。

Answer 2

这可能是一个解决方案。将数据表示为：

data = {'animal':['cat','dog'], 'food':['steak','egg']}

检查某个值 x 是动物还是食物：

t = [k for k, v in data.iteritems() if x in data[k]]

t 然后将包含特定项目所属的类。请注意，如果animal 也以food 的形式出现，那么t 将同时包含两者。有点灵活！

Answer 3

您可以创建一个接受参数的类，并传递所有用于确定内部类型的逻辑：

例如：

class Result(object):
  def __init__(self, animal=None, food=None):
    self.animal = animal
    self.food = food 

  @property
  def is_animal(self):
    return self.animal != None

  @property
  def is_food(self):
    return self.food != None

def help(x):
  if x == 1:
    return Result(animal='monkey')
  elif x == 2:
    return Result(animal='hippo')
  elif x == 3:
    return Result(food='bananas')
  else:
    return Result()

if __name__ == '__main__':
  result = help(3)
  if result.is_animal:
    print result.animal
  if result.is_food:
    print result.food

Answer 4

好吧，您可以创建一组动物和食物，例如：

animals={'hippo', 'monkey'}
food={'bananas', 'steak'}

然后与in运营商核对：

h = help('ddd')

if h in animals:
    ...
elif h in food:
    ...