python中or_在python类上重写__or__运算符

作为一个人为的例子,假设我在python中生成一个随机的水果篮.我创造了篮子：

basket = FruitBasket()

现在我想指定篮子中可能出现的特定水果组合.假设我是一个非常挑剔的家伙,篮子要么必须装满苹果和石榴,橙子和葡萄柚,要么只有香蕉.

我正在阅读python运算符重载,似乎我可以定义__or__和__and__来获得我想要的行为.我想我可以这样做：

basket.fruits = (Apple() & Pomegranate()) | (Banana()) | (Orange() & Grapefruit())

这可以很好地制作两个类(Or和And).当__or__或__and__被调用时,我只返回一个新的Or或And对象：

def __or__(self, other):

return Or(self, other)

def __and__(self, other):

return And(self, other)

我想弄清楚的是,如何在不首先实例化水果的情况下做到这一点？为什么我不能在基类Fruit类上使用静态__or__方法？我试过这个,但它不起作用：

class Fruit(object):

@classmethod

def __or__(self, other):

return Or(self, other)

并指定水果：

basket.fruits = (Apple & Pomegranate) | (Orange & Grapefruit) | (Banana)

我收到这样的错误：

TypeError: unsupported operand type(s) for |: 'type' and 'type'

有关如何使这项工作的任何想法？

解决方法:

__or__查找对象的类型;对于一个水果实例,那将是水果;对于水果,这是类型.但是,您可以使用元类更改Fruit的类型：

class FruitMeta(type):

def __or__(self, other):

return Or(self, other)

class Fruit(object):

__metaclass__ = FruitMeta

(对于Python 3,语法是类Fruit(metaclass = FruitMeta)：相反.)

这样就完成了你想要的一切. Apple |香蕉(假设这两个是Fruit的子类)将产生Or(Apple,Banana).

但是,要非常小心这种设计.它正在趋向魔法领域,很容易造成混乱.

(在Python 2.7中完成演示:)

>>> class Or(object):

... def __init__(self, a, b):

... self.a = a

... self.b = b

... def __repr__(self):

... return 'Or({!r}, {!r})'.format(self.a, self.b)

...

>>> class FruitMeta(type):

... def __or__(self, other):

... return Or(self, other)

...

>>> class Fruit(object):

... __metaclass__ = FruitMeta

...

>>> class Apple(Fruit): pass

...

>>> class Banana(Fruit): pass

...

>>> Apple | Banana

Or(, )

标签：python,operator-overloading

来源： https://codeday.me/bug/20190620/1245345.html