Как создать пользовательское строковое представление для объекта класса?

Рассмотрим этот класс:

class foo(object):
    pass

Строковое представление по умолчанию выглядит примерно так:

>>> str(foo)
"<class '__main__.foo'>"

Как я могу сделать это отображение пользовательской строкой?

Ответ 1

Внесите __str__() или __repr__() в метакласс класса.

class MC(type):
  def __repr__(self):
    return 'Wahaha!'

class C(object):
  __metaclass__ = MC

print C

Используйте __str__, если вы имеете в виду читаемую строкопись, используйте __repr__ для однозначных представлений.

Ответ 2

class foo(object):
    def __str__(self):
        return "representation"
    def __unicode__(self):
        return u"representation"

Ответ 3

Если вам нужно выбрать между __repr__ или __str__ перейдите к первому, так как по умолчанию реализация __str__ вызывает __repr__ когда он не был определен.

Пользовательский пример Vector3:

class Vector3(object):
    def __init__(self, args):
        self.x = args[0]
        self.y = args[1]
        self.z = args[2]

    def __repr__(self):
        return "Vector3([{0},{1},{2}])".format(self.x, self.y, self.z)

    def __str__(self):
        return "x: {0}, y: {1}, z: {2}".format(self.x, self.y, self.z)

В этом примере repr возвращает снова строку, которую можно непосредственно использовать/выполнять, тогда как str более полезен в качестве отладочного вывода.

v = Vector3([1,2,3])
print repr(v)    #Vector3([1,2,3])
print str(v)     #Vector(x:1, y:2, z:3)

Ответ 4

Одобренный ответ Игнасио Васкеса-Абрамса совершенно прав. Это, однако, из поколения Python 2. Обновление для текущего Python 3 будет:

class MC(type):
  def __repr__(self):
    return 'Wahaha!'

class C(object, metaclass=MC):
    pass

print(C)

Если вам нужен код, который работает как на Python 2, так и на Python 3, то шесть модулей помогут вам:

from __future__ import print_function
from six import with_metaclass

class MC(type):
  def __repr__(self):
    return 'Wahaha!'

class C(with_metaclass(MC)):
    pass

print(C)

Наконец, если у вас есть один класс, для которого вы хотите иметь пользовательское статическое repr, вышеописанный подход на основе классов отлично работает. Но если у вас их несколько, вам придется сгенерировать метакласс, похожий на MC для каждого, и это может стать утомительным. В этом случае продвижение метапрограммирования на шаг вперед и создание фабрики метаклассов делает вещи немного чище:

from __future__ import print_function
from six import with_metaclass

def custom_class_repr(name):
    """
    Factory that returns custom metaclass with a class ''__repr__'' that
    returns ''name''.
    """
    return type('whatever', (type,), {'__repr__': lambda self: name})

class C(with_metaclass(custom_class_repr('Wahaha!'))): pass

class D(with_metaclass(custom_class_repr('Booyah!'))): pass

class E(with_metaclass(custom_class_repr('Gotcha!'))): pass

print(C, D, E)

печатает:

Wahaha! Booyah! Gotcha!

Метапрограммирование - это не то, что вам обычно нужно каждый день, но когда вам это нужно, оно действительно попадает в точку!

Ответ 5

Просто добавлю ко всем прекрасным ответам мою версию с оформлением:

from __future__ import print_function
import six

def classrep(rep):
    def decorate(cls):
        class RepMetaclass(type):
            def __repr__(self):
                return rep

        class Decorated(six.with_metaclass(RepMetaclass, cls)):
            pass

        return Decorated
    return decorate


@classrep("Wahaha!")
class C(object):
    pass

print(C)

стандартный вывод:

Wahaha!

Недостатки:

  1. Вы не можете объявить C без суперкласса (без class C:
  2. Экземпляры C будут экземплярами некоторого странного происхождения, поэтому, вероятно, будет хорошей идеей добавить __repr__ для этих экземпляров.