Day 3-5 装饰器

开放-封闭原则:

• 封闭：已实现的功能代码块不应该被修改.
• 开放：对现有功能的扩展开放.

装饰器:

定义:在符合'开放-封闭'的原则下,给程序扩展其他的功能!

例：在不更改tokyo函数的情况下。给tokyo函数加上一个验证模块。

 def tokyo():
     print('日本专区'.center(30, '-'))

 def america():
     print('美洲专区'.center(30, '-'))

 def euro():
     print('欧洲专区'.center(30, '-'))

 user_status = False

 def login(func):
     def inner():
         global user_status
         if user_status == False:
             user = input('user:').strip()
             password = input('password:')
             if user == 'alex' and password == '':
                 print('welcome....')
                 user_status = True
         if user_status:
             func()  # 用户通过验证后,执行login传进来的函数.
     return inner   # 把inner的内存地址返回给login(func),所以下面tokyo = login(tokyo)的时候,其实等于tokyo = login(tokyo) = inner
 　　# 相当于用闭包和函数的重新赋值,完成了装饰器.

 tokyo = login(tokyo)
 tokyo()

上面的代码也可以简写成下面的形式:

 user_status =False
 def login(func):

    def inner(*args):   # *args 非固定参数,可以多传或者不传
        acc_user = 'alex'
        acc_pwd = ''
        global user_status
        if user_status == False:
            username = input('username>>>')
            password = input('password>>>')
            if username == acc_user and password == '':
                print('welcome,%s' % username)
                user_status = True
            else:
                print('username or password Error.')
        else:
            print('已通过验证,欢迎您!')
        if user_status == True:
            func(*args)
    return inner  # 把inner的内存地址返回给login().

 def home():
     print('首页'.center(34, '*'))

 def euro():
     print('欧洲专区'.center(30, '*'))

 @login
 def japan(stlye):
     print('日本专区'.center(30, '*'),stlye)

 @login　　　　　　　　　　# 在要加验证的函数前加上@+验证函数.相当于,@login(america).在程序运行时,会默认先运行一次.@login(america)返回的是inner的内存地址,当调用america()时,等价于america()=inner()
 def america():
     print('美洲专区'.center(30, '*'))

 japan('6666') # 如果在这里传参,相当于把这个参数传给inner('666')