如何从另一个文件向类添加方法？ [复制]

Question

如何在 python 中从另一个文件向类添加方法？ 例如，我在文件main.py 中有类main，在文件method.py 中有函数something。我怎样才能使main 类具有函数something 作为方法？

Answer 1

您可以通过将函数分配给类的成员名称来完成猴子补丁：

class Main:
    def existing(self): print(f"existing in {type(self)}")

# monkey patch (can be in another file)
def something(self): print(f"something in {type(self)}")
Main.something = something


aMain = Main()

aMain.existing()
aMain.something()

existing in <class '__main__.Main'>
something in <class '__main__.Main'>

这也适用于子类：

class Sub(Main):
    def other(self): print(f"other in  {type(self)}")

aSub  = Sub()
aSub.existing()
aSub.other()
aSub.something()

existing in <class '__main__.Sub'>
other in  <class '__main__.Sub'>
something in <class '__main__.Sub'>

Answer 2

您需要动态添加它（这里是所有数据都在同一个文件中的示例）

def function(p): return p

class A:
   def __init__(self): pass

# add as instance method
setattr(A, 'function', lambda  self, p: function(p))
# add as class method
setattr(A, 'function_cls', classmethod(lambda cls, p: function(p)))
# # add as static method
setattr(A, 'function_static', staticmethod(lambda p: function(p)))

print(a.function('p'))
print(a.function.__class__)

print(A.function_cls('p'))
print(A.function_cls.__class__)

print(A.function_static('p'))
print(A.function_static.__class__)

输出

p
<class 'method'>
p
<class 'method'>
p
<class 'function'>

区别在于类和静态方法也可以像从类调用一样从实例调用