Python3 一个类中存在大量结构接近的方法,能否用修饰器来避免大量代方法的机械性的改写?_Python_论坛

Python3 一个类中存在大量结构接近的方法,能否用修饰器来避免大量代方法的机械性的改写?

蜜***园分类：Python 人气：664 回帖：2 发布于6年前收藏

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来修改调用方法,有什么更好的解决方法吗?

标签：python

讨论这个帖子（2）垃圾回帖将一律封号处理……

Lv1 新人

qq***qq JS工程师 6年前#1

如果只是为了替代大量类似方法的机械性改写可以直接把函数名映射一下，比如

class Inst(object):
    def __getattr__(self, attr_name):
        return getattr(self, attr_name.replace('method', 'function'))

这里的replace可以替换成一个dict表或者什么其他的映射表，这样每次调用method_one都会自动映射成function_one,
至于根据是否传入p1参数决定调用哪个函数实现这个要求可以在这个类初始化的时候就把它的函数用另一个函数替换掉

class Inst(object):
    def __init__(self, p1):
        if p1:
            self.function_one = types.MethodType(function_two, self)

这样以后调用function_one时实际上都会变成调用function_two,而且这个替换仅在该实例有效，没有副作用

Lv2 入门

麦***气 JAVA开发工程师 6年前#2

你可以使用元类魔改类，这里放出个例子

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')

文明上网，理性发言！ 😉 阿里云幸运券，戳我领取