《Python编程：从入门到实践》课后习题10-13

《Python编程：从入门到实践》课后习题

10-13 验证用户：最后一个remember_me.py版本假设用户要么已输入其用户名，要么是首次运行该程序。我们应修改这个程序，以应对这样的情形：当前和最后一次 运行该程序的用户并非同一个人。 为此，在greet_user() 中打印欢迎用户回来的消息前，先询问他用户名是否是对的。如果不对，就调用get_new_username() 让用户输入正确的用户名。

import json
""" 重构rember me.py 的尝试"""

def get_username():
    """如果存储了用户名，就读取它"""
    file_name = "username.json"
    try:
        with open(file_name, "r") as f_obj:
            username = json.load(f_obj)
    except FileNotFoundError:
        return None # 如果文件username.json不存在，这个函数就返回None
    else:
        return username

def get_new_username():
    """提示用户输入用户名"""
    username = input("What is your name? ")
    file_name = "username.json"
    with open(file_name, "w") as f_obj:
        json.dump(username, f_obj)
    return username

def greet_user():
    """问候用户，并指出其名字"""
    username = get_username() # 先调用只负责获取存储的用户名函数
    while username: # 如果成功地获取了用户名，就打印一条欢迎用户回来的消息，否则就提示用户输入用户名。
        print("---Please enter 'Yes' or 'No'---")
        answer = input("Are you " + username + "?")
        if answer == "Yes".lower():
            print("Welcome back, " + username + "!")
            break

        elif answer == "No".lower():
            username = get_new_username() # 必要时调用只负责获取并存储新用户的用户名函数
            print("We'll rember you when you come back, " + username + "!")
            break

        else:
            print("Please enter 'Yes' or 'No' !\n")

greet_user()