import time
import json

Stuff = {}

# TODO:
# Assign Storage Spaces
# Assign Stuff to Storage Spaces
# Menu
# SaveState

SleepTime = 0.1
file = open("Stuffbase.json", "w")


def Menu():
    print("1 : Save the current State")
    time.sleep(SleepTime)
    print("2 : Store something")
    time.sleep(SleepTime)
    print("3 : Take something out")
    time.sleep(SleepTime)
    print("4 : Look for Something")
    time.sleep(SleepTime)
    print("5 : End the Session")
    time.sleep(SleepTime)
    RawInput = input("Your Answer: ")
    Ans = int(RawInput)

    # Selection of What to do
    if Ans == 1:
        Save()
    elif Ans == 2:
        StoreSomething()
    elif Ans == 3:
        RemoveSomething()
    elif Ans == 4:
        print("Not yet implemented")
        Menu()

    elif Ans == 5:
        quit("Session ended")
    else:
        print("No valid input!")
        Menu()


def Save():
    Data = Stuff
    json.dump(Data, file, indent = 1)
    Menu()


def Search():

    Menu()


def StoreSomething():
    Object = input("Object: ")
    Place = input("Place: ")
    Storage = input("Storage: ")
    Status = input("Status: ")
    n = 0
    while Object in Stuff:
        n = n + 1
        Object = Object + str(n)
    Stuff[Object] = {"Place": Place, "Storage": Storage, "Status": Status}
    Menu()


def RemoveSomething():
    Object = input("Object: ")
    if not Object in Stuff:
        print("Object not in the Database!")
    else:
        print(Stuff[Object])
        Stuff[Object] = {"Storage": "Removed"}
    Menu()


Menu()