标签：装饰器python
# -*- coding:utf-8 -*- import time login_state = False user_dict = {'username': None} def register(): # 注册函数 while True: username = input("请输入注册账号：").strip() password = input("请输入注册密码：").strip() with open("register", encoding="UTF-8")as reg: for i in reg: list_reg = i.strip().split(',') if username == list_reg[0]: print("用户名已经存在，请重新输入") break else: with open("register", encoding="UTF-8", mode="a")as ligo: ligo.write('\n{},{}'.format(username, password)) print("注册成功") return True def login(): # 登陆函数 global login_state global user_dict i = 0 # 计数器 while i < 3: # 超过3次后，登陆失败 username = input("请输入您的账号:").strip() password = input("请输入您的密码:").strip() # 去除空格及换号符 with open('register', encoding='UTF-8')as f1: for line in f1: # 循环读取注册文件中的内容 line_list = line.strip().split(',') if username == line_list[0] and password == line_list[1]: print("*******登陆成功*******") login_state = True user_dict = line_list[0] return True else: print("账户或密码输入错误") i += 1 def log(e): def loge(): log_time = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime()) with open("log", encoding="UTF-8", mode="a") as logg: logg.write('\n{},{}登陆了,执行了{}函数'.format(log_time, user_dict, e.__name__)) e() return loge def jian_ce(f): # 认证函数，检测用户是否登陆 def inner(): global login_state if login_state == False: print("您尚未登陆，请先登陆后在执行程序") login() else: f() return inner def logoff(): # 注销函数 global login_state if login_state == True: login_state = False print("注销成功") return login_state else: print("您尚未登陆，不需要注销") @jian_ce # 语法糖，装饰器 @log def wen_zhang(): print("这是文章页面") @jian_ce # 语法糖，装饰器 @log def ri_ji(): print("这是日记页面") @jian_ce # 语法糖，装饰器 @log def ping_lun(): print("这是评论页面") @jian_ce # 语法糖，装饰器 @log def shou_cang(): print("这是收藏页面") xu_dict = { # 定义了一个字典，存放序列号对应函数 1: login, 2: register, 3: wen_zhang, 4: ri_ji, 5: ping_lun, 6: shou_cang, 7: logoff } while True: print('''-------欢迎来到博客园------- 1:请登陆 2:请注册 3:文章页面 4:日记页面 5:评论页面 6:收藏页面 7:注销 8:退出程序 ''') xu_hao = input("请输入序列号：").strip() if xu_hao.isdigit(): xu_hao = int(xu_hao) if xu_hao > 0 and xu_hao <= len(xu_dict): xu_dict[xu_hao]() elif xu_hao == 8: break print("再见！退出程序成功") else: print("你输入的序号不存在") else: print("您输入的序列号存在非法字符")标签：装饰器python
原文地址：http://blog.51cto.com/11258494/2122229
