python - 如何在 python 返回时强制调用类方法

Question

我有一个ReportEntry类

class ReportEntry(object):
  def __init__(self):
      # Many attributes defined here

  ... # Lot many setattr/getattr here
  def validate(self):
      # Lot of validation code in here
      return self

多个其他类与ReportEntry类保持has-a关系

class A(object):
  def test1(self):
    t1 = ReportEntry()
    # Assign the attribute values to t1
    return t1.validate()

  def test2(self):
    t2 = ReportEntry()
    # Assign the attribute values to t2
    return t2.validate()

并且有多个像A这样的类。

我需要强制每个 ReportEntry 类实例在 return 或 return 之前调用 validate() >.

基本上，ReportEntry 的任何实例都不应逃避验证，因为如果缺少某些内容，最终报告生成将失败。

我怎样才能做到这一点？

Answer 1

你可以写一个类装饰器:

import inspect

def validate_entries(cls):
    def validator(fnc):  # this is a function decorator ...
        def wrapper(*args, **kwargs):
            rval = fnc(*args, **kwargs)
            if isinstance(rval, ReportEntry):
                # print('validating')
                return rval.validate()
            return rval
        return wrapper
    for name, f in inspect.getmembers(cls, predicate=inspect.isfunction):
        setattr(cls, name, validator(f))  # .. that we apply to all functions
    return cls

现在您可以定义所有类似 A 的类:

@validate_entries
class A(object):
    # ...

这将验证任何 A 的方法返回的任何 ReportEntry。