Python NotImplementedError для атрибутов экземпляра - PullRequest
Новая
       48

Python NotImplementedError для атрибутов экземпляра

1 голос
/ 11 июля 2019

Как пометить атрибут instance как не реализованный в базовом классе? (В отличие от этого вопроса, в котором обсуждается маркировка атрибутов class как не реализованных, но, возможно, я не правильно понимаю базовые классы ...)

например. Я хочу что-то вроде 

class Base():
    def __init__(self):
        self.x = NotImplemented

class GoodSub(Base):
    def __init__(self, x):
        super().__init__()
        self.x = x #good

class BadSub(Base):
    def __init__(self):
       super().__init__()
       #forgot to set self.x

good = GoodSub(5)
bad = BadSub(-1)    
good.x #returns 5
bad.x #throws error because x not implemented

Или есть ли лучший способ заставить все подклассы Base устанавливать атрибут self.x при инициализации?

Редактировать: ссылка на связанный вопрос

Ответы [ 2 ]

2 голосов
/ 12 июля 2019

Одно решение с декораторами и дескрипторами классов (метод __get__): 

def abstract_variables(*args):
    class av:
        def __init__(self, error_message):
            self.error_message = error_message

        def __get__(self, *args, **kwargs):
            raise NotImplementedError(self.error_message)

    def f(klass):
        for arg in args:
            setattr(klass, arg, av('Descendants must set variable `{}`'.format(arg)))
        return klass

    return f


@abstract_variables('x', 'y')
class Base:
    def __init__(self):
        pass

class Derived(Base):
    x = 10


b = Base()
d = Derived()
print(d.x)    # prints 10
print(d.y)    # raises NotImplementedError

Отпечатки: 

10
Traceback (most recent call last):
  File "main.py", line 28, in <module>
    print(d.y)
  File "main.py", line 7, in __get__
    raise NotImplementedError(self.error_message)
NotImplementedError: Descendants must set variable `y`
2 голосов
/ 11 июля 2019

Я бы подумал о том, чтобы сделать x свойством. 

class Base():
    def __init__(self):
        self.__x = None
        self.__x_is_set = False
    @property
    def x(self):
        if not self.__x_is_set:
            raise NotImplementedError('Descendents from Base must set x')
        else:
            return self.__x
    @x.setter
    def x(self, value):
        self.__x = value
        self.__x_is_set = True


class GoodSub(Base):
    def __init__(self):
        super().__init__()
        self.x = 5

class BadSub(Base):
    def __init__(self):
        super().__init__()
        pass

class AlsoBad(Base):
    def __init__(self):
        super().__init__()
        self.__x = 5  # sets the attribute, but not through the property

>>> g, a, b = GoodSub(), BadSub(), AlsoBad()
>>> g.x
5
>>> a.x
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "<stdin>", line 8, in x
NotImplementedError: Descendents from Base must set x
>>> b.x
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "<stdin>", line 8, in x
NotImplementedError: Descendents from Base must set x
...