带类方法的类装饰器

2 投票
2 回答
733 浏览
提问于 2025-04-16 18:43

我写了一个类装饰器,这个装饰器会修改一个类,覆盖它的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)

撰写回答