Load budget in stats controller

2017-06-30 21:41:47 +02:00
parent 6c214c266f
commit 51ab4a430e
1 changed files with 5 additions and 0 deletions
--- a/app/controllers/budgets/stats_controller.rb
+++ b/app/controllers/budgets/stats_controller.rb
@@ -1,6 +1,7 @@
 module Budgets
  class StatsController < ApplicationController

+    before_action :load_budget
    load_and_authorize_resource :budget

    def show
@@ -14,5 +15,9 @@ module Budgets
        Budget::Stats.new(@budget).generate
      end

+      def load_budget
+        @budget = Budget.find_by(slug: params[:budget_id]) || Budget.find_by(id: params[:budget_id])
+      end
+
  end
 end