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 == '123':
                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 = '1234'
       global user_status
       if user_status == False:
           username = input('username>>>')
           password = input('password>>>')
           if username == acc_user and password == '1234':
               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')