Блог

У меня есть несколько классов, в которых большинство методов просто «перепрограммируют» вызовы методов self.value и возвращают новый экземпляр:

 class someClass():
    def __init__(self, value):
        self.value = value

    def __add__(self, other):
        return self.__class__( self.value.__add__(other) )

    def someMethod(self, *args, **kwargs):
        return self.__class__( self.value.someMethod(*args, **kwargs) )

    def someOtherMethod(self, *args, **kwargs):
        return self.__class__( self.value.someOtherMethod(*args, **kwargs) )
  

Конечно, не все методы внутри такие, но большинство из них.

Вместо того, чтобы реализовывать явно someMethod , someOtherMethod и __add__ , есть ли способ сделать это систематически? Возможно, с __getattr__ помощью или __getattribute__ ?

Вот почему подкласс типа значения нежизнеспособен :

 >>> class someClass(int):
    pass
>>> a = someClass(5)
>>> isinstance(a 5, someClass)
False #I need this to be True
  

Чтобы заставить эту последнюю строку вернуть True, мне пришлось бы «перепрограммировать» все операторы, как раньше: подклассы вообще не помогают

Частичный ответ, потому что вам все равно нужно перечислить методы, которые вы хотите обернуть, но следующее значительно упрощает перенос большого количества методов, потому что вы просто добавляете их в список методов, которые нужно обернуть:

 class MyClass():

    def __init__(self, value):
        self.value = value
    
    def __str__(self):
        return f'<MyClass({repr(self.value)})>'

    
def make_wrappers(cls, methods):
    for method in methods:
        def wrapper(self, *args, _method=method, **kwargs):
            return self.__class__(getattr(self.value, _method)(*args, **kwargs))
        setattr(cls, method, wrapper)    


make_wrappers(MyClass, ['__add__', '__sub__', 'upper'])

x = MyClass(3)
print(x 2, x-2)  # <MyClass(5)> <MyClass(1)>

y = MyClass('hello')
print(y.upper())  # <MyClass('HELLO')>
  

Не похоже, что вы можете использовать __getattr__ , чтобы делать полностью то, что вы хотите. Это потому, что если вы пытаетесь оценить a b , где a находится экземпляр вашего класса, то, если вы не определили __add__ метод, но определили __getattr__ , то это вызовет TypeError и проигнорирует (не вызовет) __getattr__ . Поэтому вы не можете использовать __getattr__ для переноса «магические методы», такие как __add__ (по крайней мере, в Python 3).

Обратите _method внимание, что здесь используется для обеспечения того, чтобы значение переменной method было привязано к оболочке в момент ее определения (оно добавляется в словарь по умолчанию). Если бы вместо этого вы использовали method непосредственно внутри wrapper where _method is used , то вы обнаружили бы, что каждая оболочка будет использовать значение method , унаследованное от внешней области при его вызове, что было бы не тем, что вы хотите.

Обратите также внимание, что решение основано на «обезьяньем исправлении» класса. Это будет работать. Что не будет работать в Python 3, так это исправлять экземпляры класса — например, если вы попытались использовать setattr для добавления метода непосредственно к x тому, который не существует MyClass , тогда его вызов не будет работать.

Я бы использовал этот очень простой подход:

 class SomeClass():
    def __init__(self, value):
        self.value = value

        for method_name in ("some_method",
                            "some_other_method"):
            new_method = lambda *args, **kwargs: self.__class__(getattr(self.value, method_name)(*args, **kwargs)
            setattr(self, method_name, new_method)
  
     
  

Теперь вам нужен только способ присвоить этим методам строки документации.