Как я могу реализовать __add__ метод в родительском классе?

голоса
0

У меня объектно-ориентированного программирования для моделирования геометрических форм. У меня есть добавить метод в каждом классе , если я хочу , чтобы добавить две геометрические фигуры , но я определил в каждом подклассе. Как я могу реализовать метод добавления в родительском классе, так что я делать не определил его для каждых подклассов?

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
Задан 02/12/2019 в 23:54
источник пользователем
На других языках...                            


1 ответов

голоса
1

Это странный вопрос, потому что это на самом деле не имеет смысла , чтобы добавить два круга и есть результат будет новым кругом с суммой радиусов. У вас также есть странное поведение для добавления с , intпотому что вы меняете состояние объекта , а не создавать новую, как вы делаете при добавлении объекта.

Но есть на самом деле способ сделать это с помощью только одного метода в родительском классе, использование некоторых функций Python специфические:

  • cls = self.__class__класс текущего объекта, который может быть использован для создания нового объекта того же класса, и тест , если otherэто правильный тип.
  • d = self.__dict__ это словарь атрибутов объекта.
  • **{ ... }Распаковка оператор позволяет называть clsконструктор с помощью словаря понимания для вычисления аргументов.
  • Я также написал родовой , __repr__который показывает состояние объекта, для удобного тестирования примеров в REPL.

Вот пример:

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

Примеры:

>>> 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)

Обратите внимание, что я изменил поведение всегда возвращает новый объект, а не мутирует существующую.

Ответил 03/12/2019 в 00:08
источник пользователем

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