問題描述

class Classname(object): def __init__(self, p1, p2=’’):self.p1 = p1self.p2 = p2 # @Classname.decorator def method_one(self, p_list):return function_one(p_list) def method_one(self, p_list):return function_two(p_list) def method_one(self, p_list):return function_three(p_list)

其中大量都是這樣的method_one調(diào)用funciton_one,但是現(xiàn)在要根據(jù)用戶是否傳入p2來改變調(diào)用function_xxx的方法,希望改成這個(gè)樣子function_xxx的代碼我無法修改.

def method_two(self, p_list):if self.p2: return function_two(self.p2, p_list)else: return function_two(p_list)

考慮過使用修飾器來處理,但是使用修飾器來處理話似乎只能在函數(shù)外面包一層,而無法侵入性的改變調(diào)用方法,請(qǐng)問此處有什么合適的解決方法嗎?如果以后還可能根據(jù)self.p3 self.p4來修改調(diào)用方法,有什么更好的解決方法嗎?

問題解答

你可以使用元類魔改類，這里放出個(gè)例子

def function_one(*args): print(1, args)def function_two(*args): print(2, args)def make_method(func): # 此處填邏輯 def _method(self, plist):func(plist) return _method# 元類工廠方法，傳入包含 function_xxx 的模塊對(duì)象def meta(mod): class Meta(type):def __new__(cls, name, bases, attrs): fnames = attrs.get(’FUNCTIONS’, []) for n in fnames:func = getattr(mod, ’function_’ + n)attrs[’method_’+n] = make_method(func) return super(Meta, cls).__new__(cls, name, bases, attrs) return Metaimport sysmyself = sys.modules[__name__]class Class(metaclass=meta(myself)): FUNCTIONS = [’one’, ’two’]obj = Class()obj.method_one(’one’)obj.method_two(’two’)