from snack import Snack

class VendingMachine:

  def __init__(self):
      """
      Constructor. Initializes a list of available snacks (empty to start) and
      calls a function to load the default file of snack information
      """
      self.foods = []
      self.loadSnacks("snacks.txt")

  def loadSnacks(self, filename):
      """
      Loads snack information from a file. Clears the old snacks from the
      machine.  Param filename (string): the file to load Returns: none
      """
      ifile = open(filename, "r")
      for line in ifile:
          tokens = line.strip().split(",")
          print(tokens)
          name = tokens[0]
          quantity = int(tokens[1])
          cost = int(tokens[2])
          desc = tokens[3]
          snack = Snack(name, quantity, cost, desc)
          self.foods.append(snack)

  def listSnacks(self):
      """
      Pretty prints the snacks in the vending machine.
      Hint: Use the following string formatting to match the sample output
          "%d) %-20s %d left\t$%d\t%s"
      Params: none
      Returns: none
      """
      print()
      print("="*30, "Vending Machine", "="*30)
      for i in range(len(self.foods)):
          print(i, self.foods[i])
      print("="*75)

  def buy(self, snack, money):
      """
      Buy a snack from the vending machine.
        Prints "Cannot find food" if the snack is not a valid (e.g. None)
        Prints "None left!" if the named snack is unavailable
        Prints "You can't afford it" if the snack costs too much
        Prints "You bought a snack!" if successful and decrements the snack qty
      Param snack (Snack): the snack to buy (might be None)
      Param money (int): the available amount of money
      Returns: True is successful; False otherwise
      """
      if snack is None:
          print("Cannot find food!")
          return False
      elif snack.getQuantity() == 0:
          print("None left!")
          return False
      elif snack.getCost() > money:
          print("You can't afford it")
          return False
      else:
          print("You bought a snack!")
          quantity = snack.getQuantity()
          snack.setQuantity(quantity-1)
          return True

  def findSnack(self, name):
      """
      Searches for a snack matching name in the list of snacks
      Param name (string): the snack name to search
      Returns: found snack (type Snack) if found; None otherwise
      """
      # for every snack in our list of list of snack
      #    if the snack's name == name: return the snack
      for i in range(len(self.foods)):
          snack = self.foods[i]
          if snack.getName() == name:
              return snack
      return None # not found!

  def getSnack(self, idx):
      """
      Returns the snack at marker idx in the list of snacks
      Param idx (int): the snack id
      Returns: snack (type Snack) if found; None otherwise
      """
      if idx < 0 or idx >= len(self.foods):
          return None # not valid, not found
      return self.foods[idx]

  def getNumSnacks(self):
      """
      Returns the number of snacks in the vending machine
      Params: none
      Returns (int): the number of snacks
      """
      return len(self.foods)

if __name__ == '__main__':

   playerMoney = 10
   machine = VendingMachine()

   print("You have $%d dollars to spend at the snack vending machine"%playerMoney)
   print("What would you like to buy?")
   machine.listSnacks()
   print()
   numSnacks = machine.getNumSnacks()

   snack = machine.findSnack("Slurm")
   #print("Snack Test1: ", snack) # for debugging

   snack = machine.findSnack("Cheetos")
   #print("Snack Test2: ", snack) # for debugging

   snackIdx = int(input("Enter a choice from 0 to %d: "%(numSnacks-1)))
   snack = machine.getSnack(snackIdx)
   #print("The user choose: ", snack) # for debugging
   if machine.buy(snack, playerMoney):
      playerMoney = playerMoney - snack.getCost()
      print("You now have $%d left"%playerMoney)

"""
Dictionary practice
"""

def main():
    names2ages = {} # creates an empty dictionary
    print(names2ages)

    # suppose our pairs are str:int
    names2ages = { "marie": 65, "shane":45, "cat":38, "briana":28}
    ages2names = { 65:"marie", 45:"shane"}
    print(names2ages)

    # L[0], e.g. L[<index>]
    print(names2ages["marie"]) # syntax: dict[<key>]

    # if the key is not in the dictionary, we get an KeyError!
    # checking if a key is in a dictionary
    if "bill" in names2ages:
       print(names2ages["bill"]) # dictionary[<key>], throws error if bill not a key
    else:
       print("bill not in dictionary")

    print(names2ages.keys()) # returns all keys
    print(names2ages.values()) # returns all values

    # append to add items to a list
    names2ages["bill"] = 10 # add a key/value pair for bill:10
    print(names2ages["bill"])

    names2ages["bill"] = 15 # replace value of 10 to 15 for key "bill"
    print(names2ages["bill"])

    # safe method of getting values for keys
    age = names2ages.get("sally", -1)
    print("sally", age)

    age = names2ages.get("shane", -1)
    print("shane", age)

    print("-"*30)
    for key in names2ages.keys():
        # key can be used to lookup the value
        print(key, names2ages[key])

    print("-"*30)
    for key,value in names2ages.items():
        # item is a key,value tuple
        print(key,value)

main()

"""
Write a program that let's the user ask for the maximum speed of a vehicle
Load the vehicles/speeds from the file "speeds.csv" and store the results
in a dictionary

$ python3 vehicle-dictionary.py
Enter a vehicle (Type nothing to quit): plane
The speed of plane is 2193.000000 mph
Enter a vehicle (Type nothing to quit): bicycle
Sorry. I don't know the speed for bicycle
Enter a vehicle (Type nothing to quit):
"""

def load(filename):
    data = {}
    ifile = open(filename, "r")
    for line in ifile:
        tokens = line.strip().split(",")
        name = tokens[0]
        speed = float(tokens[1])
        data[name] = speed
    return data

def main():
    vehicles2speed = load("speeds.csv")

    done = False
    while not done:
        query = input("Enter a vehicle (Type nothing to quit): ")
        if query == "":
            done = True
        else:
            speed = vehicles2speed.get(query, -1)
            if speed != -1:
                print(f"The speed of {query} is {speed}")
            else:
                print(f"Sorry. I don't know the speed for {query}")

main()