diff options
Diffstat (limited to 'app')
-rw-r--r-- | app/models/public_body_category_link.rb | 4 | ||||
-rw-r--r-- | app/models/public_body_heading.rb | 2 |
2 files changed, 3 insertions, 3 deletions
diff --git a/app/models/public_body_category_link.rb b/app/models/public_body_category_link.rb index e5a0d7f25..d1ea5bdbb 100644 --- a/app/models/public_body_category_link.rb +++ b/app/models/public_body_category_link.rb @@ -17,11 +17,11 @@ class PublicBodyCategoryLink < ActiveRecord::Base before_validation :on => :create do unless self.category_display_order - self.category_display_order = PublicBodyCategoryLink.next_display_order(self.public_body_heading_id) + self.category_display_order = PublicBodyCategoryLink.next_display_order(public_body_heading_id) end end - def PublicBodyCategoryLink.next_display_order(heading_id) + def self.next_display_order(heading_id) if last = where(:public_body_heading_id => heading_id).order(:category_display_order).last last.category_display_order + 1 else diff --git a/app/models/public_body_heading.rb b/app/models/public_body_heading.rb index e675162f8..bbed55b29 100644 --- a/app/models/public_body_heading.rb +++ b/app/models/public_body_heading.rb @@ -58,7 +58,7 @@ class PublicBodyHeading < ActiveRecord::Base end end - def PublicBodyHeading.next_display_order + def self.next_display_order if max = maximum(:display_order) max + 1 else |