structured_data._conditional_method

Complexity

Total Complexity

8

Size/Duplication

Total Lines	36
Duplicated Lines	0 %

Importance

Changes

0

5 Methods

Rating	Name	Duplication	Size	Complexity
A	ConditionalMethod.__set_name__()	0	3	1
A	ConditionalMethod.__get__()	0	5	3
A	ConditionalMethod.__delete__()	0	3	1
A	ConditionalMethod.__init__()	0	3	1
A	ConditionalMethod.__set__()	0	3	1

import functools

from . import _attribute_constructor


class ConditionalMethod:
    name = None
    __objclass__ = None

    def __init__(self, source, field_check):
        self.source = source
        self.field_check = field_check

    def __set_name__(self, owner, name):
        self.__objclass__ = owner
        self.name = name

    def __get__(self, instance, owner):
        if getattr(owner, self.field_check):
            return getattr(self.source, self.name).__get__(instance, owner)
        target = owner if instance is None else instance
        return getattr(super(self.__objclass__, target), self.name)

    def __set__(self, instance, value):
        # Don't care about this coverage
        raise AttributeError  # pragma: nocover

    def __delete__(self, instance):
        # Don't care about this coverage
        raise AttributeError  # pragma: nocover


def conditional_method(source):
    return _attribute_constructor.AttributeConstructor(
        functools.partial(ConditionalMethod, source)
    )