python - ObjA 列表的自定义显示输出

Question

我们可以使用 ipython 显示引擎为 numpy.polynomial.polynomial 注册自定义类型，如下所示

ip = get_ipython()

foramtter = ip.display_formatter.formatters['text/latex']

foramtter.for_type_by_name('numpy.polynomial.polynomial',
                             'Polynomial', display_func)

我想使用 .for_type_by_name(...) 方法为特定类型的列表注册自定义显示，比如 ObjA 而不仅仅是类型 ObjA 本身。

我该怎么做？

顺便说一句，我无权访问返回 ObjA 列表的类。

Answer 1

如何为 list 对象创建一个仅在看到 ObjA 列表时才起作用的格式化程序？

from decimal import Decimal   # Decimal is my ObjA here

ip = get_ipython()
formatter = ip.display_formatter.formatters['text/latex']

def format_list(obj):
    if not isinstance(obj, list):
        return None
    if not all(isinstance(item, Decimal) for item in obj):
        return None
    return "$$[%s]$$" % ";".join(map(str, obj))

formatter.for_type_by_name('builtins', 'list', format_list)

似乎如果格式化函数返回None，则格式化程序将被忽略。至少它对我有用:

In[2]: [Decimal("1"), Decimal("2"), "not a decimal"]
Out[2]: [Decimal("1"), Decimal("2"), "not a decimal"]

In[3]: [Decimal("1"), Decimal("2")]
Out[3] 1, 2  # LaTeX-formatted, yeah

这是一个非常肮脏的 hack，但不幸的是我没有看到任何其他方式(除了猴子修补 DisplayFormatter，它甚至更肮脏，尽管它应该更强大)。如果有，希望有人赐教。