ciprianna/app.rb

## app.rb
# Driver

require "sqlite3"
require_relative "shoes.rb"
require_relative "location.rb"
require_relative "category.rb"

# Creates the database connection
DATABASE = SQLite3::Database.new("shoe_inventory.db")

# Creates the table
DATABASE.execute("CREATE TABLE IF NOT EXISTS shoes (id INTEGER PRIMARY KEY, name TEXT NOT NULL, cost INTEGER NOT NULL, color TEXT NOT NULL, category_id INTEGER, location_id INTEGER, location_stock INTEGER);")
DATABASE.execute("CREATE TABLE IF NOT EXISTS categories (id INTEGER PRIMARY KEY, name TEXT);")
DATABASE.execute("CREATE TABLE IF NOT EXISTS locations (id INTEGER PRIMARY KEY, name TEXT);")

# Returns the results as a Hash
DATABASE.results_as_hash = true

################################################################################

# Main menu in ux to get an initial choice from the user
puts "What would you like to do with the Cutesie Bootsie Inventory?"
60.times {print "-"}
puts "\n"
puts "1".ljust(10) + "View current stock".rjust(30)
puts "2".ljust(10) + "View quantity information".rjust(30)
puts "3".ljust(10) + "Add new product".rjust(30)
puts "4".ljust(10) + "Update product information".rjust(30)
puts "5".ljust(10) + "View products by cost".rjust(30)
puts "6".ljust(10) + "View location information".rjust(30)
puts "7".ljust(10) + "View category information".rjust(30)
puts "8".ljust(10) + "Delete product".rjust(30)
puts "0".ljust(10) + "Exit Cutesie Bootsie Inventory".rjust(30)

print ">> "
choice = gets.to_i

# Loop to get a valid response from the user
range = [1, 2, 3, 4, 5, 6, 7, 8, 0]

while !range.include?(choice)
  puts "Please choose a number from the menu"
  print ">> "
  choice = gets.to_i
end

# Begins the loop when a correct input from the user has been entered.
#   If zero, it skips the loop entirely and exits the program.
while choice != 0

##### Displays all products-----------------------------------------------------
  if choice == 1
    Shoe.all.each do |shoe_hash|
    puts "ID: #{shoe_hash['id']}, Name: #{shoe_hash['name']}, Cost: #{shoe_hash['cost']}, Color: #{shoe_hash['color']}, Category: #{shoe_hash['category_id']}, Location: #{shoe_hash['location_id']}"
    end
  end
##### Quantity information menu; gives a list of sub-options--------------------
  if choice == 2
    puts "What would you like to do?"
    40.times {print "-"}
    puts "\n"
    puts "1".ljust(10) + "View all stock quantities".rjust(30)
    puts "2".ljust(10) + "View low quantities".rjust(30)
    puts "3".ljust(10) + "Update stock quantities".rjust(30)
    puts "0".ljust(10) + "Exit quantity information".rjust(30)
    print ">> "
    quantity_choice = gets.to_i

    quantity_choice_range = [0, 1, 2, 3]
    while !quantity_choice_range.include?(quantity_choice)
      puts "Please choose a number from the menu"
      print ">> "
      quantity_choice = gets.to_i
    end

    while quantity_choice != 0

      ##### Views all stock quantities------------------------------------------
      if quantity_choice == 1
        Shoe.quantity.each do |shoe_hash|
          puts "#{shoe_hash['id']} - #{shoe_hash['name']} (#{shoe_hash['location_stock']})"
        end

        total_stock = Shoe.total_stock
        puts "Total stock quantity - #{total_stock[0][0]}"
      end

      ##### Shows all items where_quantity_is_low-------------------------------
      if quantity_choice == 2
        Shoe.where_quantity_is_low.each do |shoe_hash|
          puts "#{shoe_hash['id']} - #{shoe_hash['name']} (#{shoe_hash['location_stock']})"
        end
      end

      ##### Updates an item's quantity------------------------------------------
      if quantity_choice == 3
        puts "Which product quantity would you like to update?"
        quantity_range = []
        Shoe.all.each do |shoe_hash|
          puts "#{shoe_hash['id']} - #{shoe_hash['name']}"
          quantity_range.push(shoe_hash['id'])
        end
        print ">> "
        shoe_to_change = gets.to_i

        while !quantity_range.include?(shoe_to_change)
          puts "Please choose an id from the options:"
          print ">> "
          shoe_to_change = gets.to_i
        end

        shoe = Shoe.new(shoe_to_change)

        puts "Okay, and how many are you adding? If removing quantity, enter a negative number."
        print ">> "
        change = gets.to_i

        shoe.update_quantity(change)
      end

      ##### Re-asks for the menu options----------------------------------------
      puts "What would you like to do?"
      40.times {print "-"}
      puts "\n"
      puts "1".ljust(10) + "View all stock quantities".rjust(30)
      puts "2".ljust(10) + "View low quantities".rjust(30)
      puts "3".ljust(10) + "Update stock quantities".rjust(30)
      puts "0".ljust(10) + "Exit quantity information".rjust(30)
      print ">> "
      quantity_choice = gets.to_i

    end
  end

##### Adds a new item to the inventory------------------------------------------
  if choice == 3
    puts "Okay, please enter the product information."
    puts "Shoe name:"
    print ">> "
    name = gets.chomp
    puts "Cost:"
    print ">> "
    cost = gets.to_f
    puts "Color:"
    print ">> "
    color = gets.chomp
    puts "Category:"
    category_range = []
    Category.all.each do |category_hash|
      puts "#{category_hash['id']} - #{category_hash['name']}"
      category_range.push(category_hash['id'])
    end
    print ">> "
    category_id = gets.to_i

    while !category_range.include?(category_id)
      puts "Please choose a category id from the options:"
      print ">> "
      category_id = gets.to_i
    end

    puts "Storage location:"
    location_range = []
    Location.all.each do |location_hash|
      puts "#{location_hash['id']} - #{location_hash['name']}"
      location_range.push(location_hash['id'])
    end
    print ">> "
    location_id = gets.to_i

    while !location_range.include?(location_id)
      puts "Please choose a location id from the options:"
      print ">> "
      location_id = gets.to_i
    end

    puts "Quantity:"
    print ">> "
    quantity = gets.to_i
    Shoe.add(name, cost, color, category_id, location_id, quantity)
  end

