Python中如何將數據存儲為json格式的文件(續)

將上一篇中的例子，修改一下，將兩個程序合二為一，如果存儲了用戶喜歡的水果就顯示它，否則提示用戶輸入他喜歡的水果並將其存儲到文件中。

favorite.py

import json

filename = "favorite_fruit.json"

def greet_user():
    """問候用戶，並指出他喜歡的水果"""
    fruit = get_saved_fruit()
    if fruit:
        print("I know your favorite fruit !  It's " + fruit)
    else:
        fruit = save_user_new_fruit()
        print("I know your favorite fruit !  It's " + fruit)


    

def save_user_new_fruit():
    """提示用戶輸入喜歡的水果"""
    fruit = input(" What is your favorite fruit ? ")
    with open(filename,'w') as  file_obj:
        json.dump(file_obj,fruit)
    return fruit

def get_saved_fruit():
    """如果存儲了用戶喜歡的水果，就獲取它"""
    try :
        with open(filename) as  file_obj:
            fruit = json.load(file_obj)
    except FileNotFoundError:
        return None
    else:
        return fruit
        
greet_user()

解釋：在優化代碼過后，每個函數只需要執行單一任務，這更符合程序的設計

第一次運行程序后的控制台如下：

如果我們是第一次運行，如下圖

會讓用戶先輸入最喜歡的水果然后回應用戶他最喜歡的水果。

而之后的運行，則如圖：

因為程序在第一次運行之后已經將用戶喜歡的水果存入了json文件。

所以不會提示用戶輸入。