10-12 记住喜欢的数字

将练习10-11中的两个程序合而为一。如果存储了用户喜欢的数字，就向用户显示它，否则提示用户输入他喜欢的数字并将其存储到文件中。 运行这个程序两次，看看它是否像预期的那样工作。

import json


def add_number():
    filename = 'favorite.json'
    try:
        with open(filename) as f:
           number = json.load(f)
    except FileNotFoundError:
        return None
    else:
        return number


def save_number():
    filename = 'favorite.json'
    number = input("Please input your favorite number:")
    with open(filename, 'w') as f:
        json.dump(number, f)
        return number


def fav():
    number = add_number()
    if number:
        print("I know your favorite number!It's " + str(number))
    else:
        number = save_number()
        print("ok")


fav()