From e3b6f6cff25318973df1d0988b65df53a38e014d Mon Sep 17 00:00:00 2001 From: Ryan Daigle Date: Fri, 11 Dec 2009 17:33:26 -0500 Subject: [PATCH] Fix pks scope --- lib/utility_scopes/pks.rb | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/lib/utility_scopes/pks.rb b/lib/utility_scopes/pks.rb index 3105915..3108ba1 100644 --- a/lib/utility_scopes/pks.rb +++ b/lib/utility_scopes/pks.rb @@ -3,13 +3,18 @@ module Pks def self.included(within) - within.class_eval do - named_scope :only_pks, :select => "`#{table_name}`.#{primary_key}" + within.class_eval do extend ClassMethods end end module ClassMethods + + # Can't use named_scope b/c don't have access to table_name etc... yet + def only_pks + scoped(:select => "`#{table_name}`.#{primary_key}") + end + def pks only_pks.collect(&:"#{primary_key}") end