带类方法的类装饰器
我写了一个类装饰器,这个装饰器会修改一个类,覆盖它的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 个回答
1
使用 @staticmethod 装饰器。
@staticmethod
def staticMethod() : ...
3
@staticmethod
def staticMethod (): print "I am static"
或者
def staticMethod (): print "I am static"
cls.staticMethod = staticmethod(staticMethod)