关于明星投票系统的作业分享

按要求完成明星投票系统的相关功能.  

 功能需求: 1. 用户注册. 将用户名和密码保存在文件中.

　　　　   2. 登录功能. 不需要三次登录. 正常反复无限登录就可以. 直到登录成功为止

　　　　   3. 用户登录后可以查看所有明星信息.  明星信息保存在明星文件中. 

                 4. 给明星进行投票

　　　　   5. 每个人每24小时只能投一次票(扩展功能, 装饰器)

　　首先说下自己的情况，从一开始想先画流程图再去写代码，但是经过半天的不懈努力，我终于放弃这个想法（也不知道是自己太挑了还是不熟悉画图工具最终也没画出什么有价值的，总之对画图的效率，工具都不满意）。然后就开始按照给出的需求写，首先想到有四个主要的函数：注册，登录，显示明星信息，给明星投票。注册和登录没有做优化和扩展很简单就完成了，显示明星信息也很简单就是读取文件，然后给明星投票也是文件操作之修改文件。四个主要功能的函数完事了就开始琢磨函数间的联系及扩展逻辑：

1.注册是需要判断用户名重复（没有难度的）及重复后的相关操作

2.注册成功后是否去登录的操作（日常操作）

3.登录状态问题（字典及装饰器）用来控制显示明星信息及给明星投票

4.登录成功后直接跳转到投票页面（理论上应该是到另一个主菜单只有查看明星、投票及退出）

5.本题升级的难点（用日志记录每次操作投票的时间，其中涉及时间的转换，耗时最多）

6.最后加一个main方法把所有逻辑及方法串起来。

总结下：在目前编程基础不扎实及经验很渣的时候，应该将整体需求化繁为简（彦涛老师课上讲过），将大致功能完善了再开始打磨每个单体的需求及每个个体需求间的联系。

下面是一段花了不知多久（因为老走神，貌似间断的有6小时左右）写出来的垃圾代码（就是完成了需求，其中有很多可以改进的地方），不嫌弃的可以看俩眼。

import os
import time, datetime
from functools import wraps


