# -*- coding:utf-8 -*-
# @Time       :2022/6/24 9:42
# @SOFTWARE   :python基础

# 装饰器：在不改变原有代码的前提下，为原函数拓展新功能
"""
    @符号 装饰器的标识符:
        (1)自动把下面修饰的原函数当成参数传递给装饰器
        （2）把返回的新函数去替换原函数
"""
"""
# (1)装饰器的原型
def tuozhan(_func):
    def newfunc():
        print("你好，中国")
        _func()
        print("你好，上海")
    return newfunc
def func():
    print("你好，世界")

func = tuozhan(func) # => func = newfunc
func()
"""

# (2) @符号的使用
"""
def tuozhan(_func):
    def newfunc():
        print("你好，中国")
        _func()
        print("你好，上海")
    return newfunc

@tuozhan
def func():
    print("你好，北京")

func()
"""
# (3)装饰器的嵌套
def tuozhan1(_func):
    def newfunc():
        print("你好，中国1")
        _func()
        print("你好，上海2")
    return newfunc

def tuozhan2(_func):
    def newfunc():
        print("你好，广东3")
        _func()
        print("你好，深圳4")
    return newfunc

@tuozhan2
@tuozhan1
def func():
    print("你好，北京5")

func() # 31524  func => newfunc(tuohzan1) => newfunc(tuozhan2)


# (4)用装饰器拓展带有参数的原函数
print("===========================")
def tuozhan3(func):
    def newfunc(who,where):
        print("你好,老王")
        func(who,where)
        print("你好，小明")
    return newfunc

@tuozhan3
def func(who,where):
    print("{}在{}打游戏".format(who,where))

func("老王","网吧")


# (5)用装饰器拓展带有参数和返回值的原函数
print("<=============================>")
def tuozhan4(func):
    def newfunc(*args,**kwargs):
        print("在网吧打游戏")
        res = func(*args,**kwargs)
        print("洗浴中心捏脚")
        return res
    return newfunc

@tuozhan4
def func(*args,**kwargs):
    lst = []
    for i in args:
        print(i)

    for k,v in kwargs.items():
        strvar = k +":" + v
        lst.append(strvar)
    return lst
lst = func("小明","25","男",name = "老王",age = "16",sex = "男")
print(lst)

# （6）使用类装饰器
print("<=============================>")
class TuoZhan():
    def tuozhan(func):
        def newfunc():
            print("上午打游戏")
            func2()
            print("下午去捏脚")
        return newfunc

# @TuoZhan.tuozhan
def func():
    print("你好，我想睡觉")

# func()

# @TuoZhan()
def func2():
    print("你好，很高兴认识你")
# func2()

# (7)带有参数的函数装饰器
def outer(num):
    def tuozhan(_func):
        def newfunc1(self):
            print("你好，中国")
            _func(self)
            print("你好，世界")
        def newfunc2(self):
            print("你好，北京")
            _func(self)
            print("你好，上海")

        def newfunc3():
            print("你好，广州")
            _func()
            print("你好，江苏")
        if num == 1:
            return newfunc1
        elif num == 2:
            return newfunc2
        elif num == 3:
            return "你好，山西"
    return tuozhan

class MyClass():

    @outer(1)
    def func1(self):
        print("向前一小步")

    @outer(2)
    def func2(self):
        print("文明一大步")

    @outer(3)
    def func3(self):
        print("请瞄准后发射")

obj = MyClass()
obj.func1()
obj.func2()
res = obj.func3
print(res)

# (8)带有参数的类装饰器
"""
    参数1：给修饰的类添加成员属性和方法
    参数2：把类中的run方法变成属性
"""
print("<=============================>")

class TuoZhan():

    ad = "倚天屠龙记"

    def money(self):
        print("神雕侠侣")

    def __init__(self,num):
        self.num = num

    def __call__(self,cls):
        print(cls) # MyClass
        if self.num == 1:
            return self.tuozhan1(cls)
        elif self.num == 2:
            pass
        elif self.num == 3:
            pass
    def tuozhan1(self,cls):
        def newfunc():
            cls.money = TuoZhan.money
            cls.ad = TuoZhan.ad
            return cls()
        return newfunc





@TuoZhan(1) # @obj => MyClass => obj(MyClass)
class MyClass():
    def run(self):
        return "天龙八部"

obj = MyClass()
print(obj.ad)
obj.money()