cart.rb 1.2 KB

123456789101112131415161718192021222324252627282930313233
  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. end
  26. current_item
  27. end
  28. def total_price
  29. line_items.to_a.sum { |item| item.total_price }
  30. end
  31. end