Python3 一個類別中存在大量結構接近的方法,能否用修飾器來避免大量代方法的機械性的改寫?

Question

class Classname(object):
    def __init__(self, p1, p2=''):
        self.p1 = p1
        self.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呼叫funciton_one,但是現在要根據使用者是否傳入p2來改變呼叫function_xxx的方法,希望改成這個樣子
function_xxx的程式碼我無法修改.

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

考慮過使用修飾器來處理,但是使用修飾器來處理話似乎只能在函數外麵包一層,而無法侵入性的改變調用方法,請問此處有什麼合適的解決方法嗎?
如果以後還可能根據self.p3 self.p4來修改呼叫方法,有什麼更好的解決方法嗎?

Answer 1

你可以用元類魔改類，這裡放出一個例子

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 的模块对象
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 Meta

import sys

myself = sys.modules[__name__]

class Class(metaclass=meta(myself)): 
    FUNCTIONS = ['one', 'two']

obj = Class()
obj.method_one('one')
obj.method_two('two')