@@ -52,7 +52,7 @@ def keys
52
52
# Let #has_key? consider defined columns
53
53
#
54
54
def has_key? ( attr_name )
55
- @klass . columns_hash . has_key? ( attr_name . to_s )
55
+ @klass . columns_hash . has_key? ( attr_name )
56
56
end
57
57
58
58
alias_method :include? , :has_key?
@@ -62,11 +62,10 @@ def has_key?(attr_name)
62
62
# Lazily augment and load missing attributes
63
63
#
64
64
def []( attr_name )
65
- attr_s = attr_name . to_s
66
- if interesting_for_scrooge? ( attr_s )
67
- augment_callsite! ( attr_s )
65
+ if interesting_for_scrooge? ( attr_name )
66
+ augment_callsite! ( attr_name )
68
67
fetch_remaining
69
- @scrooge_columns << attr_s
68
+ @scrooge_columns << attr_name
70
69
end
71
70
super
72
71
end
@@ -77,7 +76,7 @@ def fetch(*args, &block)
77
76
end
78
77
79
78
def []=( attr_name , value )
80
- @scrooge_columns << attr_name . to_s
79
+ @scrooge_columns << attr_name
81
80
super
82
81
end
83
82
@@ -111,7 +110,7 @@ def update(hash)
111
110
112
111
def fetch_remaining
113
112
unless @fully_fetched
114
- columns_to_fetch = keys - @scrooge_columns . to_a
113
+ columns_to_fetch = @klass . column_names - @scrooge_columns . to_a
115
114
unless columns_to_fetch . empty?
116
115
fetch_remaining! ( columns_to_fetch )
117
116
end
0 commit comments