我有一个处理非交互式绘图的类,作为Matplotlib的前端。但是,有时需要进行交互式绘图。仅使用几个函数,我发现我能够增加图形数量,手动调用绘制等,但是我需要在每次绘制调用之前和之后执行这些操作。因此,要创建交互式绘图包装器和屏幕外绘图包装器,我发现通过元类包装适当的方法来执行此操作比执行以下操作更有效:
class PlottingInteractive:
add_slice = wrap_pylab_newplot(add_slice)
这种方法不能跟上API的更改等等,但是__init__
在重新设置类属性之前对类属性进行迭代的一种方法效率更高,并且可以保持最新状态:
class _Interactify(type):
def __init__(cls, name, bases, d):
super(_Interactify, cls).__init__(name, bases, d)
for base in bases:
for attrname in dir(base):
if attrname in d: continue # If overridden, don't reset
attr = getattr(cls, attrname)
if type(attr) == types.MethodType:
if attrname.startswith("add_"):
setattr(cls, attrname, wrap_pylab_newplot(attr))
elif attrname.startswith("set_"):
setattr(cls, attrname, wrap_pylab_show(attr))
当然,也许有更好的方法可以做到这一点,但是我发现这是有效的。当然,这也可以在__new__
或中完成__init__
,但这是我发现最直接的解决方案。