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"]=="":
           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=""
     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=""
     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],""))
          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 =="":
             Register()
         elif qt =="":
             ChangePwd()
         elif qt =="":
             ChangeRole()
         elif qt =="":
            AddUser()
         elif qt=="":
            ShowUserInfo()
         elif qt=="":
             DeleteUser()
         elif qt =="":
             SeachUser()
         elif qt=="q":
          print("再见")
          break

 main()#调用入口函数