##### Updates a product's information-------------------------------------------
  if choice == 4
    puts "Which product would you like to update?"
    shoe_range = []
    Shoe.all.each do |shoe_hash|
      puts "#{shoe_hash['id']} - #{shoe_hash['name']}"
      shoe_range.push(shoe_hash['id'])
    end
    print ">> "
    shoe = gets.to_i

    while !shoe_range.include?(shoe)
      puts "Please choose an id from the options:"
      print ">> "
      shoe = gets.to_i
    end

    shoe_to_change = Shoe.new(shoe)

    ##### Displays all information pertaining to the selected shoe--------------
    shoe_to_change.information.each do |shoe_hash|
      puts "ID: #{shoe_hash['id']}, Name: #{shoe_hash['name']}, Cost: #{shoe_hash['cost']}, Color: #{shoe_hash['color']}, Category: #{shoe_hash['category_id']}, Location: #{shoe_hash['location_id']}"
    end

    ##### Sub-menu for updating-------------------------------------------------
    puts "And what would you like to update?"
    40.times {print "-"}
    puts "\n"
    puts "1".ljust(10) + "Name".rjust(30)
    puts "2".ljust(10) + "Cost".rjust(30)
    puts "3".ljust(10) + "Color".rjust(30)
    puts "4".ljust(10) + "Category".rjust(30)
    puts "5".ljust(10) + "Location".rjust(30)
    puts "0".ljust(10) + "Exit product update".rjust(30)
    print ">> "
    to_update = gets.to_i

    range_for_updates = [0, 1, 2, 3, 4, 5]
    while !range_for_updates.include?(to_update)
      puts "Please choose a number from the menu"
      print ">> "
      to_update = gets.to_i
    end

    ##### Begins loop once a valid input is entered; if zero, exits the sub-menu
    while to_update != 0

      ##### Updates the name of the shoe----------------------------------------
      if to_update == 1
        puts "What is the new name for this shoe?"
        print ">> "
        new_name = gets.chomp
        shoe_to_change.update_name(new_name)
      end

      ##### Updates the cost of the shoe----------------------------------------
      if to_update == 2
        puts "What is the new cost for this shoe?"
        print ">> "
        new_cost = gets.to_f
        shoe_to_change.update_cost(new_cost)
      end

      ##### Updates the color of the shoe---------------------------------------
      if to_update == 3
        puts "What is the new color of the shoe?"
        print ">> "
        new_color = gets.chomp
        shoe_to_change.update_color(new_color)
      end

      ##### Updates the category_id of the shoe---------------------------------
      if to_update == 4
        puts "What is the new category of the shoe?"
        category_range = []
        Category.all.each do |category_hash|
          puts "#{category_hash['id']} - #{category_hash['name']}"
          category_range.push(category_hash['id'])
        end
        print ">> "
        new_category_id = gets.to_i

        while !category_range.include?(new_category_id)
          puts "Please choose a category id from the options:"
          print ">> "
          category_id = gets.to_i
        end

        shoe_to_change.update_category(new_category_id)
      end

      ##### Updates the location_id of the shoe---------------------------------
      if to_update == 5
        puts "What location is this shoe moving to?"
        location_range = []
        Location.all.each do |location_hash|
          puts "#{location_hash['id']} - #{location_hash['name']}"
          location_range.push(location_hash['id'])
        end
        print ">> "
        new_location_id = gets.to_i

        while !location_range.include?(new_location_id)
          puts "Please choose a location id from the options:"
          print ">> "
          new_location_id = gets.to_i
        end

        shoe_to_change.update_location(new_location_id)
      end

      ##### Re-asks what option the user would like to choose-------------------
      puts "Is there anything else to update for this product?"
      40.times {print "-"}
      puts "\n"
      puts "1".ljust(10) + "Name".rjust(30)
      puts "2".ljust(10) + "Cost".rjust(30)
      puts "3".ljust(10) + "Color".rjust(30)
      puts "4".ljust(10) + "Category".rjust(30)
      puts "5".ljust(10) + "Location".rjust(30)
      puts "0".ljust(10) + "Exit product update".rjust(30)
      print ">> "
      to_update = gets.to_i

    end

  end

##### Displays information by pricing category----------------------------------
  if choice == 5
    puts "Which pricing category would you like to see?"
    puts "High - $100+"
    puts "Medium - $50-$99"
    puts "Low - $0-$49"
    print ">> "
    pricing_category = gets.chomp.downcase

    while (pricing_category != "high") && (pricing_category != "medium") && (pricing_category != "low")
      puts "Please select 'high', 'medium', or 'low':"
      print ">> "
      pricing_category = gets.chomp.downcase
    end

    shoes_by_price = Shoe.where_cost(pricing_category)
    shoes_by_price.each do |shoe_hash|
      puts "ID: #{shoe_hash['id']}, Name: #{shoe_hash['name']}, Cost: #{shoe_hash['cost']}, Color: #{shoe_hash['color']}, Category: #{shoe_hash['category_id']}, Location: #{shoe_hash['location_id']}"
    end
  end

##### Displays options for the locations menu-----------------------------------
  if choice == 6
    puts "What would you like to do?"
    40.times {print "-"}
    puts "\n"
    puts "1".ljust(10) + "View all locations".rjust(30)
    puts "2".ljust(10) + "View all products at a location".rjust(30)
    puts "3".ljust(10) + "Change location name".rjust(30)
    puts "4".ljust(10) + "Add new location".rjust(30)
    puts "5".ljust(10) + "Delete location".rjust(30)
    puts "0".ljust(10) + "Exit location information".rjust(30)
    print ">> "
    location_choice = gets.to_i

    range_for_locations = [0, 1, 2, 3, 4, 5]
    while !range_for_locations.include?(location_choice)
      puts "Please choose a number from the menu"
      print ">> "
      location_choice = gets.to_i
    end

    ##### Begins loop once a valid input is entered; if zero, exits the sub-menu
    while location_choice != 0

      ##### Displays all locations----------------------------------------------
      if location_choice == 1
        puts "Locations:"
        Location.all.each do |location_hash|
          puts "#{location_hash['id']} - #{location_hash['name']}"
        end
      end

      ##### Displays all products at an instance of a location------------------
      if location_choice == 2
        puts "Which location would you like to view products at?"
        location_range = []
        Location.all.each do |location_hash|
          puts "#{location_hash['id']} - #{location_hash['name']}"
          location_range.push(location_hash['id'])
        end
        print ">> "
        location = gets.to_i

        while !location_range.include?(location)
          puts "Please choose a location id from the options:"
          print ">> "
          location = gets.to_i
        end

        location_to_view = Location.new(location)
        location_to_view.shoes.each do |shoes_hash|
          puts "#{shoes_hash['id']} - #{shoes_hash['name']} (#{shoes_hash['location_stock']})"
        end
      end

      ##### Changes the name of a location--------------------------------------
      if location_choice == 3
        puts "Which location would you like to change the name of?"
        location_range = []
        Location.all.each do |location_hash|
          puts "#{location_hash['id']} - #{location_hash['name']}"
          location_range.push(location_hash['id'])
        end
        print ">> "
        location = gets.to_i

        while !location_range.include?(location)
          puts "Please choose a location id from the options:"
          print ">> "
          location = gets.to_i
        end

        location_to_change = Location.new(location)

        puts "What is the new name of this location?"
        print ">> "
        new_location_name = gets.chomp

        location_to_change.update(new_location_name)
      end

      ##### Adds a new location-------------------------------------------------
      if location_choice == 4
        puts "What's the name of the new location?"
        print ">> "
        new_location = gets.chomp
        Location.add(new_location)
      end

      ##### Deletes a location, after checking to ensure nothing is stored there
      if location_choice == 5
        puts "Which location would you like to delete?"
        location_range = []
        Location.all.each do |location_hash|
          puts "#{location_hash['id']} - #{location_hash['name']}"
          location_range.push(location_hash['id'])
        end
        print ">> "
        location = gets.to_i

        while !location_range.include?(location)
          puts "Please choose a location id from the options:"
          print ">> "
          location = gets.to_i
        end

        location_to_delete = Location.new(location)

        location_to_delete.delete

        if location_to_delete.empty?
          puts "Location deleted"
        else
          puts "Cannot delete this location while shoes are stored here."
          location_to_delete.shoes.each do |shoes_hash|
            puts "#{shoes_hash['id']} - #{shoes_hash['name']} (#{shoes_hash['location_stock']})"
          end
        end

      end

      ##### Re-asks the user to input an option---------------------------------
      puts "What would you like to do?"
      40.times {print "-"}
      puts "\n"
      puts "1".ljust(10) + "View all locations".rjust(30)
      puts "2".ljust(10) + "View all products at a location".rjust(30)
      puts "3".ljust(10) + "Change location name".rjust(30)
      puts "4".ljust(10) + "Add new location".rjust(30)
      puts "5".ljust(10) + "Delete location".rjust(30)
      puts "0".ljust(10) + "Exit location information".rjust(30)
      print ">> "
      location_choice = gets.to_i

    end

  end

