Add payment after order create for users

app/controllers/orders_controller.rb

@@ -22,6 +22,7 @@ class OrdersController < ApplicationController
     @products = Product.all
     @order_products = @order.order_products
     if @order.save
+      @user.pay(@order.price)
       redirect_to root_path
     else
       render 'new'

app/models/order.rb

@@ -30,4 +30,12 @@ class Order < ActiveRecord::Base
 
   accepts_nested_attributes_for :order_products, reject_if: proc { |op| op[:count].to_i <= 0 }
 
+  def price
+    price = 0
+    products.each do |p|
+      price += p.price * p.count(self)
+    end
+    price
+  end
+
 end

app/models/user.rb

@@ -23,5 +23,9 @@ class User < ActiveRecord::Base
     "#{name} #{last_name}"
   end
 
+  def pay(amount)
+    self.increment!(:balance, - amount)
+  end
+
   has_secure_password
 end