随笔- 116 文章- 0 评论- 10 阅读- 10万

10-13 验证用户

1. 项目

最后一个 remember_me.py 版本假设用户要么已输入其用户名，要么是首次运行该程序。

我们应修改这个程序，以应对这样的情形：当前和最后一次运行该程序的用户并非同一个人。

为此，在 greet_user()中打印欢迎用户回来的消息前，先询问他用户名是否是对的。

如果不对，就调用 get_new_username()让用户输入正确的用户名。

2. 代码

import json
 
def get_stored_username():
    """如果存储了用户名，就获取它"""
    filename = "username.json"
    try:
        with open(filename) as f_obj:
            username = json.load(f_obj)
    except FileNotFoundError:
        return None
    else:
        return username
 
def get_new_username():
    """提示用户输入用户名"""
    username = input("What is your name? ")
    filename = 'username.json'
    with open(filename, 'w') as f_obj:
        json.dump(username, f_obj)
    return username
 
def greet_user():
    """问候用户，并指出其名字"""
    username = get_stored_username()
    if username:
        flag = input(f"your name is {username}, yes or no? >")
        if flag == "yes":
            print("Welcome back, " + username + "!")
        elif flag == "no":
            username = get_new_username()
            print("We'll remember you when you come back, " + username + "!")
        else:
            print("input error!")
    else:
        username = get_stored_username()
        print("We'll remember you when you come back, " + username + "!")
 
greet_user()