##### Displays the options in the Category Menu---------------------------------
  if choice == 7
    puts "What would you like to do?"
    40.times {print "-"}
    puts "\n"
    puts "1".ljust(10) + "View all categories".rjust(30)
    puts "2".ljust(10) + "View all products in a category".rjust(30)
    puts "3".ljust(10) + "Change category name".rjust(30)
    puts "4".ljust(10) + "Add new category".rjust(30)
    puts "5".ljust(10) + "Delete category".rjust(30)
    puts "0".ljust(10) + "Exit category information".rjust(30)
    print ">> "
    category_choice = gets.to_i

    range_for_categories = [0, 1, 2, 3, 4, 5]
    while !range_for_categories.include?(category_choice)
      puts "Please choose a number from the menu"
      print ">> "
      category_choice = gets.to_i
    end

    ##### Begins loop once a valid input is entered; if zero, exits the sub-menu
    while category_choice != 0

      ##### Displays all categories---------------------------------------------
      if category_choice == 1
        puts "Categories:"
        Category.all.each do |categories_hash|
          puts "#{categories_hash['id']} - #{categories_hash['name']}"
        end
      end

      ##### Displays all products in a category instance------------------------
      if category_choice == 2
        puts "Which category would you like to view products in?"
        category_range = []
        Category.all.each do |category_hash|
          puts "#{category_hash['id']} - #{category_hash['name']}"
          category_range.push(category_hash['id'])
        end
        print ">> "
        category = gets.to_i

        while !category_range.include?(category)
          puts "Please choose a category id from the options:"
          print ">> "
          category = gets.to_i
        end

        category_to_view = Category.new(category)
        category_to_view.shoes.each do |shoes_hash|
          puts "#{shoes_hash['id']} - #{shoes_hash['name']} (#{shoes_hash['location_stock']})"
        end
      end

      ##### Changes the name of a category--------------------------------------
      if category_choice == 3
        puts "Which category would you like to change the name of?"
        category_range = []
        Category.all.each do |category_hash|
          puts "#{category_hash['id']} - #{category_hash['name']}"
          category_range.push(category_hash['id'])
        end
        print ">> "
        category = gets.to_i

        while !category_range.include?(category)
          puts "Please choose a category id from the options:"
          print ">> "
          category = gets.to_i
        end

        category_to_change = Category.new(category)

        puts "What is the new name of this category?"
        print ">> "
        new_category_name = gets.chomp

        category_to_change.update(new_category_name)
      end

      ##### Adds a new category-------------------------------------------------
      if category_choice == 4
        puts "What's the name of the new category?"
        print ">> "
        new_category = gets.chomp
        Category.add(new_category)
      end

      ##### Deletes a category after checking to ensure no products are assigned
      #####   to that category--------------------------------------------------
      if category_choice == 5
        puts "Which category would you like to delete?"
        category_range = []
        Category.all.each do |category_hash|
          puts "#{category_hash['id']} - #{category_hash['name']}"
          category_range.push(category_hash['id'])
        end
        print ">> "
        category = gets.to_i

        while !category_range.include?(category)
          puts "Please choose a category id from the options:"
          print ">> "
          category = gets.to_i
        end

        category_to_delete = Category.new(category)

        category_to_delete.delete

        if category_to_delete.empty?
          puts "Category deleted"
        else
          puts "Cannot delete this category while shoes are assigned."
          category_to_delete.shoes.each do |shoes_hash|
            puts "#{shoes_hash['id']} - #{shoes_hash['name']} (#{shoes_hash['location_stock']})"
          end
        end
      end

      ##### Re-asks the user for an option--------------------------------------
      puts "What would you like to do?"
      40.times {print "-"}
      puts "\n"
      puts "1".ljust(10) + "View all categories".rjust(30)
      puts "2".ljust(10) + "View all products in a category".rjust(30)
      puts "3".ljust(10) + "Change category name".rjust(30)
      puts "4".ljust(10) + "Add new category".rjust(30)
      puts "5".ljust(10) + "Delete category".rjust(30)
      puts "0".ljust(10) + "Exit category information".rjust(30)
      print ">> "
      category_choice = gets.to_i

    end

  end

##### Deletes a product from the inventory--------------------------------------
  if choice == 8
    puts "Which product would you like to delete?"
    shoe_range = []
    Shoe.all.each do |shoe_hash|
      puts "#{shoe_hash['id']} - #{shoe_hash['name']}"
      shoe_range.push(shoe_hash['id'])
    end
    print ">> "
    shoe_choice = gets.to_i

    while !shoe_range.include?(shoe_choice)
      puts "Please choose an id from the menu:"
      print ">> "
      shoe_choice = gets.to_i
    end

    shoe_to_delete = Shoe.new(shoe_choice)

    shoe_to_delete.information.each do |shoe_hash|
      puts "ID: #{shoe_hash['id']}, Name: #{shoe_hash['name']}, Cost: #{shoe_hash['cost']}, Color: #{shoe_hash['color']}, Category: #{shoe_hash['category_id']}, Location: #{shoe_hash['location_id']}"
    end

    puts "Are you sure you wish to delete this product? (yes/no)"
    print ">> "
    sure = gets.chomp.downcase

    if sure == "yes"
      shoe_to_delete.delete
    end

  end

##### Re-asks the user for an option--------------------------------------------
  60.times {print "-"}
  puts "\n"
  puts "What would you like to do with the Cutesie Bootsie Inventory?"
  60.times {print "-"}
  puts "\n"
  puts "1".ljust(10) + "View current stock".rjust(30)
  puts "2".ljust(10) + "View quantity information".rjust(30)
  puts "3".ljust(10) + "Add new product".rjust(30)
  puts "4".ljust(10) + "Update product information".rjust(30)
  puts "5".ljust(10) + "View products by cost".rjust(30)
  puts "6".ljust(10) + "View location information".rjust(30)
  puts "7".ljust(10) + "View category information".rjust(30)
  puts "8".ljust(10) + "Delete product".rjust(30)
  puts "0".ljust(10) + "Exit Cutesie Bootsie Inventory".rjust(30)
  print ">> "
  choice = gets.to_i

  range = [1, 2, 3, 4, 5, 6, 7, 8, 0]
  while !range.include?(choice)
    puts "Please choose a number from the menu"
    print ">> "
    choice = gets.chomp
  end

end

##### Exit menu message---------------------------------------------
puts "Bye!"

## category.rb
# Category Class

