Налаштування доступу до атрибутів

Середній рівень об'єктно-орієнтованого програмування в Python

Jake Roach

Data Engineer

AttributeError

class Student:
  def __init__(self, student_name, major):
    self.student_name = student_name
    self.major = major
...

karina = Student("Karina", "Literature")
student.residence_hall  # Спроба доступитися до атрибута, якого не існує

...
AttributeError: 'Student' object has no attribute 'residence_hall'

getattr()

# Rest of the class definition above

  def __getattr__(self, name):
    # Implement logic here
    ...

Простір імен об'єкта — це набір атрибутів, пов'язаних із цим об'єктом

__getattr__() виконується, коли робиться спроба звернутися до БУДЬ-ЯКОГО атрибута поза простором імен об'єкта

Магічний метод, напряму не викликається
Приймає параметр name
Реалізує власну логіку замість генерації AttributeError

Усунення AttributeError

class Student:
  def __init__(self, student_name, major):
    self.student_name = student_name
    self.major = major

  def __getattr__(self, name):
      print(f"""{name} does not exist in this object's namespace, try setting 
            a value for {name} first""")

karina.residence_hall  # Тепер знову спробуйте отримати атрибут residence_hall

residence_hall does not exist in this object's namespace, try setting a value for 
  residence_hall first

setattr()

__setattr__() — це магічний метод, що виконується, коли атрибут встановлюється або оновлюється (новий чи наявний)

Включає атрибути, задані через __init__()
Приймає name атрибута та value
Використовує атрибут об'єкта __dict__

Контроль змін атрибутів, валідація, трансформація

# Rest of the class definition above

  def __setattr__(self, name, value):
    # Implement logic here
    ...

    # Use __dict__ to create/update 
    # the attribute
    self.__dict__[name] = value

__dict__ зберігає всі атрибути об'єкта; його можна використовувати для читання й запису даних

Кастомізація зберігання атрибутів

class Student:
  def __init__(self, student_name, major):
    self.student_name = student_name
    self.major = major

  def __setattr__(self, name, value):
    # If value is a string, set the attribute using the __dict__ attribute
    if isinstance(value, str):
      print(f"Setting {name} = {value}")
      self.__dict__[name] = value

    else: # Otherwise, raise an exception noting an incorrect data type
        raise Exception("Unexpected data type!")

setattr у дії

# Set an attribute using a value of type 'str'
karina.residence_hall = "Honors College South"
print(karina.residence_hall)

Setting residence_hall = Honors College South
Honors College South

# Set an attribute using a value of type 'int'
karina.student_id = 19301872

...
    raise Exception("Unexpected data type!")
Exception: Unexpected data type!

Використання getattr та setattr разом

class Student:
    ...   

    def __getattr__(self, name):
        # Set the attribute with a placeholder
        self.__setattr__(name, None)
        return None

    def __setattr__(self, name, value):
        if value is None:  # Print a message denoting a placeholder
            print(f"Setting placeholder for {name}")

        self.__dict__[name] = value  # Set the attribute

Давайте потренуємось!