Получение класса объекта в Python - варианты и их применение

Расширенные примеры использования

Ниже приведены более сложные сценарии, демонстрирующие получение класса объекта в Python.

Создание класса динамически с помощью type()

Функция type() с тремя аргументами создаёт новый класс. Это мощный приём метапрограммирования.

# Создаём класс с именем 'Person' и одним методом
Person = type('Person', (object,), {'greet': lambda self: 'Hello!'})
p = Person()
print(type(p).__name__)   # Person
print(p.greet())          # Hello!

Person
Hello!

Получение класса в декораторе для логирования

Декоратор может использовать type(self).__name__ для логирования имени класса, в котором вызывается метод.

import functools

def log_call(func):
    @functools.wraps(func)
    def wrapper(self, *args, **kwargs):
        cls_name = type(self).__name__
        print(f'Calling {cls_name}.{func.__name__}')
        return func(self, *args, **kwargs)
    return wrapper

class MyClass:
    @log_call
    def method(self):
        pass

obj = MyClass()
obj.method()

Calling MyClass.method

Проверка иерархии с issubclass для сериализации

При сериализации объектов полезно проверить, что класс объекта является подклассом некоторого базового класса сериализации.

class Serializable:
    pass

class User(Serializable):
    def __init__(self, name):
        self.name = name

class NotSerializable:
    pass

def serialize(obj):
    if issubclass(type(obj), Serializable):
        return {'class': type(obj).__name__, 'data': obj.__dict__}
    else:
        raise TypeError('Object is not serializable')

user = User('Alice')
print(serialize(user))

ns = NotSerializable()
# print(serialize(ns))  # TypeError

{'class': 'User', 'data': {'name': 'Alice'}}

Использование __class__ в методах для создания копий

Метод может создать новый экземпляр того же класса, даже если класс неизвестен заранее, используя self.__class__.

class Point:
    def __init__(self, x, y):
        self.x = x
        self.y = y

    def copy(self):
        return self.__class__(self.x, self.y)

p = Point(1, 2)
p_copy = p.copy()
print(type(p_copy).__name__)  # Point
print(p_copy.x, p_copy.y)      # 1 2

Point
1 2

Получение класса с помощью isinstance и абстрактных базовых классов

Модуль abc позволяет регистрировать классы как виртуальные подклассы, что влияет на isinstance.

from collections.abc import Sequence

class MyList:
    pass

Sequence.register(MyList)

ml = MyList()
print(isinstance(ml, Sequence))  # True
print(type(ml).__name__)         # MyList
print(type(ml).__mro__)          # видно только MyList и object, без Sequence

True
MyList
(<class '__main__.MyList'>, <class 'object'>)

Обратите внимание: type(ml).__mro__ не включает Sequence, хотя isinstance вернул True. Это особенность виртуального наследования.

Получение имени класса для сообщений об ошибках

В пользовательских исключениях можно включать имя класса объекта, вызвавшего ошибку.

class ValidationError(Exception):
    pass

def validate_age(obj):
    if not hasattr(obj, 'age'):
        raise ValidationError(f'Object of class {type(obj).__name__} has no age')

class Person:
    pass

p = Person()
try:
    validate_age(p)
except ValidationError as e:
    print(e)

Object of class Person has no age

Динамическое получение класса через getattr

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

import builtins

class_name = 'int'
cls = getattr(builtins, class_name)
obj = cls(42)
print(type(obj))  # <class 'int'>

<class 'int'>