python3利用元类批量给所有继承类增加装饰器

python3利用元类批量给所有继承类增加装饰器

这是之前写的代码了,这次给传到csdn上面,大家有什么更好的建议可以回复我

不多说了,贴代码

# -*- coding: utf-8 -*-
# author:CY
# datetime:2019/4/25 16:37
import six
import types
import inspect
from functools import wraps
from collections import OrderedDict

def xx(func):
    @wraps(func)
    def wrap(*args,**kwargs):
        print('*'*150)
        print('执行了decorator')
        res=func(*args,**kwargs)
        if res:
            return res
        print('执行完毕')
        print('*'*150)
    return wrap


class Meta(type):

    @classmethod
    def options(cls,bases,attrs):
        new_attrs=OrderedDict()
        #循环自己的所有属性
        for key,value in attrs.items():
        	#对各种类型的方法进行分别处理
            if hasattr(value,'__func__') or isinstance(value,types.FunctionType):
                if isinstance(value,staticmethod):
                    new_attrs[key]=staticmethod(cls.func(value.__func__))
                elif isinstance(value,classmethod):
                    new_attrs[key] = classmethod(cls.func(value.__func__))
                elif isinstance(value, property):
                            new_attrs[key] = property(fget=cls.func(value.fget), fset=cls.func(value.fset), fdel=cls.func(value.fdel),doc=value.__doc__)
                elif not key.startswith('__') :
                    new_attrs[key]=cls.func(value)
                continue
            new_attrs[key]=value
		#循环所有继承类
        for base in bases:
            for key,value in base.__dict__.items():
                if key not in new_attrs:
                    if hasattr(value, '__func__') or isinstance(value,types.FunctionType):
                        if isinstance(value, staticmethod):
                            new_attrs[key] = staticmethod(cls.func(value.__func__))
                        elif isinstance(value, classmethod):
                            new_attrs[key] = classmethod(xx(value.__func__))
                        elif isinstance(value, property):
                            new_attrs[key] = property(fget=cls.func(value.fget), fset=cls.func(value.fset), fdel=cls.func(value.fdel),doc=value.__doc__)
                        elif not key.startswith('__'):
                            new_attrs[key] = xx(value)
                        continue
                    new_attrs[key] = value

        return new_attrs

    def __new__(cls, name, bases, attrs):
    	"""
    	name:类名,
    	bases:类所继承的父类
    	attrs:类所有的属性
    	"""
        cls.func=attrs.get('meta_decoator')
        assert inspect.isfunction(cls.func),(
            '传入的meta装饰器不正确'
        )
        #在类生成的时候做一些手脚
        new_attrs = cls.options(bases, attrs)
        return super().__new__(cls,name,bases, new_attrs)
        
class obj(object):

    def __init__(self):
        print('obj.__init__')

    @classmethod
    def one(self):
        print('obj.one')

@six.add_metaclass(Meta)
class obj1(obj):
	#只要继承类中有meta_decoator属性,这个属性的方法就会自动装饰下面所有的方法
	#包括类属性,实例属性,property属性,静态属性
    meta_decoator=xx
    aa=1
    @classmethod
    def three(cls):
        print('obj1.three')

    @staticmethod
    def four():
        print('obj1.four')

    def two(self):
        print(self.pro)
        print('obj1.two')

    @property
    def pro(self):
        return 1

b=obj1()
b.one()
b.two()
b.three()
b.four()

a=obj1
a.four()
a.three()

Console输出:

obj.__init__
******************************************************************************************************************************************************
执行了decorator
obj.one
执行完毕
******************************************************************************************************************************************************
******************************************************************************************************************************************************
执行了decorator
1
obj1.two
执行完毕
******************************************************************************************************************************************************
******************************************************************************************************************************************************
执行了decorator
obj1.three
执行完毕
******************************************************************************************************************************************************
******************************************************************************************************************************************************
执行了decorator
obj1.four
执行完毕
******************************************************************************************************************************************************
******************************************************************************************************************************************************
执行了decorator
obj1.four
执行完毕
******************************************************************************************************************************************************
******************************************************************************************************************************************************
执行了decorator
obj1.three
执行完毕
******************************************************************************************************************************************************