class Category

  # Assigns an id for identification in instance methods
  #
  # id - Integer assigned as the primary key from the id column
  #
  # Returns the category object created
  def initialize(id)
    @id = id
  end

  # Creates a new category (row) in the categories table
  #
  # category_name - String
  #
  # Returns an empty Array
  def self.add(category_name)
    DATABASE.execute("INSERT INTO categories (name) VALUES ('#{category_name}');")
  end

  # Read method for the categories table
  #
  # Returns all data from the categories table as an Array of Hashes. Each Hash
  #   corresponds with one row of data from the shoes table (one category).
  def self.all
    DATABASE.execute("SELECT * FROM categories;")
  end

  # Reads all shoes in a category object
  #
  # Returns all shoe information in a category from the shoes table as an Array
  #   of Hashes. Each Hash corresponds to a row of data which is stored at the
  #   passed id.
  def shoes
    DATABASE.execute("SELECT * FROM shoes WHERE category_id = #{@id};")
  end

  # Update method for the categories table
  #
  # new_category_name - String
  #
  # Returns an empty Array
  def update(new_category_name)
    DATABASE.execute("UPDATE categories SET name = '#{new_category_name}' WHERE id = #{@id};")
  end

  # Checks to see if the category contains no products.
  #
  # Returns true or false - Binary
  def empty?
    self.shoes == []
  end

  # Delete a category from the categories table
  #
  # Returns an empty Array
  def delete
    if self.empty?
      DATABASE.execute("DELETE FROM categories WHERE id = #{@id};")
    end
  end

end

## location.rb
# Location Class

class Location

  # Assigns an id for identification in instance methods
  #
  # id - Integer assigned as the primary key from the id column
  #
  # Returns location object created
  def initialize(id)
    @id = id
  end

  # Creates a new location (row) in the locations table
  #
  # location_name - String
  #
  # Returns an empty Array
  def self.add(location_name)
    DATABASE.execute("INSERT INTO locations (name) VALUES ('#{location_name}');")
  end

  # Read method for the locations table
  #
  # Returns all information from the locations table as an Array of Hashes. Each
  #   Hash corresponds to one row of data (one location).
  def self.all
    DATABASE.execute("SELECT * FROM locations;")
  end

  # Reads all shoes at a location object
  #
  # Returns all shoe information at one location from the shoes table as an
  #  Array of Hashes. Each Hash corresponds to a row of data which is stored at #  the passed id.
  def shoes
    DATABASE.execute("SELECT * FROM shoes WHERE location_id = #{@id};")
  end

  # Update method for the locations table
  #
  # new_location_name - String
  #
  # Returns an empty Array
  def update(new_location_name)
    DATABASE.execute("UPDATE locations SET name = '#{new_location_name}' WHERE id = #{@id};")
  end

  # Checks to see if the location contains no products.
  #
  # Returns true or false - Binary
  def empty?
    self.shoes == []
  end

  # Delete a category from the categories table
  #
  # Returns an empty Array
  def delete
    if self.empty?
      DATABASE.execute("DELETE FROM locations WHERE id = #{@id};")
    end
  end

end

## shoes.rb
# Shoe Class

class Shoe

  # Assigns an id for identification in instance methods
  #
  # id - Integer assigned as the primary key from the id column
  #
  # Returns object created
  def initialize(id)
    @id = id
  end

  # Creates a new shoe (row) in the shoes table
  #
  # shoe_name - String
  # cost - Integer
  # color - String
  # category_id - Integer, foreign key from the categories table
  # location_id - Integer, foreign key from the locations table
  # quantity - Integer, to be added to the location_stock column
  #
  # Returns an empty Array
  def self.add(shoe_name, cost, color, category_id, location_id, quantity)
    DATABASE.execute("INSERT INTO shoes (name, cost, color, category_id, location_id, location_stock) VALUES ('#{shoe_name}', #{cost}, '#{color}', '#{category_id}', '#{location_id}', #{quantity});")
  end

  # Read method for the shoes table
  #
  # Returns all rows from the shoes table as an Array of Hashes. Each Hash is
  #   a row of data.
  def self.all
    DATABASE.execute("SELECT * FROM shoes;")
  end

  # Read method for stock quantities
  #
  # Returns id - Integer, name - String, and location_stock - Integer, from
  #   shoes table as an Array of Hashes. Each Hash corresponds to a row of data.
  def self.quantity
    DATABASE.execute("SELECT id, name, location_stock FROM shoes;")
  end

  # Sums all of the product inventory
  #
  # Returns the sum of all location_stock values - Integer
  def self.total_stock
    DATABASE.execute("SELECT SUM(location_stock) FROM shoes;")
  end

  # Shows all products by cost categories
  #
  # cost_category - String, should be categories of high, medium, or low
  #
  # Returns all row information where the condition is true as an Array of
  #   Hashes. Each Hash corresponds to a row of data in the shoes table that
  #   meets the criteria in the WHERE statement.
  def self.where_cost(cost_category)
    if cost_category == "high"
      DATABASE.execute("SELECT * FROM shoes WHERE cost >= 100;")
    elsif cost_category == "medium"
      DATABASE.execute("SELECT * FROM shoes WHERE cost >= 50 AND cost < 100;")
    else
      DATABASE.execute("SELECT * FROM shoes WHERE cost < 50;")
    end
  end

  # Reads all products with low quantities in location_stock.
  #
  # Returns all row information, for only rows where location_stock is less than
  #   5, as an Array of Hashes. Each Hash corresponds to a row of data in the
  #   shoes table that meets the criteria in the WHERE statement.
  def self.where_quantity_is_low
    DATABASE.execute("SELECT * FROM shoes WHERE location_stock < 5;")
  end

  # Read method for a single shoe product (row) in the shoes table
  #
  # Returns row of information as an Array with a single Hash. The Hash
  #   corresponds to the row of data that matches the given id.
  def information
    DATABASE.execute("SELECT * FROM shoes WHERE id = #{@id};")
  end

  # Update any value for a given field that takes Strings
  #
  # field_name - String
  # new_value - String
  #
  # Returns an empty Array
  def update_strings(field_name, new_value)
    DATABASE.execute("UPDATE shoes SET #{field_name} = '#{new_value}' WHERE id = #{@id};")
  end

  # Update any value for a given field that takes Integers
  #
  # field_name - String
  # new_value - Integer
  #
  # Returns an empty Array
  def update_integers(field_name, new_value)
    DATABASE.execute("UPDATE shoes SET #{field_name} = #{new_value} WHERE id = #{@id};")
  end

  # Updates the name value in the shoes table
  #
  # new_name - String
  #
  # Returns an empty Array
  def update_name(new_name)
    update_strings("name", new_name)
  end

  # Updates the cost value in the shoes table
  #
  # new_cost - Integer
  #
  # Returns an empty Array
  def update_cost(new_cost)
    update_integers("cost", new_cost)
  end

  # Updates the color value in the shoes table
  #
  # new_color - String
  #
  # Returns an empty Array
  def update_color(new_color)
    update_strings("color", new_color)
  end

  # Assigns/updates the location_id of a shoe instance
  #
  # new_location_id - Integer
  #
  # Returns an empty Array
  def update_location(new_location_id)
    update_integers("location_id", new_location_id)
  end

  # Updates the quantity of a product for a given id
  #
  # to_add - Integer
  #
  # Returns an empty Array
  def update_quantity(to_add)
    current_quantity = DATABASE.execute("SELECT location_stock FROM shoes WHERE id = #{@id};").first['location_stock']
    DATABASE.execute("UPDATE shoes SET location_stock = #{current_quantity + to_add} WHERE id = #{@id};")
  end

  # Assigns/updates the category_id of a shoe instance
  #
  # new_category_id - Integer
  #
  # Returns an empty Array
  def update_category(new_category_id)
    update_integers("category_id", new_category_id)
  end

  # Deletes a shoe row from the shoes table
  #
  # Returns an empty Array
  def delete
    DATABASE.execute("DELETE FROM shoes WHERE id = #{@id};")
  end