def register():
    '''
    用来注册，好像是句废话
    :return:
    '''
    flag = True
    while flag:  # 如果重复继续输入要注册的用户名
        user = input('请输入用户名：')
        with open('userinfo', 'r', encoding='utf-8')as f1:
            for i in f1:
                if user == i.split(',')[0]:  # 判断用户名是否重复
                    print('用户名重复，请重新输入！')
                    break
                else:
                    flag = False

    pwd = input('请输入密码：')
    with open('userinfo', 'a', encoding='utf-8')as f2:  # 将用户名及密码写入用户信息中
        f2.write(user + ',' + pwd + '
')  # 最后加个换行符可以保证每个用户为一行的数据规范
    print(f'注册成功，您的用户名是{user}，密码是{pwd}')
    user_choose = input('是否去登录？(y/Y继续，其他任意键退出程序)')
    if user_choose.upper() == 'Y':
        login()
    else:
        exit()


# register()
dic = {'user': None, 'status': None}  # 存储登录状态


def wrapper_login(fn):
    '''
    用来验证是否已经登录
    :param fn: 被装饰的方法
    :return: inner
    '''

    def inner(*args, **kwargs):
        '''
        装饰器的真正执行者
        :param args:
        :param kwargs:
        :return: fn函数执行的结果
        '''
        if dic['status'] == True:  # 用来验证是否登录成功
            ret = fn(*args, **kwargs)
            return ret
        else:
            print('对不起，您还没有登录！')
        login()

    return inner


def login():
    '''
    登录函数
    :return:
    '''
    while True:
        username = input('账号：')
        password = input('密码：')
        with open('userinfo', 'r', encoding='utf-8')as f:
            for i in f:
                user_file, pwd_file = i.strip().split(',')
                if username == user_file and password == pwd_file:
                    print(f'恭喜{username}登录成功！')
                    dic['user'] = username
                    dic['status'] = True
                    vote_star()
                    return
            else:
                print('用户名或者密码错误请重新输入！')


# login()
@wrapper_login
def show_star_info():
    '''
    用来显示明星信息
    :return:
    '''
    with open('star_info', 'r', encoding='utf-8')as f:
        for i in f:
            print(i.strip())


# show_star_info('star_info')
def wrapper_time(fn):
    '''
    用来记录访问日志
    :param fn: 被装饰的方法
    :return: inner
    '''

    @wraps(fn)
    def inner(*args, **kwargs):
        '''
        装饰器的执行者
        :param args:
        :param kwargs:
        :return: fn执行的结果
        '''
        with open('log', 'r', encoding='utf-8')as f2:
            data = f2.readlines()  # 将数据读取到列表中
            str_time_last = data[-1].strip().split('|')[1]  # 用切片找到最后一列的时间
        time_last = time.mktime(time.strptime(str_time_last, "%Y-%m-%d %X"))  # 将最后一次投票的时间戳
        time_start = time.time()
        time_difference = time_start - time_last  # 计算时间差
        if time_difference >= 60:  # 判断时间差
            ret = fn(*args, **kwargs)
            str_time_end = time.strftime("%Y-%m-%d %X")
            with open('log', 'a', encoding='utf-8')as f1:
                f1.write('在|%s|访问了|%s
' % (str_time_end, fn.__name__))
            return ret
        else:
            print('距离上次投票时间不足1分钟，无法进行下次投票！')

    return inner


@wrapper_login
@wrapper_time
def vote_star():
    '''
    用于给明星投票
    :return:
    '''
    show_star_info()
    user_id = input('请选择您要投票的明星编号：')
    with open('star_info', 'r', encoding='utf-8')as f1, 
            open('new_star_info', 'w', encoding='utf-8')as f2:
        for i in f1:
            file_id, file_name, file_poll = i.strip().split('|')
            if str(user_id) == file_id:
                file_poll = str(int(file_poll) + 1)
            f2.write('%s|%s|%s
' % (file_id, file_name, file_poll))
    os.remove('star_info')
    os.rename('new_star_info', 'star_info')
    show_star_info()
    print('稍事休息,1分钟后才可以继续投票！')


def main():
    '''
    系统整个的执行逻辑
    :return:
    '''
    global register, login, show_star_info, vote_star  # 将函数中需要调用的方法全部引用为全局变量
    register, login, show_star_info, vote_star = register, login, show_star_info, vote_star  # 将同名的变量赋值成原来的方法
    print('欢迎进入明细投票系统：')
    tu_menu = ('注册', '登录', '查看明星', '明星投票', '退出系统')
    tu_def = (register, login, show_star_info, vote_star)
    for index, element in enumerate(tu_menu):
        print(f'{index+1}.{element}')
    choose = int(input('请选择您要进行的操作：'))
    for index, element in enumerate(tu_def):  # 此处用循环代替大幅的if语句
        if choose == index + 1:
            element()
            break
        elif choose == 5:
            exit()
    else:
        print('输入有误，请重新输入！')


# vote_star('star_info',3)
if __name__ == '__main__':
    main()

 对于单个用户登录后时间的计算问题重新优化部分：

def wrapper_time(fn):
    '''
    用来记录访问日志
    :param fn: 被装饰的方法
    :return: inner
    '''

    @wraps(fn)
    def inner(*args, **kwargs):
        '''
        装饰器的执行者
        :param args:
        :param kwargs:
        :return: fn执行的结果
        '''
        with open('log', 'r', encoding='utf-8')as f2:
            #     data = f2.readlines()  # 将数据读取到列表中
            #     str_time_last = data[-1].strip().split('|')[2]  # 用切片找到最后一列的时间
            # time_last = time.mktime(time.strptime(str_time_last, "%Y-%m-%d %H:%M:%S"))  # 将最后一次投票的字符串时间转化成时间戳
            lst = []  # 空列表用来放单个用户的访问时间戳
            for i in f2:
                username, str_time_last = i.strip().split('|')[0], i.strip().split('|')[2]  # 拿到俩个关键字段，名字和字符串的时间
                if username == dic['user']:  # 如果日志里的用户名和登录的用户相同，就进行把时间都拿出来
                    stamp_time_last = time.mktime(time.strptime(str_time_last, "%Y-%m-%d %H:%M:%S"))  # 和之前一样转换下字符串的时间变为时间戳
                    lst.append(stamp_time_last)  # 添加到列表里
            else:  # 如果日志里还没有该用户名那就直接访问，并记录日志
                ret = fn(*args, **kwargs)
            str_time_end = time.strftime("%Y-%m-%d %H:%M:%S")  # 获取当前的字符串时间
            with open('log', 'a', encoding='utf-8')as f1:
                f1.write('%s|在|%s|访问了|%s
' % (
                dic['user'], str_time_end, fn.__name__))  # 日志记录 chang|在|2018-06-19 19:34:09|访问了|vote_star
            return ret
        time_last = max(lst)  # 取到列表中最大的时间戳为最后一次的投票时间
        time_start = time.time()  # 访问函数前的时间戳
        time_difference = time_start - time_last  # 计算时间差
        if time_difference >= 60:  # 判断时间差
            ret = fn(*args, **kwargs)
            str_time_end = time.strftime("%Y-%m-%d %H:%M:%S")
            with open('log', 'a', encoding='utf-8')as f1:
                f1.write('%s|在|%s|访问了|%s
' % (dic['user'], str_time_end, fn.__name__))
            return ret
        else:
            print('距离上次投票时间不足1分钟，无法进行下次投票！')

    return inner