码迷,mamicode.com
首页 > 其他好文 > 详细

装饰器(一)

时间:2019-04-05 23:51:36      阅读:176      评论:0      收藏:0      [点我收藏+]

标签:eric   def   开放   wrong   password   home   user   too   原来   

一个完整的程序需要一个团队很多人的努力,当你的上一位程序员发给你一段没有问题的代码时,你要在这段代码的基础上添加功能,而且要遵循一个原则----开放-封闭原则。

开放:对现有功能扩展开放

封闭:已经实现功能的代码不应该被修改

假如,现在已经有了一段上一位程序员写好的代码

def jap():
    print("----日韩专区----")

def america():
    print("----欧美专区----")

def homemade():
    print("----国产专区----")

def cartoon():
    print("----动漫专区----")

现在因为日韩和国产的访问量较大,需要用户登录才能继续观看,所以要做到一下功能:

1.点击日韩或国产需要进入登录界面。

2.进入日韩会弹出登录界面,再进入国产显示登录成功不需要再登录,反之亦然。

 

你这样写代码可以实现这个功能:

user_state = False
name = "sunjing"
password = "123"
def login():
global user_state
if user_state == False:
user_name = input("name:")
user_password = input("password:")
if user_name == name and user_password == password:
print("login success")
user_state = True
else:
print("wrong name or password")
if user_state:
print("login success")
def jap():
login()
print("----日韩专区----")

def america():
print("----欧美专区----")

def homemade():
login()
print("----国产专区----")

def cartoon():
print("----动漫专区----")

jap()      
homemade()

结果:

name:sunjing
password:123
login success
login success
----日韩专区----
login success
----国产专区----

发现调用jap()后再调用homemade()就不用再登陆了,实现了预想的功能,但是违反了不修改原来代码的原则于是再对程序进行修改

user_state = False
name = "sunjing"
password = "123"
def login(func):
    global user_state
    if user_state == False:
        user_name = input("name:")
        user_password = input("password:")
        if user_name == name and user_password == password:
            print("login success")
            user_state = True
        else:
            print("wrong name or password")
    if user_state:
        print("login success")
def jap():
    print("----日韩专区----")

def america():
    print("----欧美专区----")

def homemade():
    print("----国产专区----")

def cartoon():
    print("----动漫专区----")

jap = login(jap)
homemade = login(homemade)

结果:

name:sunjing
password:123
login success
login success
login success

修改后程序能实现上述功能但是发现没有调用函数,函数却自己运行了,而且改变了原来函数jap()的调用方式,所以这种方式还是不对,但是可以在上述代码的基础上加以修改

user_state = False
name = "sunjing"
password = "123"
def login(func):
    def inner():
        global user_state
        if user_state == False:
            user_name = input("name:")
            user_password = input("password:")
            if user_name == name and user_password == password:
                print("login success")
                user_state = True
            else:
                print("wrong name or password")
        if user_state:
            func()
            print("login success")
    return inner
def jap():
    print("----日韩专区----")

def america():
    print("----欧美专区----")

def homemade():
    print("----国产专区----")

def cartoon():
    print("----动漫专区----")

jap = login(jap)            #把jap当做一个参数传给login,此时login(jap)就是inner函数的内存地址,再把内存地址赋给jap
homemade = login(homemade)  

jap()                       #jap()相当于调用inner()
homemade()

这样就实现了不修改源程序,不修改调用方式,通过扩展的方式实现了要求的功能。该程序还可以通过别的方式化简:

user_state = False
name = "sunjing"
password = "123"

def login(func1):
    def inner():
        global user_state
        if user_state == False:
            user_name = input("name:")
            user_password = input("password:")
            if user_name == name and user_password == password:
                print("login success")
                user_state = True
            else:
                print("wrong name or password")
        if user_state == True:
            func1()
    return inner
@login                #相当于把@login下面的函数当做参数传入login()函数中,变成login(jap),再把login(jap)变成新函数执行
def jap():
    print("----日韩专区----")

def america():
    print("----欧美专区----")

@login                #@login就相当于jap = login(jap) == inner
def homemade(): print("----国产专区----")

def cartoon():
    print("----动漫专区----")

这就是装饰器的原理

 

 

 

 

 

 

 

 

 

 

 


 

装饰器(一)

标签:eric   def   开放   wrong   password   home   user   too   原来   

原文地址:https://www.cnblogs.com/sunj-96/p/10660259.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!