如何在类内部装饰方法？ [英] How to decorate a method inside a class?

问题描述

我正在尝试装饰类中的方法，但是python抛出错误。我的班级看起来像这样：

I am attempting to decorate a method inside a class but python is throwing an error. My class looks like this:

from pageutils import formatHeader

class myPage(object):
   def __init__(self):
      self.PageName = ''

   def createPage(self):
      pageHeader = self.createHeader()

   @formatHeader   #<----- decorator
   def createHeader(self):
       return "Page Header ",self.PageName

if __name__=="__main__":
   page = myPage()
   page.PageName = 'My Page'
   page.createPage()

---

pageutils.py ：

def formatHeader(fn):
   def wrapped():
       return '<div class="page_header">'+fn()+'</div>'
   return wrapped

Python抛出以下错误

Python throws the following error

self.createHeader()
TypeError: wrapped() takes no arguments (1 given)

推荐答案

Python自动将类实例作为引用传递。 （ self 自变量在所有类方法中都可见）。

Python automatically passes the class instance as reference. (The self argument which is seen in all class methods).

您可以这样做：

def formatHeader(fn):
    def wrapped(self=None):
        return '<div class="page_header">'+fn(self)+'</div>'
    return wrapped

这篇关于如何在类内部装饰方法？的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！