类装饰器与类方法

时间:2011-06-01 02:43:21

标签: python decorator

我编写了一个类装饰器,它修补了一个类,覆盖了 init 并添加了一个方法persist()。到目前为止一切正常。

现在我需要在装饰类中添加一个类方法(静态方法)。我需要在代码中更改什么才能使staticMethod()成为装饰类的静态方法?

这是我的代码:

#! /usr/bin/env python
# -*- coding: utf-8 -*-

class Persistent (object):
    def __init__ (self, table = None, rmap = None):
        self.table = table
        self.rmap = rmap

    def __call__ (self, cls):
        cls.table = self.table
        cls.rmap = self.rmap
        oinit = cls.__init__
        def finit (self, *args, **kwargs):
            print "wrapped ctor"
            oinit (self, *args, **kwargs)
        def persist (self):
            print "insert into %s" % self.table
            pairs = []
            for k, v in self.rmap.items (): pairs.append ( (v, getattr (self, k) ) )
            print "(%s)" % ", ".join (zip (*pairs) [0] )
            print "values (%s)" % ", ".join (zip (*pairs) [1] )
        def staticMethod (): print "I am static"
        cls.staticMethod = staticMethod
        cls.__init__ = finit
        cls.persist = persist
        return cls

@Persistent (table = "tblPerson", rmap = {"name": "colname", "age": "colage"} )
class Test (object):
    def __init__ (self, name, age):
        self.name = name
        self.age = age

a = Test ('John Doe', '23')
a.persist ()
Test.staticMethod ()

输出是:

wrapped ctor
insert into tblPerson
(colage, colname)
values (23, John Doe)
Traceback (most recent call last):
  File "./w2.py", line 39, in <module>
    Test.staticMethod ()
TypeError: unbound method staticMethod() must be called with Test instance as first argument (got nothing instead)

2 个答案:

答案 0 :(得分:3)

    @staticmethod
    def staticMethod (): print "I am static"

    def staticMethod (): print "I am static"
    cls.staticMethod = staticmethod(staticMethod)

答案 1 :(得分:1)

使用@staticmethod装饰器。

@staticmethod
def staticMethod() : ...