快捷導航

Python實現(xiàn)帶參數(shù)的用戶驗證功能裝飾器示例

更新時間：2018年12月14日 12:08:31 作者：我是馬克思小清新

這篇文章主要介紹了Python實現(xiàn)帶參數(shù)的用戶驗證功能裝飾器,結(jié)合實例形式分析了Python用戶驗證裝飾器具體定義及使用技巧,需要的朋友可以參考下

本文實例講述了Python實現(xiàn)帶參數(shù)的用戶驗證功能裝飾器。分享給大家供大家參考，具體如下：

user_list = [
  {'name': 'sb1', 'passwd': '123'},
  {'name': 'sb2', 'passwd': '123'},
  {'name': 'sb3', 'passwd': '123'},
  {'name': 'sb4', 'passwd': '123'}
]
# 初始狀態(tài)，用來保存登陸的用戶，
client_dic = {'username': None, 'login': False}
# 添加新功能
def auth(auth_type='filedb'):
  def auth_func(func):
    def wrapper(*args, **kwargs):
      print(auth_type)
      if auth_type == 'fildb':
      # 參數(shù)檢查，判斷是否有用戶登錄，如果有，不用驗證，直接執(zhí)行函數(shù)的功能
        if client_dic['username'] and client_dic['login']:
          res = func(*args, **kwargs)
          return res
        # 輸入用戶名和密碼
        username = input('用戶名:').strip()
        passwd = input('passwd:').strip()
        # 對比列表，檢查用戶名和密碼是否正確
        for user_dic in user_list:
          if username == user_dic['name'] and passwd == user_dic['passwd']:
            client_dic['username'] = user_dic['name']
            client_dic['login'] = True
            res = func(*args, **kwargs)
            return res
        else:
          print('用戶名或者密碼錯誤!')
      elif auth_type == 'pass':
        print('不知道什么驗證方式')
        res = func(*args, **kwargs)
        return res
      else:
        print('一臉蒙蔽的驗證方式')
        res = func(*args, **kwargs)
        return res
    return wrapper
  return auth_func
@auth(auth_type='filedb')
def index():
  print("歡迎來到主頁")
@auth(auth_type='user')
def home(name):
  print("歡迎回家:%s" % name)
@auth(auth_type='pass')
def shoppping_car():
  print('購物車里有[%s,%s,%s]' % ('奶茶', '妹妹', '娃娃'))
print(client_dic)
index()
print(client_dic)
home('root')

運行結(jié)果：