python基础学习1-双层装饰器(实现登陆注册)

LOGIN_USER = {"IsLogin":False}

 

def check_login(func): #检查登陆的装饰器
    def inner(*args,**kwargs):
        if LOGIN_USER["IsLogin"]:
            r = func(*args,**kwargs)
            return r
        else:
            print("请先登录!")
    return  inner

 

def check_role(func):#检查权限的装饰器
    def inner(*args,**kwargs):
       if LOGIN_USER["role"]=="2":
          r = func(*args,**kwargs)
          return r
       else:
          print("请用管理员帐户登录!")
    return  inner

 

def login(username,pwd):#登陆函数
    f = open('db1',"r",encoding="utf-8")
    for line in f:
        r= line.split("|")
        if r[0]==username and r[1] == pwd:
            LOGIN_USER["IsLogin"] = True
            LOGIN_USER["username"] = username
            LOGIN_USER["pwd"]=r[1]
            LOGIN_USER["mail"]=r[2]
            LOGIN_USER["phone"]=r[3]
            LOGIN_USER["addr"]=r[4]
            LOGIN_USER["role"]=r[5].replace('\n','')
            break
    f.close()

 

@check_login
def Register():#用户注册函数
    print("欢迎用户%s" %LOGIN_USER["username"])
    uname =input("输入注册用户名称:") #--取得新加注册记录的信息
    upwd = input("请输入注册用户密码")
    umail = input("请输入注册用户邮箱")
    uphone = input("请输入注册用户电话")
    uaddr = input("请输入注册用户地址")
    urole="1"
    fr= open('db1','r',encoding='utf-8') #---读取现有用户记录
    frs = fr.readlines()
    fr.close()
    fw = open('db1','w',encoding='utf-8') #--覆盖写入现有用户记录和新加用户记录
    for line in frs:
      fw.write(line)
    fw.write("\n"+"%s|%s|%s|%s|%s|%s" % (uname,upwd,umail,uphone,uaddr,urole))#--写入新加用户记录
    fw.close()
    print("注册用户成功!")

 

@check_login
def ChangePwd(): #用户修改密码函数
     print("欢迎用户%s" %LOGIN_USER["username"])
     mpw=input("请输入修改的密码:")
     fr= open('db1','r',encoding='utf-8')  #---读取现有用户记录
     frs = fr.readlines()
     fr.close()
     fw = open('db1','w',encoding='utf-8') #--覆盖写入现有用户记录和新加用户记录
     for line in frs:
         ulist= line.split('|')
         if ulist[0]== LOGIN_USER["username"]:
            fw.write("%s|%s|%s|%s|%s|%s" % (ulist[0],mpw,ulist[2],ulist[3],ulist[4],ulist[5]))
         else:
            fw.write(line)
     fw.close()
     print("修改密码成功")

 

@check_login
def ShowUserInfo():#用户查看详细信息函数
    print("欢迎用户%s" %LOGIN_USER["username"])
    fr= open('db1','r',encoding='utf-8')#---读取现有用户记录
    frs = fr.readlines()
    fr.close()

   for line in frs:
         ulist= line.split('|')
         if ulist[0]== LOGIN_USER["username"]:
            uinfo =""" 用户信息
            ------------------------------
            username:%s
            email   :%s
            phone   :%s
            address :%s
            role    :%s
            ------------------------------
            """ %(LOGIN_USER["username"],
                     LOGIN_USER["mail"],
                     LOGIN_USER["phone"],
                     LOGIN_USER["addr"],
                     LOGIN_USER["role"])
    print(uinfo)

 

@check_login #调用双层装饰器 先执行前面的装饰器 先检查登陆 ，在检查是否有权限
@check_role
def DeleteUser():#管理员删除用户信息
     print("欢迎用户%s" %LOGIN_USER["username"])
     duser = input("请输入需要删除用户:")
     fr= open('db1','r',encoding='utf-8')   #---读取现有用户记录
     frs = fr.readlines()
     fr.close()
     fw = open('db1','w',encoding='utf-8')
     for line in frs:
         ulist =line.split('|')
         if ulist[0]==duser:
             continue
         else:
             fw.write(line)
     fw.close()
     print("删除成功!")

 

@check_login#调用双层装饰器
@check_role
def AddUser():#管理员添加用户信息
    print("欢迎用户%s" %LOGIN_USER["username"])
    uname =input("输入用户名称:")#--取得新加注册记录的信息
    upwd = input("请输入用户密码")
    umail = input("请输入用户邮箱")
    uphone = input("请输入用户电话")
    uaddr = input("请输入用户地址")
    urole="1"
    fr= open('db1','r',encoding='utf-8') #---读取现有用户记录
    frs = fr.readlines()
    fr.close()
    fw = open('db1','w',encoding='utf-8')#--覆盖写入现有用户记录和新加用户记录
    for line in frs:
      fw.write(line)
    fw.write("\n"+"%s|%s|%s|%s|%s|%s" % (uname,upwd,umail,uphone,uaddr,urole)) #--写入新加用户记录
    fw.close()
    print("添加用户成功!")

 

@check_login#调用双层装饰器
@check_role
def SeachUser():#管理员根据输入的关键字模糊查询用户信息
     print("欢迎用户%s" %LOGIN_USER["username"])
     search = input("请输入搜索关键字:")
     fr= open('db1','r',encoding='utf-8')#---读取现有用户记录
     frs = fr.readlines()
     fr.close()
     for line in frs:
         ss = line.split('|')
         if search in ss:
             print("用户信息".center(50,'-'))
             uinfo ="""
            username:%s
            email   :%s
            phone   :%s
            address :%s
            role    :%s
            """ %(ss[0],
                     ss[2],
                     ss[3],
                     ss[4],
                     ss[5])
             print(uinfo)
             print("End".center(50,'-'))

 

@check_login#调用双层装饰器
@check_role
def ChangeRole():#管理员提升普通用户权限为管理员权限
     print("欢迎用户%s" %LOGIN_USER["username"])
     gname=input("请输入需提高权限的用户:")
     fr= open('db1','r',encoding='utf-8')  #---读取现有用户记录
     frs = fr.readlines()
     fr.close()
     fw = open('db1','w',encoding='utf-8')#--覆盖写入现有用户记录和新加用户记录
     for line in frs:
         ulist= line.split('|')
         if ulist[0]== gname:
            fw.write("%s|%s|%s|%s|%s|%s" % (ulist[0],ulist[1],ulist[2],ulist[3],ulist[4],"2"))
         else:
            fw.write(line)
     fw.close()
     print("提高权限成功")

 

def main():#程序主入口函数
    while True:
        print("请选择操作模块")
        qt= input( "1.注册 ；2.修改用户密码 ；3.修改用户权限 ；4.添加普通用户；5.查看本用户信息；6.删除用户；7.查询用户信息；Q-退出")
        if not LOGIN_USER["IsLogin"] and qt !="q":
            un = input("用户名:")
            pd = input("密码:")
            login(un,pd)
        if qt =="1":
            Register()
        elif qt =="2":
            ChangePwd()
        elif qt =="3":
            ChangeRole()
        elif qt =="4":
           AddUser()
        elif qt=="5":
           ShowUserInfo()
        elif qt=="6":
            DeleteUser()
        elif qt =="7":
            SeachUser()
        elif qt=="q":
         print("再见")
         break

 

main()#调用入口函数