python decorator扩展_python如何扩展一个decorator-CSDN博客

本文链接：https://blog.csdn.net/qq_40666620/article/details/120204811

该博客探讨了Python中函数注册和修饰器的使用。通过`register_func`装饰器实现函数字典动态注册，允许在运行时根据随机数调用不同方法。同时，展示了类修饰器`Components`用于在类初始化时注入属性。此外，还介绍了一个计数器修饰器`counter`，用于记录类实例的数量。

摘要由CSDN通过智能技术生成

扩展1, 接收参数

# coding=utf-8


MOVE_FUNC_DICT = {}


# 相当于一个函数执行, 并返回一个decorator
# 只会初始化的时候执行一次
def register_func(move_type):
	def _inner(fun):
		MOVE_FUNC_DICT[move_type] = fun
		return fun
	return _inner


def decorator(func):
	def __inner(*args, **kwargs):
		print '修饰'
		func(*args, **kwargs)
	return __inner



class Test(object):

	def run(self):
		import random
		while True:
			MOVE_FUNC_DICT[random.randint(0, 2)](self)
			# self.move_a()

	@register_func(0)
	def move_a(self):
		print 'a'

	@register_func(1)
	def move_b(self):
		print 'b'

	@register_func(2)
	def move_c(self):
		print 'c'


if __name__ == '__main__':
	t = Test()
	t.run()

扩展2,修饰类

def Components(*args, **kwargs):
	def _inner(kclass):
		kclass.init_dict(*args, **kwargs)
		return kclass

	return _inner


@Components({1: '1'})
class A(object):
	test = None

	def __init__(self):
		pass

	@classmethod
	def init_dict(cls, _dict):
		cls.test = _dict

a = A()

print a.__class__.test

没有参数, 2就是在3基础上包了一层


def counter(cls):
    obj_list = []
    def wrapper(*args, **kwargs):
        new_obj = cls(*args, **kwargs)
        obj_list.append(new_obj)
        print "class:%s'object number is %d" % (cls.__name__, len(obj_list))
        return new_obj
    return wrapper

@counter
class my_cls(object):
    STATIC_MEM = 'This is a static member of my_cls'
    def __init__(self, *args, **kwargs):
        print self, args, kwargs
        print my_cls.STATIC_MEM

c = my_cls()