cart.rb 1.3 KB

12345678910111213141516171819202122232425262728293031323334
  1. #---
  2. # Excerpted from "Agile Web Development with Rails",
  3. # published by The Pragmatic Bookshelf.
  4. # Copyrights apply to this code. It may not be used to create training material,
  5. # courses, books, articles, and the like. Contact us if you are in doubt.
  6. # We make no guarantees that this code is fit for any purpose.
  7. # Visit http://www.pragmaticprogrammer.com/titles/rails4 for more book information.
  8. #---
  9. #---
  10. # Excerpted from "Agile Web Development with Rails, 4rd Ed.",
  11. # published by The Pragmatic Bookshelf.
  12. # Copyrights apply to this code. It may not be used to create training material,
  13. # courses, books, articles, and the like. Contact us if you are in doubt.
  14. # We make no guarantees that this code is fit for any purpose.
  15. # Visit http://www.pragmaticprogrammer.com/titles/rails4 for more book information.
  16. #---
  17. class Cart < ActiveRecord::Base
  18. has_many :line_items, :dependent => :destroy
  19. def add_product(product_id)
  20. current_item = line_items.find_by_product_id(product_id)
  21. if current_item
  22. current_item.quantity += 1
  23. else
  24. current_item = line_items.build(:product_id => product_id)
  25. current_item.price = current_item.product.price
  26. end
  27. current_item
  28. end
  29. def total_price
  30. line_items.to_a.sum { |item| item.total_price }
  31. end
  32. end