Como posso implementar __add__ método na classe pai?

votos
0

Eu tenho de modelagem de programação orientada a objetos para formas geométricas. Tenho adicionar método em cada classe se eu quiser adicionar até duas formas geométricas, mas tenho definido em cada subclasse. Como posso implementar o método add na classe pai, para que eu não a definiu para cada subclasses?

import numpy as np

class Shape(object):


    def __repr__(self):
        return type(self).__name__

    def __str__(self):
        return type(self).__name__

class Circle(Shape):
    

    
       # constructor

    def __init__(self, radius):
        self.radius = radius
    def __add__(self, other):
        if type(other) == int:
            self.radius = self.radius + other            
        else: 
            newRadius = self.radius + other.radius
            return Circle(newRadius)
    def __radd__(self, other):
        return self.__add__(other)
    def area(self):
        return np.pi * self.radius**2

class Rectangle(Shape):
    # constructor

    def __init__(self, width,height):
        self.width , self.height = width, height
    def __add__(self, other):
        if type(other) == int:
            self.width = self.width + other  
            self.height = self.height + other  
        else: 
            newWidth = self.width + other.width
            newHeight = self.Height + other.Height
            return Rectangle(newWidth,newHeight)
    def __radd__(self, other):
        return self.__add__(other)


    def area(self):
        
        Function to compute the area of triangle.
        
        return self.width * self.height
Publicado 02/12/2019 em 23:54
fonte usuário
Em outras línguas...                            


1 respostas

votos
1

Esta é uma pergunta estranha, porque ele realmente não faz sentido adicionar dois círculos e ter o resultado ser um novo círculo com a soma dos raios. Você também tem comportamento estranho para adicionar com um int, porque você está mudando o estado do objeto em vez de criar um novo, como você faz quando a adição de um objeto.

Mas há de fato uma maneira de fazer isso com apenas um método na classe pai, o uso de alguns recursos específicos do Python:

  • cls = self.__class__é a classe do atual objeto, que pode ser usado para criar um novo objeto da mesma classe, e testar se otheré o tipo certo.
  • d = self.__dict__ é um dicionário de atributos do objeto.
  • O **{ ... }operador de desembalar permite chamar o clsconstrutor usando uma compreensão dicionário para calcular os argumentos.
  • Eu também escrevi um genérico __repr__que mostra o estado do objeto, para testar convenientemente exemplos na REPL.

Aqui está um exemplo:

class Shape:
    def __add__(self, other):
        cls = self.__class__
        d = self.__dict__
        if isinstance(other, int):
            return cls(**{ k: v + other for k, v in d.items() })
        elif isinstance(other, cls):
            return cls(**{ k: v + other.__dict__[k] for k, v in d.items() })
        else:
            raise TypeError()

    def __radd__(self, other):
        return self.__add__(other)

    def __repr__(self):
        d = self.__dict__
        return '{0}({1})'.format(
            self.__class__.__name__,
            ', '.join('{0}={1!r}'.format(k, v) for k, v in d.items())
        )

class Circle(Shape):
    def __init__(self, radius):
        self.radius = radius

class Rectangle(Shape):
    def __init__(self, width, height):
        self.width, self.height = width, height

Exemplos:

>>> Circle(4) + Circle(5)
Circle(radius=9)
>>> Circle(6) + 2
Circle(radius=8)
>>> 3 + Circle(2)
Circle(radius=5)
>>> Rectangle(2, 3) + Rectangle(4, 5)
Rectangle(width=6, height=8)
>>> Rectangle(2, 3) + 1
Rectangle(width=3, height=4)
>>> 5 + Rectangle(2, 3)
Rectangle(width=7, height=8)

Note que eu mudei o comportamento para retornar sempre um novo objeto, em vez de mutação do já existente.

Respondeu 03/12/2019 em 00:08
fonte usuário

Cookies help us deliver our services. By using our services, you agree to our use of cookies. Learn more