Как написать getatrribute, чтобы переменные в dict могли быть возвращены? - PullRequest
Новая
       32

Как написать getatrribute, чтобы переменные в dict могли быть возвращены?

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

У меня есть класс, и в нем есть диктат. Я хочу, чтобы посетить переменные в dict прямым "instance.key" вместо "instance.d [ключ]". Как это сделать? 

class A:
    def __init__(self):
        self.a = 1
        self.b = 2
        self.fields = {'c': 3, 'd': 4}


def main():
    a = A()
    print(a.a) # 1
    print(a.b) # 2
    print(a.c) # currently raise Exception, I want it to print 3
    print(a.d) # currently raise Exception, I want it to print 4


if __name__ == '__main__':
    main()

Ответы [ 3 ]

6 голосов
/ 04 июля 2019

Вы можете переопределить метод object.__getattr__(self, name) (такие методы могут быть определены для настройки значения доступа к атрибутам для экземпляров классов): 

class A:
    def __init__(self):
        self.a = 1
        self.b = 2
        self.fields = {'c': 3, 'd': 4}

    def __getattr__(self, item):
        return self.__dict__.get(item) or self.__dict__['fields'].get(item)

a = A()
print(a.a) # 1
print(a.b) # 2
print(a.c) # 3
print(a.d) # 4

Ссылка на ссылку: __getattr__

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

Вы также можете изменить __dict__: 

class A:
    def __init__(self):
        self.a = 1
        self.b = 2
        self.fields = {'c': 3, 'd': 4}
        self.__dict__ = {**self.__dict__, **self.fields}

А теперь: 

def main():
    a = A()
    print(a.a)
    print(a.b)
    print(a.c)
    print(a.d)


if __name__ == '__main__':
    main()

Дает: 

1
2
3
4
0 голосов
/ 04 июля 2019

Реализуйте __getattr__, как предложено @RomanPerekhrest.Единственная проблема, с которой вы можете столкнуться, заключается в том, что атрибуты вашего экземпляра переопределяют ваши записи dict.Если это не ожидаемое поведение, вы должны реализовать __getattribute__ вместо __getattr__

Примерно так: 

class A:
    def __init__(self):
        self.a = 1
        self.b = 2
        self.e = 5
        self.fields = {'c': 3, 'd': 4, 'e': 6}

    def __getattribute__(self, key):
        fields = super().__getattribute__('fields')
        if key == 'fields':
            return fields

        try:
            return fields[key]
        except KeyError:
            return super().__getattribute__(key)

a = A()
print(a.a) # 1
print(a.b) # 2
print(a.c) # 3
print(a.d) # 4
print(a.e) # 6 (Note that it returns the entry of the dict rather that the instance attribute)
print(a.f) # Raises exception (not in the instance attributes nor in the dict)
...