Update customers_controller.rb

app/controllers/customers_controller.rb

@@ -38,42 +38,48 @@ class CustomersController < ApplicationController
   
   # display a specific customer
   def show
+    begin
       @customer = Customer.find_by_id(params[:id])
-    if @customer
       @vehicles = @customer.vehicles.paginate(:page => params[:page])
       @issues = @customer.issues
-    else
+    rescue ActiveRecord::RecordNotFound
-      flash[:error] = "Customer Not Found"  
+      render_404
-      render :index
     end
   end
   
   # return an HTML form for editing a customer
   def edit
+    begin
       @customer = Customer.find_by_id(params[:id])
+    rescue ActiveRecord::RecordNotFound
+      render_404
+    end
   end
   
   # update a specific customer
   def update
+    begin
       @customer = Customer.find_by_id(params[:id])
       if @customer.update_attributes(params[:vehicle])
         flash[:notice] = "Customer updated"
         redirect_to @customer
       else
+        flash[:error] = @customer.errors.full_messages.to_sentence if @customer.errors
         redirect_to edit_customer_path
       end
-    flash[:error] = @customer.errors.full_messages.to_sentence if @customer.errors
+    rescue ActiveRecord::RecordNotFound
+      render_404
+    end
   end
   
   def destroy
     begin
       Customer.find_by_id(params[:id]).destroy
       flash[:notice] = "Customer deleted successfully"
-    rescue
-      flash[:error] = "No Customer Found"
-    end
       redirect_to action: :index
-
+    rescue ActiveRecord::RecordNotFound
+      render_404
+    end
   end
 
 end