Lỗi gặp phải khi sử dụng trình trang trí để cập nhật trình bao bọc

Question

Tôi đã gặp phải thông báo lỗi khá khó hiểu (với tôi ít nhất) khi cố sử dụng trình trang trí để cập nhật trình bao bọc của hàm. Bất kỳ ý tưởng làm thế nào tôi có thể khắc phục điều này?

Tôi đã cố gắng làm cho mã của mình càng chung càng tốt để nó cũng sẽ áp dụng cho các tình huống khác.

def decorator(d): 
    """Make function d a decorator: d wraps a function fn.""" 

    def _d(fn): 
     return functools.update_wrapper(d(fn), fn) 
    functools.update_wrapper(_d, d) 
    return _d 


@decorator 
def f(fn): 
    """Converts the string fn to a function and returns it. 
    Because of the @decorator decorator, _f.__name__ should 
    be identical to f.__name__""" 

    f.__name__ = fn 
    def _f(fn): 
     return eval(fn) 
    return _f 

g = f('x**2') 
print g.__name__ 

đầu ra mong muốn:

>>>x**2 

đầu ra thực tế:

Traceback (most recent call last): 
    File "C:\python\swampy-2.0\testcode.py", line 18, in <module> 
    g = f('x**2') 
    File "C:\python\swampy-2.0\testcode.py", line 6, in _d 
    return functools.update_wrapper(d(fn), fn) 
    File "C:\Python27\lib\functools.py", line 33, in update_wrapper 
    setattr(wrapper, attr, getattr(wrapped, attr)) 
AttributeError: 'str' object has no attribute '__module__' 

Answer 1

Một trang trí có một chức năng như một tham số và trả về một "trang trí" chức năng. Bạn đang truyền một chuỗi và cố gắng trả về một hàm thực sự là một nhà máy chức năng. functools.wraps và functools.update_wrapper mong đợi một chức năng. Một đối tượng hàm sẽ có thuộc tính __module__ trong khi các cá thể của str không có thuộc tính __module__.

Bạn có muốn tạo hàm từ chuỗi "x ** 2" không?

Việc triển khai decorator của bạn là không cần thiết. Chỉ cần sử dụng functools.wraps:

def f(fn): 
    """Converts the string fn to a function and returns it.""" 
    @functools.wraps(fn) 
    def _f(fn): 
     return eval(fn) 
    return _f 

Tuy nhiên, bạn không muốn trang trí trong trường hợp này nhưng là nhà máy chức năng.

def factory(exp): 
    def f(**kwargs): 
     return eval(exp, globals(), kwargs) 
    f.__name__ = exp 
    return f 

Bây giờ bạn có thể sử dụng như thế này:

>>> x_squared = factory("x**2") 
>>> x_squared(x=7) 
49 

Cảnh báo: Surgeon General đã xác định rằng eval là nguy hiểm đến sức khỏe của bạn