如何在类中为递归函数实现装饰器

时间:2019-04-24 19:53:36

标签: python recursion decorator

我正在编写一个在初始化时采用整数输入列表的类。该类具有许多排序方法。我想添加一个装饰器,该装饰器将在每次方法调用之前对输入列表进行混洗。尝试实现递归气泡排序时,装饰器会导致RecursionError: maximum recursion depth exceeded in comparison

我试图传递self参数,以便装饰器可以访问class变量。但是我需要有关如何让递归函数与装饰器一起工作的帮助

import functools
from searching import timer
import random


def shuffle(func):
    @functools.wraps(func)
    def wrapper(self, *args, **kwargs):
        random.shuffle(self.ip_list)
        value = func(self, *args, **kwargs)
        return value
    return wrapper


class sorting:
    def __init__(self, ip_list):
        self.ip_list = ip_list
        self.n = len(self.ip_list)
        self.timer_dict = {}

    @shuffle
    @timer
    def recursive_bubble_sort(self):
        print(self.ip_list)
        for j in range(self.n):
            try:
                if self.ip_list[j] > self.ip_list[j+1]:
                    self.ip_list[j], self.ip_list[j + 1] = self.ip_list[j + 1], self.ip_list[j]
                    self.recursive_bubble_sort()
            except IndexError:
                pass
        print(self.ip_list)


x = [i for i in range(0,30)]
s = sorting(x)
s.recursive_bubble_sort()

1 个答案:

答案 0 :(得分:3)

装饰像您的示例中那样的递归方法是一个非常糟糕的主意。对于某些方法和装饰器,它可以工作,但不能使用排序算法。问题在于,每个递归调用最终都会通过装饰器的包装器进行调用。使用shuffle装饰器,这意味着您将在每个递归调用中重新排列列表,这就是为什么列表永远不会得到排序的原因。即使不会在每次调用时都重新排序,您的timer装饰器也可能会遇到类似的问题,因为它将尝试计时每个递归调用,而不仅仅是对函数的顶级调用。 >

一个选项可能是将递归方法和修饰方法分开。这通常是为将要通过递归实现的功能设计API的好方法,因为您经常需要将额外的参数传递给递归调用,但顶级调用不需要它们。

@shuffle
@timer
def bubble_sort_recursive(self):        # despite the name, this function is not recursive itself
    self.bubble_sort_recursive_helper()

def bubble_sort_recursive_helper(self): # all the recursion happens in this helper method
    ... # recursive code goes here, recursive calls should be to the helper!