覆盖python方法以返回具有扩展数据的正确类的对象

Question

假设我有一个类似的python类：

class Meal(object):
   def __init__(self, starter, main="steak"):
       self.starter = starter
       self.main = main

   def new_meal_eaten_starter(self):
       return Meal("Eaten-%s" % self.starter, main=self.main)

   def __repr__(self):
       return "%s %s" % (self.starter, self.main)

现在我想继承这个类，添加一些数据：

class MealWithDessert(Meal):
    def __init__(self, dessert, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.dessert = dessert

    def __repr__(self):
        return "%s %s %s" % (self.starter, self.main, self.dessert)

这当然会发生：

>>> x = MealWithDessert("cake", "salad", "pizza")
salad pizza cake
>>> x.new_meal_eaten_starter()
eaten-salad pizza

当然我想看的是吃沙拉披萨蛋糕。 我可以通过重写方法来使用基本方法来解决这个问题：

    def new_meal_eaten_starter(self):
        new_meal_no_dessert = super().new_eaten_starter()
        new_meal = MealWithDessert(self.dessert, new_meal_no_dessert.starter, main=new_meal_no_dessert.main)

    return new_meal

... 但这该死的丑陋。或者通过基本上复制基本方法（违反 DRY）：

def new_meal_eaten_starter(self):
      return MealWithDessert(self.dessert, "Eaten-%s" % self.starter, main=self.main)

...对于更复杂的功能，这两个选项都非常可怕。肯定有更好的方法吗？

Answer 1

下面的sn-p实现了你想做的，希望对你有帮助

class Meal(object):
    def __init__(self, starter, main="steak"):
            self.starter = starter
            self.main = main

    def new_meal_same_starter(self, new_main):
        meal_class = globals()[self.__class__.__name__]
        new_args = self.__dict__.copy()
        new_args['starter'] = "Eaten-%s" % new_args['starter']
        new_args['main'] = new_main
        return meal_class(**new_args)

    def __repr__(self):
        return "%s %s" % (self.starter, self.main)



class MealWithDessert(Meal):
    def __init__(self, dessert, *args, **kwargs):
            super(MealWithDessert, self).__init__(*args, **kwargs)
            self.dessert = dessert

    def __repr__(self):
        return "%s %s %s" % (self.starter, self.main, self.dessert)


x = MealWithDessert("cake", "salad", "pizza")
print(x)
y = x.new_meal_same_starter("tofu")

a = Meal("Rice")
b = a.new_meal_same_starter("Bread")

print(a)
print(b)

print(y)

print(x)

Answer 2

我认为这里的问题是课堂用餐中的以下方法

   def new_meal_same_starter(self, new_main):
       return Meal(self.starter, main=new_main)

你只需要用 new_main 更新主菜。所以试试这个：

   def new_meal_same_starter(self, new_main):
       self.main=new_main

如果需要从方法中返回，则覆盖它：

   def new_meal_same_starter(self, new_main):
      return MealWithDessert(self.dessert, self.starter, main=new_main)