这个装饰器怎么写呀，怎么感觉都不对。

def a ():
    def ap():
        print('前面')
        b()
        print('后面')
    return ap

@a
def b (c,d):
    if d ==1 :
        print(c)
    elif d==2:
        print(c+1)
b(5,1)

1. def a(func):
   
2.     def ap(*args, **kwargs):
   
3.         print('前面')
   
4.         func(*args, **kwargs)
   
5.         print('后面')
   
6.     return ap
   
7. 
   
8. @a
   
9. def b (c,d):
   
10.     if d ==1 :
    
11.         print(c)
    
12.     elif d==2:
    
13.         print(c+1)
    
14. b(5,1)

复制代码