# Week 13 in-class code
#
# This code builds on the homework from week 11 by saving the contents
# of the dictionary into a file called dictionary.json when the
# program exits, and attempting to read from that file when the
# program runs. It also adds error handling so that the program
# doesn't crash if the JSON file does not exist
#
# This is Python code that would go in a .py file that you would edit
# in Atom and run on the command line with the following command:
#
# $ python3 week12-json.py

import os
import json

dictionary = {}

dictionary["apple"] = "a red or green fruit"
dictionary["banana"] = "a yellow or brown fruit"
dictionary["carrot"] = "an orange vegetable"

# Error handling technique 1:
if os.path.exists("dictionary.json"):
    f = open("dictionary.json","r")
    file_contents = f.read()
    dictionary = json.loads(file_contents)

print( "Welcome to the dictionary.")
num_words = len(dictionary)
print("I know " + str(num_words) + " words." )

user_text = input("Please type a word: ")
while user_text != "exit":

    if user_text in dictionary:
        print("The definition of that word is:")
        print("  " + dictionary[user_text])
    else:
        print("I don't know that word.")
        yn = input("Would you like to add it? y/n")
        if yn == "y":
            definition = input("Great! Type the definition: ")
            dictionary[user_text] = definition
        else:
            print("OK that's fine")

    user_text = input("Please type a word: ")

print("Bye!")

f = open("dictionary.json","w")
output = json.dumps(dictionary)
f.write(output)
f.close()