end
	# Driver

	require "sqlite3"
	require_relative "shoes.rb"
	require_relative "location.rb"
	require_relative "category.rb"

	# Creates the database connection
	DATABASE = SQLite3::Database.new("shoe_inventory.db")

	# Creates the table
	DATABASE.execute("CREATE TABLE IF NOT EXISTS shoes (id INTEGER PRIMARY KEY, name TEXT NOT NULL, cost INTEGER NOT NULL, color TEXT NOT NULL, category_id INTEGER, location_id INTEGER, location_stock INTEGER);")
	DATABASE.execute("CREATE TABLE IF NOT EXISTS categories (id INTEGER PRIMARY KEY, name TEXT);")
	DATABASE.execute("CREATE TABLE IF NOT EXISTS locations (id INTEGER PRIMARY KEY, name TEXT);")

	# Returns the results as a Hash
	DATABASE.results_as_hash = true

	################################################################################

	# Main menu in ux to get an initial choice from the user
	puts "What would you like to do with the Cutesie Bootsie Inventory?"
	60.times {print "-"}
	puts "\n"
	puts "1".ljust(10) + "View current stock".rjust(30)
	puts "2".ljust(10) + "View quantity information".rjust(30)
	puts "3".ljust(10) + "Add new product".rjust(30)
	puts "4".ljust(10) + "Update product information".rjust(30)
	puts "5".ljust(10) + "View products by cost".rjust(30)
	puts "6".ljust(10) + "View location information".rjust(30)
	puts "7".ljust(10) + "View category information".rjust(30)
	puts "8".ljust(10) + "Delete product".rjust(30)
	puts "0".ljust(10) + "Exit Cutesie Bootsie Inventory".rjust(30)

	print ">> "
	choice = gets.to_i

	# Loop to get a valid response from the user
	range = [1, 2, 3, 4, 5, 6, 7, 8, 0]

	while !range.include?(choice)
	puts "Please choose a number from the menu"
	print ">> "
	choice = gets.to_i
	end

	# Begins the loop when a correct input from the user has been entered.
	# If zero, it skips the loop entirely and exits the program.
	while choice != 0

	##### Displays all products-----------------------------------------------------
	if choice == 1
	Shoe.all.each do \|shoe_hash\|
	puts "ID: #{shoe_hash['id']}, Name: #{shoe_hash['name']}, Cost: #{shoe_hash['cost']}, Color: #{shoe_hash['color']}, Category: #{shoe_hash['category_id']}, Location: #{shoe_hash['location_id']}"
	end
	end
	##### Quantity information menu; gives a list of sub-options--------------------
	if choice == 2
	puts "What would you like to do?"
	40.times {print "-"}
	puts "\n"
	puts "1".ljust(10) + "View all stock quantities".rjust(30)
	puts "2".ljust(10) + "View low quantities".rjust(30)
	puts "3".ljust(10) + "Update stock quantities".rjust(30)
	puts "0".ljust(10) + "Exit quantity information".rjust(30)
	print ">> "
	quantity_choice = gets.to_i

	quantity_choice_range = [0, 1, 2, 3]
	while !quantity_choice_range.include?(quantity_choice)
	puts "Please choose a number from the menu"
	print ">> "
	quantity_choice = gets.to_i
	end

	while quantity_choice != 0

	##### Views all stock quantities------------------------------------------
	if quantity_choice == 1
	Shoe.quantity.each do \|shoe_hash\|
	puts "#{shoe_hash['id']} - #{shoe_hash['name']} (#{shoe_hash['location_stock']})"
	end

	total_stock = Shoe.total_stock
	puts "Total stock quantity - #{total_stock[0][0]}"
	end

	##### Shows all items where_quantity_is_low-------------------------------
	if quantity_choice == 2
	Shoe.where_quantity_is_low.each do \|shoe_hash\|
	puts "#{shoe_hash['id']} - #{shoe_hash['name']} (#{shoe_hash['location_stock']})"
	end
	end

	##### Updates an item's quantity------------------------------------------
	if quantity_choice == 3
	puts "Which product quantity would you like to update?"
	quantity_range = []
	Shoe.all.each do \|shoe_hash\|
	puts "#{shoe_hash['id']} - #{shoe_hash['name']}"
	quantity_range.push(shoe_hash['id'])
	end
	print ">> "
	shoe_to_change = gets.to_i

	while !quantity_range.include?(shoe_to_change)
	puts "Please choose an id from the options:"
	print ">> "
	shoe_to_change = gets.to_i
	end

	shoe = Shoe.new(shoe_to_change)

	puts "Okay, and how many are you adding? If removing quantity, enter a negative number."
	print ">> "
	change = gets.to_i

	shoe.update_quantity(change)
	end

	##### Re-asks for the menu options----------------------------------------
	puts "What would you like to do?"
	40.times {print "-"}
	puts "\n"
	puts "1".ljust(10) + "View all stock quantities".rjust(30)
	puts "2".ljust(10) + "View low quantities".rjust(30)
	puts "3".ljust(10) + "Update stock quantities".rjust(30)
	puts "0".ljust(10) + "Exit quantity information".rjust(30)
	print ">> "
	quantity_choice = gets.to_i

	end
	end

	##### Adds a new item to the inventory------------------------------------------
	if choice == 3
	puts "Okay, please enter the product information."
	puts "Shoe name:"
	print ">> "
	name = gets.chomp
	puts "Cost:"
	print ">> "
	cost = gets.to_f
	puts "Color:"
	print ">> "
	color = gets.chomp
	puts "Category:"
	category_range = []
	Category.all.each do \|category_hash\|
	puts "#{category_hash['id']} - #{category_hash['name']}"
	category_range.push(category_hash['id'])
	end
	print ">> "
	category_id = gets.to_i

	while !category_range.include?(category_id)
	puts "Please choose a category id from the options:"
	print ">> "
	category_id = gets.to_i
	end

	puts "Storage location:"
	location_range = []
	Location.all.each do \|location_hash\|
	puts "#{location_hash['id']} - #{location_hash['name']}"
	location_range.push(location_hash['id'])
	end
	print ">> "
	location_id = gets.to_i

	while !location_range.include?(location_id)
	puts "Please choose a location id from the options:"
	print ">> "
	location_id = gets.to_i
	end

	puts "Quantity:"
	print ">> "
	quantity = gets.to_i
	Shoe.add(name, cost, color, category_id, location_id, quantity)
	end

	##### Updates a product's information-------------------------------------------
	if choice == 4
	puts "Which product would you like to update?"
	shoe_range = []
	Shoe.all.each do \|shoe_hash\|
	puts "#{shoe_hash['id']} - #{shoe_hash['name']}"
	shoe_range.push(shoe_hash['id'])
	end
	print ">> "
	shoe = gets.to_i

	while !shoe_range.include?(shoe)
	puts "Please choose an id from the options:"
	print ">> "
	shoe = gets.to_i
	end

	shoe_to_change = Shoe.new(shoe)

	##### Displays all information pertaining to the selected shoe--------------
	shoe_to_change.information.each do \|shoe_hash\|
	puts "ID: #{shoe_hash['id']}, Name: #{shoe_hash['name']}, Cost: #{shoe_hash['cost']}, Color: #{shoe_hash['color']}, Category: #{shoe_hash['category_id']}, Location: #{shoe_hash['location_id']}"
	end

	##### Sub-menu for updating-------------------------------------------------
	puts "And what would you like to update?"
	40.times {print "-"}
	puts "\n"
	puts "1".ljust(10) + "Name".rjust(30)
	puts "2".ljust(10) + "Cost".rjust(30)
	puts "3".ljust(10) + "Color".rjust(30)
	puts "4".ljust(10) + "Category".rjust(30)
	puts "5".ljust(10) + "Location".rjust(30)
	puts "0".ljust(10) + "Exit product update".rjust(30)
	print ">> "
	to_update = gets.to_i

	range_for_updates = [0, 1, 2, 3, 4, 5]
	while !range_for_updates.include?(to_update)
	puts "Please choose a number from the menu"
	print ">> "
	to_update = gets.to_i
	end

	##### Begins loop once a valid input is entered; if zero, exits the sub-menu
	while to_update != 0

	##### Updates the name of the shoe----------------------------------------
	if to_update == 1
	puts "What is the new name for this shoe?"
	print ">> "
	new_name = gets.chomp
	shoe_to_change.update_name(new_name)
	end

	##### Updates the cost of the shoe----------------------------------------
	if to_update == 2
	puts "What is the new cost for this shoe?"
	print ">> "
	new_cost = gets.to_f
	shoe_to_change.update_cost(new_cost)
	end

	##### Updates the color of the shoe---------------------------------------
	if to_update == 3
	puts "What is the new color of the shoe?"
	print ">> "
	new_color = gets.chomp
	shoe_to_change.update_color(new_color)
	end

	##### Updates the category_id of the shoe---------------------------------
	if to_update == 4
	puts "What is the new category of the shoe?"
	category_range = []
	Category.all.each do \|category_hash\|
	puts "#{category_hash['id']} - #{category_hash['name']}"
	category_range.push(category_hash['id'])
	end
	print ">> "
	new_category_id = gets.to_i

	while !category_range.include?(new_category_id)
	puts "Please choose a category id from the options:"
	print ">> "
	category_id = gets.to_i
	end

	shoe_to_change.update_category(new_category_id)
	end

	##### Updates the location_id of the shoe---------------------------------
	if to_update == 5
	puts "What location is this shoe moving to?"
	location_range = []
	Location.all.each do \|location_hash\|
	puts "#{location_hash['id']} - #{location_hash['name']}"
	location_range.push(location_hash['id'])
	end
	print ">> "
	new_location_id = gets.to_i

	while !location_range.include?(new_location_id)
	puts "Please choose a location id from the options:"
	print ">> "
	new_location_id = gets.to_i
	end

	shoe_to_change.update_location(new_location_id)
	end

	##### Re-asks what option the user would like to choose-------------------
	puts "Is there anything else to update for this product?"
	40.times {print "-"}
	puts "\n"
	puts "1".ljust(10) + "Name".rjust(30)
	puts "2".ljust(10) + "Cost".rjust(30)
	puts "3".ljust(10) + "Color".rjust(30)
	puts "4".ljust(10) + "Category".rjust(30)
	puts "5".ljust(10) + "Location".rjust(30)
	puts "0".ljust(10) + "Exit product update".rjust(30)
	print ">> "
	to_update = gets.to_i

	end

	end

	##### Displays information by pricing category----------------------------------
	if choice == 5
	puts "Which pricing category would you like to see?"
	puts "High - $100+"
	puts "Medium - $50-$99"
	puts "Low - $0-$49"
	print ">> "
	pricing_category = gets.chomp.downcase

	while (pricing_category != "high") && (pricing_category != "medium") && (pricing_category != "low")
	puts "Please select 'high', 'medium', or 'low':"
	print ">> "
	pricing_category = gets.chomp.downcase
	end

	shoes_by_price = Shoe.where_cost(pricing_category)
	shoes_by_price.each do \|shoe_hash\|
	puts "ID: #{shoe_hash['id']}, Name: #{shoe_hash['name']}, Cost: #{shoe_hash['cost']}, Color: #{shoe_hash['color']}, Category: #{shoe_hash['category_id']}, Location: #{shoe_hash['location_id']}"
	end
	end

	##### Displays options for the locations menu-----------------------------------
	if choice == 6
	puts "What would you like to do?"
	40.times {print "-"}
	puts "\n"
	puts "1".ljust(10) + "View all locations".rjust(30)
	puts "2".ljust(10) + "View all products at a location".rjust(30)
	puts "3".ljust(10) + "Change location name".rjust(30)
	puts "4".ljust(10) + "Add new location".rjust(30)
	puts "5".ljust(10) + "Delete location".rjust(30)
	puts "0".ljust(10) + "Exit location information".rjust(30)
	print ">> "
	location_choice = gets.to_i

	range_for_locations = [0, 1, 2, 3, 4, 5]
	while !range_for_locations.include?(location_choice)
	puts "Please choose a number from the menu"
	print ">> "
	location_choice = gets.to_i
	end

	##### Begins loop once a valid input is entered; if zero, exits the sub-menu
	while location_choice != 0

	##### Displays all locations----------------------------------------------
	if location_choice == 1
	puts "Locations:"
	Location.all.each do \|location_hash\|
	puts "#{location_hash['id']} - #{location_hash['name']}"
	end
	end

	##### Displays all products at an instance of a location------------------
	if location_choice == 2
	puts "Which location would you like to view products at?"
	location_range = []
	Location.all.each do \|location_hash\|
	puts "#{location_hash['id']} - #{location_hash['name']}"
	location_range.push(location_hash['id'])
	end
	print ">> "
	location = gets.to_i

	while !location_range.include?(location)
	puts "Please choose a location id from the options:"
	print ">> "
	location = gets.to_i
	end

	location_to_view = Location.new(location)
	location_to_view.shoes.each do \|shoes_hash\|
	puts "#{shoes_hash['id']} - #{shoes_hash['name']} (#{shoes_hash['location_stock']})"
	end
	end

	##### Changes the name of a location--------------------------------------
	if location_choice == 3
	puts "Which location would you like to change the name of?"
	location_range = []
	Location.all.each do \|location_hash\|
	puts "#{location_hash['id']} - #{location_hash['name']}"
	location_range.push(location_hash['id'])
	end
	print ">> "
	location = gets.to_i

	while !location_range.include?(location)
	puts "Please choose a location id from the options:"
	print ">> "
	location = gets.to_i
	end

	location_to_change = Location.new(location)

	puts "What is the new name of this location?"
	print ">> "
	new_location_name = gets.chomp

	location_to_change.update(new_location_name)
	end

	##### Adds a new location-------------------------------------------------
	if location_choice == 4
	puts "What's the name of the new location?"
	print ">> "
	new_location = gets.chomp
	Location.add(new_location)
	end

	##### Deletes a location, after checking to ensure nothing is stored there
	if location_choice == 5
	puts "Which location would you like to delete?"
	location_range = []
	Location.all.each do \|location_hash\|
	puts "#{location_hash['id']} - #{location_hash['name']}"
	location_range.push(location_hash['id'])
	end
	print ">> "
	location = gets.to_i

	while !location_range.include?(location)
	puts "Please choose a location id from the options:"
	print ">> "
	location = gets.to_i
	end

	location_to_delete = Location.new(location)

	location_to_delete.delete

	if location_to_delete.empty?
	puts "Location deleted"
	else
	puts "Cannot delete this location while shoes are stored here."
	location_to_delete.shoes.each do \|shoes_hash\|
	puts "#{shoes_hash['id']} - #{shoes_hash['name']} (#{shoes_hash['location_stock']})"
	end
	end

	end

	##### Re-asks the user to input an option---------------------------------
	puts "What would you like to do?"
	40.times {print "-"}
	puts "\n"
	puts "1".ljust(10) + "View all locations".rjust(30)
	puts "2".ljust(10) + "View all products at a location".rjust(30)
	puts "3".ljust(10) + "Change location name".rjust(30)
	puts "4".ljust(10) + "Add new location".rjust(30)
	puts "5".ljust(10) + "Delete location".rjust(30)
	puts "0".ljust(10) + "Exit location information".rjust(30)
	print ">> "
	location_choice = gets.to_i

	end

	end

	##### Displays the options in the Category Menu---------------------------------
	if choice == 7
	puts "What would you like to do?"
	40.times {print "-"}
	puts "\n"
	puts "1".ljust(10) + "View all categories".rjust(30)
	puts "2".ljust(10) + "View all products in a category".rjust(30)
	puts "3".ljust(10) + "Change category name".rjust(30)
	puts "4".ljust(10) + "Add new category".rjust(30)
	puts "5".ljust(10) + "Delete category".rjust(30)
	puts "0".ljust(10) + "Exit category information".rjust(30)
	print ">> "
	category_choice = gets.to_i

	range_for_categories = [0, 1, 2, 3, 4, 5]
	while !range_for_categories.include?(category_choice)
	puts "Please choose a number from the menu"
	print ">> "
	category_choice = gets.to_i
	end

	##### Begins loop once a valid input is entered; if zero, exits the sub-menu
	while category_choice != 0

	##### Displays all categories---------------------------------------------
	if category_choice == 1
	puts "Categories:"
	Category.all.each do \|categories_hash\|
	puts "#{categories_hash['id']} - #{categories_hash['name']}"
	end
	end

	##### Displays all products in a category instance------------------------
	if category_choice == 2
	puts "Which category would you like to view products in?"
	category_range = []
	Category.all.each do \|category_hash\|
	puts "#{category_hash['id']} - #{category_hash['name']}"
	category_range.push(category_hash['id'])
	end
	print ">> "
	category = gets.to_i

	while !category_range.include?(category)
	puts "Please choose a category id from the options:"
	print ">> "
	category = gets.to_i
	end

	category_to_view = Category.new(category)
	category_to_view.shoes.each do \|shoes_hash\|
	puts "#{shoes_hash['id']} - #{shoes_hash['name']} (#{shoes_hash['location_stock']})"
	end
	end

	##### Changes the name of a category--------------------------------------
	if category_choice == 3
	puts "Which category would you like to change the name of?"
	category_range = []
	Category.all.each do \|category_hash\|
	puts "#{category_hash['id']} - #{category_hash['name']}"
	category_range.push(category_hash['id'])
	end
	print ">> "
	category = gets.to_i

	while !category_range.include?(category)
	puts "Please choose a category id from the options:"
	print ">> "
	category = gets.to_i
	end

	category_to_change = Category.new(category)

	puts "What is the new name of this category?"
	print ">> "
	new_category_name = gets.chomp

	category_to_change.update(new_category_name)
	end

	##### Adds a new category-------------------------------------------------
	if category_choice == 4
	puts "What's the name of the new category?"
	print ">> "
	new_category = gets.chomp
	Category.add(new_category)
	end

	##### Deletes a category after checking to ensure no products are assigned
	##### to that category--------------------------------------------------
	if category_choice == 5
	puts "Which category would you like to delete?"
	category_range = []
	Category.all.each do \|category_hash\|
	puts "#{category_hash['id']} - #{category_hash['name']}"
	category_range.push(category_hash['id'])
	end
	print ">> "
	category = gets.to_i

	while !category_range.include?(category)
	puts "Please choose a category id from the options:"
	print ">> "
	category = gets.to_i
	end

	category_to_delete = Category.new(category)

	category_to_delete.delete

	if category_to_delete.empty?
	puts "Category deleted"
	else
	puts "Cannot delete this category while shoes are assigned."
	category_to_delete.shoes.each do \|shoes_hash\|
	puts "#{shoes_hash['id']} - #{shoes_hash['name']} (#{shoes_hash['location_stock']})"
	end
	end
	end

	##### Re-asks the user for an option--------------------------------------
	puts "What would you like to do?"
	40.times {print "-"}
	puts "\n"
	puts "1".ljust(10) + "View all categories".rjust(30)
	puts "2".ljust(10) + "View all products in a category".rjust(30)
	puts "3".ljust(10) + "Change category name".rjust(30)
	puts "4".ljust(10) + "Add new category".rjust(30)
	puts "5".ljust(10) + "Delete category".rjust(30)
	puts "0".ljust(10) + "Exit category information".rjust(30)
	print ">> "
	category_choice = gets.to_i

	end

	end

	##### Deletes a product from the inventory--------------------------------------
	if choice == 8
	puts "Which product would you like to delete?"
	shoe_range = []
	Shoe.all.each do \|shoe_hash\|
	puts "#{shoe_hash['id']} - #{shoe_hash['name']}"
	shoe_range.push(shoe_hash['id'])
	end
	print ">> "
	shoe_choice = gets.to_i

	while !shoe_range.include?(shoe_choice)
	puts "Please choose an id from the menu:"
	print ">> "
	shoe_choice = gets.to_i
	end

	shoe_to_delete = Shoe.new(shoe_choice)

	shoe_to_delete.information.each do \|shoe_hash\|
	puts "ID: #{shoe_hash['id']}, Name: #{shoe_hash['name']}, Cost: #{shoe_hash['cost']}, Color: #{shoe_hash['color']}, Category: #{shoe_hash['category_id']}, Location: #{shoe_hash['location_id']}"
	end

	puts "Are you sure you wish to delete this product? (yes/no)"
	print ">> "
	sure = gets.chomp.downcase

	if sure == "yes"
	shoe_to_delete.delete
	end

	end

	##### Re-asks the user for an option--------------------------------------------
	60.times {print "-"}
	puts "\n"
	puts "What would you like to do with the Cutesie Bootsie Inventory?"
	60.times {print "-"}
	puts "\n"
	puts "1".ljust(10) + "View current stock".rjust(30)
	puts "2".ljust(10) + "View quantity information".rjust(30)
	puts "3".ljust(10) + "Add new product".rjust(30)
	puts "4".ljust(10) + "Update product information".rjust(30)
	puts "5".ljust(10) + "View products by cost".rjust(30)
	puts "6".ljust(10) + "View location information".rjust(30)
	puts "7".ljust(10) + "View category information".rjust(30)
	puts "8".ljust(10) + "Delete product".rjust(30)
	puts "0".ljust(10) + "Exit Cutesie Bootsie Inventory".rjust(30)
	print ">> "
	choice = gets.to_i

	range = [1, 2, 3, 4, 5, 6, 7, 8, 0]
	while !range.include?(choice)
	puts "Please choose a number from the menu"
	print ">> "
	choice = gets.chomp
	end

	end

	##### Exit menu message---------------------------------------------
	puts "Bye!"
	# Category Class

	class Category

	# Assigns an id for identification in instance methods
	#
	# id - Integer assigned as the primary key from the id column
	#
	# Returns the category object created
	def initialize(id)
	@id = id
	end

	# Creates a new category (row) in the categories table
	#
	# category_name - String
	#
	# Returns an empty Array
	def self.add(category_name)
	DATABASE.execute("INSERT INTO categories (name) VALUES ('#{category_name}');")
	end

	# Read method for the categories table
	#
	# Returns all data from the categories table as an Array of Hashes. Each Hash
	# corresponds with one row of data from the shoes table (one category).
	def self.all
	DATABASE.execute("SELECT * FROM categories;")
	end

	# Reads all shoes in a category object
	#
	# Returns all shoe information in a category from the shoes table as an Array
	# of Hashes. Each Hash corresponds to a row of data which is stored at the
	# passed id.
	def shoes
	DATABASE.execute("SELECT * FROM shoes WHERE category_id = #{@id};")
	end

	# Update method for the categories table
	#
	# new_category_name - String
	#
	# Returns an empty Array
	def update(new_category_name)
	DATABASE.execute("UPDATE categories SET name = '#{new_category_name}' WHERE id = #{@id};")
	end

	# Checks to see if the category contains no products.
	#
	# Returns true or false - Binary
	def empty?
	self.shoes == []
	end

	# Delete a category from the categories table
	#
	# Returns an empty Array
	def delete
	if self.empty?
	DATABASE.execute("DELETE FROM categories WHERE id = #{@id};")
	end
	end

	end
	# Location Class

	class Location

	# Assigns an id for identification in instance methods
	#
	# id - Integer assigned as the primary key from the id column
	#
	# Returns location object created
	def initialize(id)
	@id = id
	end

	# Creates a new location (row) in the locations table
	#
	# location_name - String
	#
	# Returns an empty Array
	def self.add(location_name)
	DATABASE.execute("INSERT INTO locations (name) VALUES ('#{location_name}');")
	end

	# Read method for the locations table
	#
	# Returns all information from the locations table as an Array of Hashes. Each
	# Hash corresponds to one row of data (one location).
	def self.all
	DATABASE.execute("SELECT * FROM locations;")
	end

	# Reads all shoes at a location object
	#
	# Returns all shoe information at one location from the shoes table as an
	# Array of Hashes. Each Hash corresponds to a row of data which is stored at # the passed id.
	def shoes
	DATABASE.execute("SELECT * FROM shoes WHERE location_id = #{@id};")
	end

	# Update method for the locations table
	#
	# new_location_name - String
	#
	# Returns an empty Array
	def update(new_location_name)
	DATABASE.execute("UPDATE locations SET name = '#{new_location_name}' WHERE id = #{@id};")
	end

	# Checks to see if the location contains no products.
	#
	# Returns true or false - Binary
	def empty?
	self.shoes == []
	end

	# Delete a category from the categories table
	#
	# Returns an empty Array
	def delete
	if self.empty?
	DATABASE.execute("DELETE FROM locations WHERE id = #{@id};")
	end
	end

	end
	# Shoe Class

	class Shoe

	# Assigns an id for identification in instance methods
	#
	# id - Integer assigned as the primary key from the id column
	#
	# Returns object created
	def initialize(id)
	@id = id
	end

	# Creates a new shoe (row) in the shoes table
	#
	# shoe_name - String
	# cost - Integer
	# color - String
	# category_id - Integer, foreign key from the categories table
	# location_id - Integer, foreign key from the locations table
	# quantity - Integer, to be added to the location_stock column
	#
	# Returns an empty Array
	def self.add(shoe_name, cost, color, category_id, location_id, quantity)
	DATABASE.execute("INSERT INTO shoes (name, cost, color, category_id, location_id, location_stock) VALUES ('#{shoe_name}', #{cost}, '#{color}', '#{category_id}', '#{location_id}', #{quantity});")
	end

	# Read method for the shoes table
	#
	# Returns all rows from the shoes table as an Array of Hashes. Each Hash is
	# a row of data.
	def self.all
	DATABASE.execute("SELECT * FROM shoes;")
	end

	# Read method for stock quantities
	#
	# Returns id - Integer, name - String, and location_stock - Integer, from
	# shoes table as an Array of Hashes. Each Hash corresponds to a row of data.
	def self.quantity
	DATABASE.execute("SELECT id, name, location_stock FROM shoes;")
	end

	# Sums all of the product inventory
	#
	# Returns the sum of all location_stock values - Integer
	def self.total_stock
	DATABASE.execute("SELECT SUM(location_stock) FROM shoes;")
	end

	# Shows all products by cost categories
	#
	# cost_category - String, should be categories of high, medium, or low
	#
	# Returns all row information where the condition is true as an Array of
	# Hashes. Each Hash corresponds to a row of data in the shoes table that
	# meets the criteria in the WHERE statement.
	def self.where_cost(cost_category)
	if cost_category == "high"
	DATABASE.execute("SELECT * FROM shoes WHERE cost >= 100;")
	elsif cost_category == "medium"
	DATABASE.execute("SELECT * FROM shoes WHERE cost >= 50 AND cost < 100;")
	else
	DATABASE.execute("SELECT * FROM shoes WHERE cost < 50;")
	end
	end

	# Reads all products with low quantities in location_stock.
	#
	# Returns all row information, for only rows where location_stock is less than
	# 5, as an Array of Hashes. Each Hash corresponds to a row of data in the
	# shoes table that meets the criteria in the WHERE statement.
	def self.where_quantity_is_low
	DATABASE.execute("SELECT * FROM shoes WHERE location_stock < 5;")
	end

	# Read method for a single shoe product (row) in the shoes table
	#
	# Returns row of information as an Array with a single Hash. The Hash
	# corresponds to the row of data that matches the given id.
	def information
	DATABASE.execute("SELECT * FROM shoes WHERE id = #{@id};")
	end

	# Update any value for a given field that takes Strings
	#
	# field_name - String
	# new_value - String
	#
	# Returns an empty Array
	def update_strings(field_name, new_value)
	DATABASE.execute("UPDATE shoes SET #{field_name} = '#{new_value}' WHERE id = #{@id};")
	end

	# Update any value for a given field that takes Integers
	#
	# field_name - String
	# new_value - Integer
	#
	# Returns an empty Array
	def update_integers(field_name, new_value)
	DATABASE.execute("UPDATE shoes SET #{field_name} = #{new_value} WHERE id = #{@id};")
	end

	# Updates the name value in the shoes table
	#
	# new_name - String
	#
	# Returns an empty Array
	def update_name(new_name)
	update_strings("name", new_name)
	end

	# Updates the cost value in the shoes table
	#
	# new_cost - Integer
	#
	# Returns an empty Array
	def update_cost(new_cost)
	update_integers("cost", new_cost)
	end

	# Updates the color value in the shoes table
	#
	# new_color - String
	#
	# Returns an empty Array
	def update_color(new_color)
	update_strings("color", new_color)
	end

	# Assigns/updates the location_id of a shoe instance
	#
	# new_location_id - Integer
	#
	# Returns an empty Array
	def update_location(new_location_id)
	update_integers("location_id", new_location_id)
	end

	# Updates the quantity of a product for a given id
	#
	# to_add - Integer
	#
	# Returns an empty Array
	def update_quantity(to_add)
	current_quantity = DATABASE.execute("SELECT location_stock FROM shoes WHERE id = #{@id};").first['location_stock']
	DATABASE.execute("UPDATE shoes SET location_stock = #{current_quantity + to_add} WHERE id = #{@id};")
	end

	# Assigns/updates the category_id of a shoe instance
	#
	# new_category_id - Integer
	#
	# Returns an empty Array
	def update_category(new_category_id)
	update_integers("category_id", new_category_id)
	end

	# Deletes a shoe row from the shoes table
	#
	# Returns an empty Array
	def delete
	DATABASE.execute("DELETE FROM shoes WHERE id = #{@id};")
	end

	end